agents

Documentation

Overview

Package agents contains business logic of working with pmm-agent.

Index

• type Registry
  • func NewRegistry(db *reform.DB, prometheus prometheus, qanClient qanClient) *Registry
  • func (r *Registry) CheckConnectionToService(ctx context.Context, service *models.Service, agent *models.Agent) error
  • func (r *Registry) Collect(ch chan<- prom.Metric)
  • func (r *Registry) Describe(ch chan<- *prom.Desc)
  • func (r *Registry) IsConnected(pmmAgentID string) bool
  • func (r *Registry) Kick(ctx context.Context, pmmAgentID string)
  • func (r *Registry) Run(stream agentpb.Agent_ConnectServer) error
  • func (r *Registry) SendSetStateRequest(ctx context.Context, pmmAgentID string)

Types

type Registry

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

Registry keeps track of all connected pmm-agents.

TODO Split into several types?

func NewRegistry

func NewRegistry(db *reform.DB, prometheus prometheus, qanClient qanClient) *Registry

NewRegistry creates a new registry with given database connection.

func (*Registry) CheckConnectionToService

func (r *Registry) CheckConnectionToService(ctx context.Context, service *models.Service, agent *models.Agent) error

CheckConnectionToService sends request to pmm-agent to check connection to service.

func (*Registry) Collect

func (r *Registry) Collect(ch chan<- prom.Metric)

Collect implement prometheus.Collector.

func (*Registry) Describe

func (r *Registry) Describe(ch chan<- *prom.Desc)

Describe implements prometheus.Collector.

func (*Registry) IsConnected

func (r *Registry) IsConnected(pmmAgentID string) bool

IsConnected returns true if pmm-agent with given ID is currently connected, false otherwise.

func (*Registry) Kick

func (r *Registry) Kick(ctx context.Context, pmmAgentID string)

Kick disconnects pmm-agent with given ID.

func (*Registry) Run

func (r *Registry) Run(stream agentpb.Agent_ConnectServer) error

Run takes over pmm-agent gRPC stream and runs it until completion.

func (*Registry) SendSetStateRequest

func (r *Registry) SendSetStateRequest(ctx context.Context, pmmAgentID string)

SendSetStateRequest sends SetStateRequest to pmm-agent with given ID.