Fix for issue #77, create telegraf database if not exists

influxdata · Aug 10, 2015 · bee1f17 · bee1f17
1 parent a3e20ab
commit bee1f17
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/agent.go b/agent.go
@@ -6,6 +6,7 @@ import (
 	"net/url"
 	"os"
 	"sort"
+	"strings"
 	"sync"
 	"time"
 
@@ -84,6 +85,14 @@ func (a *Agent) Connect() error {
 		return err
 	}
 
+	_, err = c.Query(client.Query{
+		Command: fmt.Sprintf("CREATE DATABASE telegraf"),
+	})
+
+	if err != nil && !strings.Contains(err.Error(), "database already exists") {
+		log.Fatal(err)
+	}
+
 	a.conn = c
 
 	return nil