diff --git a/agent.go b/agent.go index f86e27920..1c90bf1a4 100644 --- a/agent.go +++ b/agent.go @@ -67,7 +67,7 @@ func NewAgent(config *Config) (*Agent, error) { return agent, nil } -// Connect connects to the agent's config URL +// Connect connects to all configured outputs func (a *Agent) Connect() error { for _, o := range a.outputs { err := o.output.Connect() @@ -78,6 +78,15 @@ func (a *Agent) Connect() error { return nil } +// Close closes the connection to all configured outputs +func (a *Agent) Close() error { + var err error + for _, o := range a.outputs { + err = o.output.Close() + } + return err +} + // LoadOutputs loads the agent's outputs func (a *Agent) LoadOutputs() ([]string, error) { var names []string diff --git a/outputs/influxdb/influxdb.go b/outputs/influxdb/influxdb.go index 1153f064b..96505a4d7 100644 --- a/outputs/influxdb/influxdb.go +++ b/outputs/influxdb/influxdb.go @@ -52,6 +52,11 @@ func (i *InfluxDB) Connect() error { return nil } +func (i *InfluxDB) Close() error { + // InfluxDB client does not provide a Close() function + return nil +} + func (i *InfluxDB) Write(bp client.BatchPoints) error { bp.Database = i.Database if _, err := i.conn.Write(bp); err != nil { diff --git a/outputs/registry.go b/outputs/registry.go index ccc40f9b2..a2f22f73b 100644 --- a/outputs/registry.go +++ b/outputs/registry.go @@ -6,6 +6,7 @@ import ( type Output interface { Connect() error + Close() error Write(client.BatchPoints) error }