agent_tracker

type AgentPodInfo struct {
	AgentId int64 `protobuf:"varint,1,opt,name=agent_id" json:"agent_id,omitempty"`
	PodId   int64 `protobuf:"varint,2,opt,name=pod_id" json:"pod_id,omitempty"`
	// contains filtered or unexported fields
}

type ConnectedAgentkInfo struct {
	AgentMeta    *agentk.Meta           `protobuf:"bytes,1,opt,name=agent_meta" json:"agent_meta,omitempty"`
	ConnectedAt  *timestamppb.Timestamp `protobuf:"bytes,2,opt,name=connected_at" json:"connected_at,omitempty"`
	ConnectionId int64                  `protobuf:"varint,3,opt,name=connection_id" json:"connection_id,omitempty"`
	AgentId      int64                  `protobuf:"varint,4,opt,name=agent_id" json:"agent_id,omitempty"`
	ProjectId    int64                  `protobuf:"varint,5,opt,name=project_id" json:"project_id,omitempty"`
	// contains filtered or unexported fields
}

type ConnectedAgentwInfo struct {
	AgentMeta    *agentw.Meta           `protobuf:"bytes,1,opt,name=agent_meta" json:"agent_meta,omitempty"`
	ConnectedAt  *timestamppb.Timestamp `protobuf:"bytes,2,opt,name=connected_at" json:"connected_at,omitempty"`
	ConnectionId int64                  `protobuf:"varint,3,opt,name=connection_id" json:"connection_id,omitempty"`
	WorkspaceId  int64                  `protobuf:"varint,4,opt,name=workspace_id" json:"workspace_id,omitempty"`
	// contains filtered or unexported fields
}

type ConnectedRunnerControllerInfo struct {
	AgentMeta    *runnerc.Meta          `protobuf:"bytes,1,opt,name=agent_meta" json:"agent_meta,omitempty"`
	ConnectedAt  *timestamppb.Timestamp `protobuf:"bytes,2,opt,name=connected_at" json:"connected_at,omitempty"`
	ConnectionId int64                  `protobuf:"varint,3,opt,name=connection_id" json:"connection_id,omitempty"`
	AgentId      int64                  `protobuf:"varint,4,opt,name=agent_id" json:"agent_id,omitempty"`
	// contains filtered or unexported fields
}

type DisconnectAgentkInfo struct {
	AgentMeta    *agentk.Meta `protobuf:"bytes,1,opt,name=agent_meta" json:"agent_meta,omitempty"`
	ConnectionId int64        `protobuf:"varint,2,opt,name=connection_id" json:"connection_id,omitempty"`
	AgentId      int64        `protobuf:"varint,3,opt,name=agent_id" json:"agent_id,omitempty"`
	ProjectId    int64        `protobuf:"varint,4,opt,name=project_id" json:"project_id,omitempty"`
	// contains filtered or unexported fields
}

type DisconnectAgentwInfo struct {
	AgentMeta    *agentw.Meta `protobuf:"bytes,1,opt,name=agent_meta" json:"agent_meta,omitempty"`
	ConnectionId int64        `protobuf:"varint,2,opt,name=connection_id" json:"connection_id,omitempty"`
	WorkspaceId  int64        `protobuf:"varint,3,opt,name=workspace_id" json:"workspace_id,omitempty"`
	// contains filtered or unexported fields
}

type DisconnectRunnerControllerInfo struct {
	AgentMeta    *runnerc.Meta `protobuf:"bytes,1,opt,name=agent_meta" json:"agent_meta,omitempty"`
	ConnectionId int64         `protobuf:"varint,2,opt,name=connection_id" json:"connection_id,omitempty"`
	AgentId      int64         `protobuf:"varint,3,opt,name=agent_id" json:"agent_id,omitempty"`
	// contains filtered or unexported fields
}

type ExpiringRegisterer interface {
	// RegisterAgentkExpiring registers agentk connection with the tracker.
	// Registration will expire if not refreshed using this method.
	RegisterAgentkExpiring(ctx context.Context, info *ConnectedAgentkInfo) error
	// RegisterAgentwExpiring registers agentw connection with the tracker.
	// Registration will expire if not refreshed using this method.
	RegisterAgentwExpiring(ctx context.Context, info *ConnectedAgentwInfo) error
	// RegisterRunnerControllerExpiring registers runnerc connection with the tracker.
	// Registration will expire if not refreshed using this method.
	RegisterRunnerControllerExpiring(ctx context.Context, info *ConnectedRunnerControllerInfo) error
	// UnregisterAgentk unregisters agentk connection with the tracker.
	// This may be called to intentionally unregister the agentk,
	// even though it may eventually expire.
	UnregisterAgentk(ctx context.Context, info *DisconnectAgentkInfo) error
	// UnregisterAgentw unregisters agentw connection with the tracker.
	// This may be called to intentionally unregister the agentw,
	// even though it may eventually expire.
	UnregisterAgentw(ctx context.Context, info *DisconnectAgentwInfo) error
	// UnregisterRunnerController unregisters runnerc connection with the tracker.
	// This may be called to intentionally unregister the runnerc,
	// even though it may eventually expire.
	UnregisterRunnerController(ctx context.Context, info *DisconnectRunnerControllerInfo) error
}

type Querier interface {
	GetAgentkConnectionsByID(ctx context.Context, agentID int64) iter.Seq[*ConnectedAgentkInfo]
	GetAgentwConnectionsByID(ctx context.Context, agentID int64) iter.Seq[*ConnectedAgentwInfo]
	GetRunnerControllerConnectionsByID(ctx context.Context, agentID int64) iter.Seq[*ConnectedRunnerControllerInfo]
	GetConnectedAgentsCount(ctx context.Context) (int64, error)
	CountAgentksByAgentVersions(ctx context.Context) (map[string]int64, error)
}

type RedisTracker struct {
	// contains filtered or unexported fields
}

type Tracker interface {
	ExpiringRegisterer
	Querier
	Run(ctx context.Context) error
}