feat: add connect stream to connection manager

This commit is contained in:
2026-05-23 14:03:43 +03:00
parent b157d7c32d
commit c41cbc3c2f
7 changed files with 65 additions and 19 deletions
@@ -13,10 +13,10 @@ import (
)
type AgentConnection struct {
stream streamConn
stream StreamConn
heartbeat heartbeatStore
log zerolog.Logger
status statusAgent
status StatusAgent
AgentID string
response *ResponseStore
ctx context.Context
@@ -24,7 +24,7 @@ type AgentConnection struct {
heartbeatTimeoutMS int
}
func newAgentConnection(agentID string, stream streamConn, heartbeat heartbeatStore, status statusAgent, heartbeatTimeoutMS int, logger zerolog.Logger) *AgentConnection {
func newAgentConnection(agentID string, stream StreamConn, heartbeat heartbeatStore, status StatusAgent, heartbeatTimeoutMS int, logger zerolog.Logger) *AgentConnection {
response := NewResponseStore()
logger = logger.With().Str("agentID", agentID).Logger()
ctx, cancel := context.WithCancel(stream.Context())