Fix ce ee sync status bug (#740)

* Fix ce ee sync status bug

* Fix mocks

---------

Co-authored-by: Dev Agent <dev-agent@example.com>
Co-authored-by: zhzhang <pulltheflower@163.com>

Author: 3 people

_mocks/opencsg.com/csghub-server/builder/store/database/mock_MirrorTaskStore.go

@@ -19,12 +19,14 @@ type MirrorTaskStore interface {
 	CancelOtherTasksAndCreate(ctx context.Context, task MirrorTask) (MirrorTask, error)
 	Create(ctx context.Context, task MirrorTask) (MirrorTask, error)
 	Update(ctx context.Context, task MirrorTask) (MirrorTask, error)
+	UpdateStatusAndRepoSyncStatus(ctx context.Context, task MirrorTask, syncStatus types.RepositorySyncStatus) (MirrorTask, error)
 	FindByMirrorID(ctx context.Context, mirrorID int64) (*MirrorTask, error)
 	Delete(ctx context.Context, ID int64) error
 	GetHighestPriorityByTaskStatus(ctx context.Context, status []types.MirrorTaskStatus) (MirrorTask, error)
 	SetMirrorCurrentTaskID(ctx context.Context, task MirrorTask) error
 	FindByID(ctx context.Context, ID int64) (*MirrorTask, error)
 	ListByStatusWithPriority(ctx context.Context, status []types.MirrorTaskStatus, per, page int) ([]MirrorTask, error)
+	ResetRunningTasks(ctx context.Context, fromStatus types.MirrorTaskStatus, toStatus types.MirrorTaskStatus) (int, error)
 }
 
 func NewMirrorTaskStore() MirrorTaskStore {
@@ -295,3 +297,44 @@ func (m *mirrorTaskStoreImpl) CancelOtherTasksAndCreate(ctx context.Context, tas
 	})
 	return task, errorx.HandleDBError(err, nil)
 }
+
+func (m *mirrorTaskStoreImpl) ResetRunningTasks(ctx context.Context, fromStatus types.MirrorTaskStatus, toStatus types.MirrorTaskStatus) (int, error) {
+	var task MirrorTask
+	result, err := m.db.Operator.Core.NewUpdate().
+		Model(&task).
+		Set("status = ?", toStatus).
+		Where("status = ?", fromStatus).
+		Exec(ctx)
+	if err != nil {
+		return 0, errorx.HandleDBError(err, nil)
+	}
+	rowsAffected, err := result.RowsAffected()
+	if err != nil {
+		return 0, errorx.HandleDBError(err, nil)
+	}
+	return int(rowsAffected), nil
+}
+
+func (m *mirrorTaskStoreImpl) UpdateStatusAndRepoSyncStatus(ctx context.Context, task MirrorTask, syncStatus types.RepositorySyncStatus) (MirrorTask, error) {
+	err := m.db.Operator.Core.RunInTx(ctx, nil, func(ctx context.Context, tx bun.Tx) error {
+		_, err := tx.NewUpdate().
+			Model(&task).
+			WherePK().
+			Exec(ctx)
+		if err != nil {
+			return err
+		}
+
+		_, err = tx.NewUpdate().
+			Model(&Repository{}).
+			Set("sync_status = ?", syncStatus).
+			Where("id = ?", task.Mirror.RepositoryID).
+			Exec(ctx)
+		if err != nil {
+			return err
+		}
+
+		return nil
+	})
+	return task, errorx.HandleDBError(err, nil)
+}