Method SetScaleSetLastMessageID

database/sql/scalesets.go:459–494

(_ context.Context, scaleSetID uint, lastMessageID int64)

}

func (s *sqlDatabase) SetScaleSetLastMessageID(_ context.Context, scaleSetID uint, lastMessageID int64) (err error) {
	var scaleSet params.ScaleSet
	var rowsAffected int64
	defer func() {
		if err == nil && scaleSet.ID != 0 && rowsAffected > 0 {
			s.sendNotify(common.ScaleSetEntityType, common.UpdateOperation, scaleSet)
		}
	}()
	if err := s.conn.Transaction(func(tx *gorm.DB) error {
		dbSet, err := s.getScaleSetByID(tx, scaleSetID, "Instances", "Enterprise", "Organization", "Repository")
		if err != nil {
			return fmt.Errorf("error fetching scale set: %w", err)
		}
		result := tx.Model(&dbSet).Updates(map[string]interface{}{
			"last_message_id": lastMessageID,
		})
		if result.Error != nil {
			return fmt.Errorf("error saving database entry: %w", result.Error)
		}
		rowsAffected = result.RowsAffected

		// Reload to get the updated value
		dbSet, err = s.getScaleSetByID(tx, scaleSetID, "Instances", "Enterprise", "Organization", "Repository")
		if err != nil {
			return fmt.Errorf("error reloading scale set: %w", err)
		}
		scaleSet, err = s.sqlToCommonScaleSet(dbSet)
		if err != nil {
			return fmt.Errorf("error converting scale set: %w", err)
		}
		return nil
	}); err != nil {
		return fmt.Errorf("error setting last message ID: %w", err)
	}
	return nil
}

func (s *sqlDatabase) SetScaleSetDesiredRunnerCount(_ context.Context, scaleSetID uint, desiredRunnerCount int) (err error) {
	var scaleSet params.ScaleSet