agent_tracker

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

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

type ConnectedAgentkInfoCallback func(*ConnectedAgentkInfo) (done bool, err error)

type ConnectedAgentkInfoCollector []*ConnectedAgentkInfo

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

type ExpiringRegisterer interface {
	// RegisterExpiring registers connection with the tracker.
	// Registration will expire if not refreshed using this method.
	RegisterExpiring(ctx context.Context, info *ConnectedAgentkInfo) error
	// Unregister unregisters connection with the tracker.
	// This may be called to intentionally unregister the agent,
	// even though it may eventually expire.
	Unregister(ctx context.Context, info *DisconnectAgentkInfo) error
}

type Querier interface {
	GetConnectionsByAgentID(ctx context.Context, agentKey api.AgentKey, cb ConnectedAgentkInfoCallback) error
	GetConnectionsByProjectID(ctx context.Context, projectID int64, cb ConnectedAgentkInfoCallback) error
	GetConnectedAgentsCount(ctx context.Context) (int64, error)
	CountAgentsByAgentVersions(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
}