fix: [CDE-202]: update gitspace database funcs (#2809)

* remove named returns
* update gitspace database funcs
pull/3576/head
Deepak Bhatt 2024-10-16 04:17:04 +00:00 committed by Harness
parent 80dc6ff0b0
commit 7790c19a7f
3 changed files with 6 additions and 53 deletions

View File

@ -20,7 +20,6 @@ import (
"time"
"github.com/harness/gitness/types"
"github.com/harness/gitness/types/enum"
)
func (c *Service) UpdateInstance(
@ -34,16 +33,3 @@ func (c *Service) UpdateInstance(
}
return nil
}
func (c *Service) UpdateAllStateInstance(
ctx context.Context,
status enum.GitspaceInstanceStateType,
instancesIDs []int64,
) error {
updatedUnixTime := time.Now().UnixMilli()
err := c.gitspaceInstanceStore.BulkUpdateState(ctx, status, updatedUnixTime, instancesIDs)
if err != nil {
return fmt.Errorf("failed to update all statue of gitspace instances: %w", err)
}
return nil
}

View File

@ -676,19 +676,11 @@ type (
// Update tries to update a gitspace instance in the datastore with optimistic locking.
Update(ctx context.Context, gitspaceInstance *types.GitspaceInstance) error
// BulkUpdateState updates state of given gitspace instance IDs to given state
BulkUpdateState(
ctx context.Context,
state enum.GitspaceInstanceStateType,
updateTimeUnix int64,
gitspaceInstanceIDs []int64,
) error
// List lists the gitspace instance present in a parent space ID in the datastore.
List(ctx context.Context, filter *types.GitspaceFilter) ([]*types.GitspaceInstance, error)
// ListDead lists dead gitspace instances whose heartbeat stopped coming after the given time.
ListDead(ctx context.Context, filter *types.GitspaceFilter) (gitInstanceIDs []int64, err error)
ListDead(ctx context.Context, filter *types.GitspaceFilter) ([]*types.GitspaceInstance, error)
// FetchInactiveGitspaceConfigs lists the inactive gitspace instance present in the datastore
FetchInactiveGitspaceConfigs(ctx context.Context, filter *types.GitspaceFilter) ([]int64, error)

View File

@ -256,32 +256,6 @@ func (g gitspaceInstanceStore) Update(
return nil
}
func (g gitspaceInstanceStore) BulkUpdateState(
ctx context.Context,
state enum.GitspaceInstanceStateType,
updateTimeUnix int64,
gitspaceInstanceIDs []int64,
) error {
stmt := database.Builder.
Update(gitspaceInstanceTable).
Set("gits_state", state).
Set("gits_updated", updateTimeUnix).
Where(squirrel.Eq{"gits_id": gitspaceInstanceIDs})
sqlStr, args, err := stmt.ToSql()
if err != nil {
return errors.Wrap(err, "Failed to convert squirrel builder to sql")
}
db := dbtx.GetAccessor(ctx, g.db)
if _, err = db.ExecContext(ctx, sqlStr, args...); err != nil {
return database.ProcessSQLErrorf(ctx, err,
"Failed to update gitspace instances for %v", gitspaceInstanceIDs)
}
return nil
}
func (g gitspaceInstanceStore) FindLatestByGitspaceConfigID(
ctx context.Context,
gitspaceConfigID int64,
@ -333,9 +307,9 @@ func (g gitspaceInstanceStore) List(
func (g gitspaceInstanceStore) ListDead(
ctx context.Context,
filter *types.GitspaceFilter,
) (gitInstanceIDs []int64, err error) {
) ([]*types.GitspaceInstance, error) {
stmt := database.Builder.
Select("gits_id").
Select(gitspaceInstanceSelectColumns).
From(gitspaceInstanceTable).
Where(squirrel.Lt{"gits_last_heartbeat": filter.LastHeartBeatBefore}).
Where(squirrel.Eq{"gits_state": filter.State}).
@ -346,11 +320,12 @@ func (g gitspaceInstanceStore) ListDead(
return nil, errors.Wrap(err, "Failed to convert squirrel builder to sql")
}
var gitspaceInstances []*types.GitspaceInstance
db := dbtx.GetAccessor(ctx, g.db)
if err = db.SelectContext(ctx, &gitInstanceIDs, sqlStr, args...); err != nil {
if err = db.SelectContext(ctx, &gitspaceInstances, sqlStr, args...); err != nil {
return nil, database.ProcessSQLErrorf(ctx, err, "Failed executing gitspace instance list query")
}
return gitInstanceIDs, nil
return gitspaceInstances, nil
}
func (g gitspaceInstanceStore) FetchInactiveGitspaceConfigs(