Adding a Close() function to the Output interface and to the agent

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

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 {

outputs/registry.go

@@ -6,6 +6,7 @@ import (
 
 type Output interface {
 	Connect() error
+	Close() error
 	Write(client.BatchPoints) error
 }