mirror of
https://git.mirrors.martin98.com/https://github.com/SigNoz/signoz
synced 2025-07-24 05:04:27 +08:00
cb155a1172
* feat: add interface for opamp.AgentConfigProvider * feat: add iface and plumbing for generating recommended conf in opamp/agent * feat: get opamp server config provider tests started * chore: add test scenario for agent connection without a config recommendation * chore: add test scenario for agent connection with a config recommendation * chore: add test for validating config deployment status gets reported * chore: add test for rolling out latest config recommendations when config changes * chore: wrap up opamp server lifecycle tests * chore: some tests cleanup * chore: get all tests passing * chore: update opamp server init logic in ee query service * chore: some cleanup * chore: some final cleanup
129 lines
3.1 KiB
Go
129 lines
3.1 KiB
Go
package opamp
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/open-telemetry/opamp-go/protobufs"
|
|
"github.com/open-telemetry/opamp-go/server"
|
|
"github.com/open-telemetry/opamp-go/server/types"
|
|
model "go.signoz.io/signoz/pkg/query-service/app/opamp/model"
|
|
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
var opAmpServer *Server
|
|
|
|
type Server struct {
|
|
server server.OpAMPServer
|
|
agents *model.Agents
|
|
logger *zap.Logger
|
|
capabilities int32
|
|
|
|
agentConfigProvider AgentConfigProvider
|
|
|
|
// cleanups to be run when stopping the server
|
|
cleanups []func()
|
|
}
|
|
|
|
const capabilities = protobufs.ServerCapabilities_ServerCapabilities_AcceptsEffectiveConfig |
|
|
protobufs.ServerCapabilities_ServerCapabilities_OffersRemoteConfig |
|
|
protobufs.ServerCapabilities_ServerCapabilities_AcceptsStatus
|
|
|
|
func InitializeServer(
|
|
agents *model.Agents, agentConfigProvider AgentConfigProvider,
|
|
) *Server {
|
|
if agents == nil {
|
|
agents = &model.AllAgents
|
|
}
|
|
|
|
opAmpServer = &Server{
|
|
agents: agents,
|
|
agentConfigProvider: agentConfigProvider,
|
|
}
|
|
opAmpServer.server = server.New(zap.S())
|
|
return opAmpServer
|
|
}
|
|
|
|
func (srv *Server) Start(listener string) error {
|
|
settings := server.StartSettings{
|
|
Settings: server.Settings{
|
|
Callbacks: server.CallbacksStruct{
|
|
OnMessageFunc: srv.OnMessage,
|
|
OnConnectionCloseFunc: srv.onDisconnect,
|
|
},
|
|
},
|
|
ListenEndpoint: listener,
|
|
}
|
|
|
|
unsubscribe := srv.agentConfigProvider.SubscribeToConfigUpdates(func() {
|
|
err := srv.agents.RecommendLatestConfigToAll(srv.agentConfigProvider)
|
|
if err != nil {
|
|
zap.S().Errorf(
|
|
"could not roll out latest config recommendation to connected agents: %w", err,
|
|
)
|
|
}
|
|
})
|
|
srv.cleanups = append(srv.cleanups, unsubscribe)
|
|
|
|
return srv.server.Start(settings)
|
|
}
|
|
|
|
func (srv *Server) Stop() {
|
|
for _, cleanup := range srv.cleanups {
|
|
defer cleanup()
|
|
}
|
|
|
|
srv.server.Stop(context.Background())
|
|
}
|
|
|
|
func (srv *Server) onDisconnect(conn types.Connection) {
|
|
srv.agents.RemoveConnection(conn)
|
|
}
|
|
|
|
func (srv *Server) OnMessage(conn types.Connection, msg *protobufs.AgentToServer) *protobufs.ServerToAgent {
|
|
agentID := msg.InstanceUid
|
|
|
|
agent, created, err := srv.agents.FindOrCreateAgent(agentID, conn)
|
|
if err != nil {
|
|
zap.S().Error("Failed to find or create agent %q: %v", agentID, err)
|
|
// TODO: handle error
|
|
}
|
|
|
|
if created {
|
|
agent.CanLB = model.ExtractLbFlag(msg.AgentDescription)
|
|
zap.S().Debugf(
|
|
"New agent added:",
|
|
zap.Bool("canLb", agent.CanLB),
|
|
zap.String("ID", agent.ID),
|
|
zap.Any("status", agent.CurrentStatus),
|
|
)
|
|
}
|
|
|
|
var response *protobufs.ServerToAgent
|
|
response = &protobufs.ServerToAgent{
|
|
InstanceUid: agentID,
|
|
Capabilities: uint64(capabilities),
|
|
}
|
|
|
|
agent.UpdateStatus(msg, response, srv.agentConfigProvider)
|
|
|
|
return response
|
|
}
|
|
|
|
// global var methods to support singleton pattern. we want to discourage
|
|
// allow multiple servers in one installation
|
|
func Ready() bool {
|
|
if opAmpServer == nil {
|
|
return false
|
|
}
|
|
if opAmpServer.agents.Count() == 0 {
|
|
zap.S().Warnf("no agents available, all agent config requests will be rejected")
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
func Subscribe(agentId string, hash string, f model.OnChangeCallback) {
|
|
model.ListenToConfigUpdate(agentId, hash, f)
|
|
}
|