Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions database/sql/scalesets.go
Original file line number Diff line number Diff line change
Expand Up @@ -389,7 +389,7 @@ func (s *sqlDatabase) SetScaleSetLastMessageID(_ context.Context, scaleSetID uin
}
}()
if err := s.conn.Transaction(func(tx *gorm.DB) error {
dbSet, err := s.getScaleSetByID(tx, scaleSetID)
dbSet, err := s.getScaleSetByID(tx, scaleSetID, "Instances", "Enterprise", "Organization", "Repository")
if err != nil {
return errors.Wrap(err, "fetching scale set")
}
Expand All @@ -416,7 +416,7 @@ func (s *sqlDatabase) SetScaleSetDesiredRunnerCount(_ context.Context, scaleSetI
}
}()
if err := s.conn.Transaction(func(tx *gorm.DB) error {
dbSet, err := s.getScaleSetByID(tx, scaleSetID)
dbSet, err := s.getScaleSetByID(tx, scaleSetID, "Instances", "Enterprise", "Organization", "Repository")
if err != nil {
return errors.Wrap(err, "fetching scale set")
}
Expand Down
4 changes: 2 additions & 2 deletions database/sql/util.go
Original file line number Diff line number Diff line change
Expand Up @@ -362,12 +362,12 @@ func (s *sqlDatabase) sqlToCommonScaleSet(scaleSet ScaleSet) (params.ScaleSet, e
}
}

if scaleSet.OrgID != nil && scaleSet.Organization.Name != "" {
if scaleSet.OrgID != nil {
ret.OrgID = scaleSet.OrgID.String()
ret.OrgName = scaleSet.Organization.Name
}

if scaleSet.EnterpriseID != nil && scaleSet.Enterprise.Name != "" {
if scaleSet.EnterpriseID != nil {
ret.EnterpriseID = scaleSet.EnterpriseID.String()
ret.EnterpriseName = scaleSet.Enterprise.Name
}
Expand Down
2 changes: 1 addition & 1 deletion params/params.go
Original file line number Diff line number Diff line change
Expand Up @@ -570,7 +570,7 @@ func (p ScaleSet) GetEntity() (ForgeEntity, error) {
EntityType: ForgeEntityTypeEnterprise,
}, nil
}
return ForgeEntity{}, fmt.Errorf("pool has no associated entity")
return ForgeEntity{}, fmt.Errorf("scale set has no associated entity")
}

func (p *ScaleSet) ScaleSetType() ForgeEntityType {
Expand Down
7 changes: 6 additions & 1 deletion runner/pool/pool.go
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,12 @@ const (
)

func NewEntityPoolManager(ctx context.Context, entity params.ForgeEntity, instanceTokenGetter auth.InstanceTokenGetter, providers map[string]common.Provider, store dbCommon.Store) (common.PoolManager, error) {
ctx = garmUtil.WithSlogContext(ctx, slog.Any("pool_mgr", entity.String()), slog.Any("pool_type", entity.EntityType))
ctx = garmUtil.WithSlogContext(
ctx,
slog.Any("pool_mgr", entity.String()),
slog.Any("endpoint", entity.Credentials.Endpoint.Name),
slog.Any("pool_type", entity.EntityType),
)
ghc, err := ghClient.Client(ctx, entity)
if err != nil {
return nil, errors.Wrap(err, "getting github client")
Expand Down
2 changes: 1 addition & 1 deletion workers/cache/cache.go
Original file line number Diff line number Diff line change
Expand Up @@ -365,7 +365,7 @@ func (w *Worker) handleScaleSetEvent(event common.ChangePayload) {
}
entity, err := scaleSet.GetEntity()
if err != nil {
slog.DebugContext(w.ctx, "getting entity from pool", "error", err)
slog.DebugContext(w.ctx, "getting entity from scale set", "error", err)
return
}

Expand Down
10 changes: 7 additions & 3 deletions workers/cache/gitea_tools.go
Original file line number Diff line number Diff line change
Expand Up @@ -15,9 +15,11 @@
package cache

import (
"context"
"encoding/json"
"fmt"
"io"
"log/slog"
"net/http"
"strings"
"time"
Expand Down Expand Up @@ -139,7 +141,7 @@ func (g GiteaEntityTools) MinimumVersion() (GiteaEntityTool, bool) {
return GiteaEntityTool{}, false
}

func getTools() ([]commonParams.RunnerApplicationDownload, error) {
func getTools(ctx context.Context) ([]commonParams.RunnerApplicationDownload, error) {
resp, err := http.Get(GiteaRunnerReleasesURL)
if err != nil {
return nil, err
Expand Down Expand Up @@ -170,11 +172,13 @@ func getTools() ([]commonParams.RunnerApplicationDownload, error) {
for _, asset := range latest.Assets {
arch, err := asset.GetArch()
if err != nil {
return nil, fmt.Errorf("getting arch: %w", err)
slog.InfoContext(ctx, "ignoring unrecognized tools arch", "tool", asset.Name)
continue
}
os, err := asset.GetOS()
if err != nil {
return nil, fmt.Errorf("getting os: %w", err)
slog.InfoContext(ctx, "ignoring unrecognized tools os", "tool", asset.Name)
continue
}
ret = append(ret, commonParams.RunnerApplicationDownload{
OS: os,
Expand Down
4 changes: 2 additions & 2 deletions workers/cache/tool_cache.go
Original file line number Diff line number Diff line change
Expand Up @@ -162,7 +162,7 @@ func (t *toolsUpdater) giteaUpdateLoop() {
randInt = big.NewInt(0)
}
t.sleepWithCancel(time.Duration(randInt.Int64()) * time.Millisecond)
tools, err := getTools()
tools, err := getTools(t.ctx)
if err != nil {
t.addStatusEvent(fmt.Sprintf("failed to update gitea tools: %q", err), params.EventError)
} else {
Expand All @@ -181,7 +181,7 @@ func (t *toolsUpdater) giteaUpdateLoop() {
case <-t.ctx.Done():
return
case <-ticker.C:
tools, err := getTools()
tools, err := getTools(t.ctx)
if err != nil {
t.addStatusEvent(fmt.Sprintf("failed to update gitea tools: %q", err), params.EventError)
slog.DebugContext(t.ctx, "failed to update gitea tools", "error", err)
Expand Down
Loading