mirror of
https://github.com/tiennm99/goclaw.git
synced 2026-10-13 06:15:25 +00:00
fix(heartbeat): log provider override and include in trace tags
This commit is contained in:
1 parent
5c8e0b7af2
commit
985d2fa50d
1 file changed
+11
-1
@@ -252,6 +252,8 @@ func (t *Ticker) runOne(ctx context.Context, hb store.AgentHeartbeat) {
|
||||
if provData, err := t.providerStore.GetProvider(ctx, *hb.ProviderID); err == nil {
|
||||
if prov, err := t.providerReg.GetForTenant(ag.TenantID, provData.Name); err == nil {
|
||||
providerOverride = prov
|
||||
slog.Info("heartbeat.provider_override",
|
||||
"agent", agentKey, "provider", provData.Name)
|
||||
} else {
|
||||
slog.Warn("heartbeat.provider_not_in_registry",
|
||||
"agent", agentKey, "provider_id", hb.ProviderID, "error", err)
|
||||
@@ -275,7 +277,7 @@ func (t *Ticker) runOne(ctx context.Context, hb store.AgentHeartbeat) {
|
||||
ProviderOverride: providerOverride,
|
||||
LightContext: hb.LightContext,
|
||||
TraceName: fmt.Sprintf("Heartbeat [%s]", agentKey),
|
||||
TraceTags: []string{"heartbeat"},
|
||||
TraceTags: heartbeatTraceTags(providerOverride),
|
||||
})
|
||||
|
||||
outcome := <-outCh
|
||||
@@ -507,3 +509,11 @@ func deref(s *string) string {
|
||||
}
|
||||
return *s
|
||||
}
|
||||
|
||||
// heartbeatTraceTags returns trace tags, including provider name when overridden.
|
||||
func heartbeatTraceTags(providerOverride providers.Provider) []string {
|
||||
if providerOverride != nil {
|
||||
return []string{"heartbeat", "provider:" + providerOverride.Name()}
|
||||
}
|
||||
return []string{"heartbeat"}
|
||||
}
|
||||
Reference in new issue
Block a user