Add ability to set sql file as query source for postgresql_extensible input

plugins/inputs/postgresql_extensible/README.md

@@ -44,6 +44,9 @@ The example below has two queries are specified, with the following parameters:
   # Be careful that if the withdbname is set to false you don't have to define
   # the where clause (aka with the dbname)
   #
+  # The script option can be used to speciafy the .sql file path,
+  # in case if script and sqlquery options specified at same time, sqlquery will be used 
+  #
   # the tagvalue field is used to define custom tags (separated by comas).
   # the query is expected to return columns which match the names of the
   # defined tags. The values in these columns must be of a string-type,
@@ -61,7 +64,7 @@ The example below has two queries are specified, with the following parameters:
     withdbname=false
     tagvalue=""
   [[inputs.postgresql_extensible.query]]
-    sqlquery="SELECT * FROM pg_stat_bgwriter"
+    script="your_sql-filepath.sql"
     version=901
     withdbname=false
     tagvalue=""

plugins/inputs/postgresql_extensible/postgresql_extensible.go

@@ -3,7 +3,9 @@ package postgresql_extensible
 import (
 	"bytes"
 	"fmt"
+	"io/ioutil"
 	"log"
+	"os"
 	"strings"
 
 	// register in driver.
@@ -25,6 +27,7 @@ type Postgresql struct {
 
 type query []struct {
 	Sqlquery    string
+	Script      string
 	Version     int
 	Withdbname  bool
 	Tagvalue    string
@@ -75,7 +78,10 @@ var sampleConfig = `
   ## field is used to define custom tags (separated by commas)
   ## The optional "measurement" value can be used to override the default
   ## output measurement name ("postgresql").
-  #
+  ##
+  ## The script option can be used to speciafy the .sql file path,
+  ## in case if script and sqlquery options specified at same time, sqlquery will be used
+  ##
   ## Structure :
   ## [[inputs.postgresql_extensible.query]]
   ##   sqlquery string
@@ -108,6 +114,20 @@ func (p *Postgresql) IgnoredColumns() map[string]bool {
 	return ignoredColumns
 }
 
+func ReadQueryFromFile(filePath string) (string, error) {
+	file, err := os.Open(filePath)
+	if err != nil {
+		return "", err
+	}
+	defer file.Close()
+
+	qyery, err := ioutil.ReadAll(file)
+	if err != nil {
+		return "", err
+	}
+	return string(qyery), err
+}
+
 func (p *Postgresql) Gather(acc telegraf.Accumulator) error {
 	var (
 		err         error
@@ -130,7 +150,14 @@ func (p *Postgresql) Gather(acc telegraf.Accumulator) error {
 	// Query is not run if Database version does not match the query version.
 	for i := range p.Query {
 		sql_query = p.Query[i].Sqlquery
+		if sql_query == "" {
+			sql_query, err = ReadQueryFromFile(p.Query[i].Script)
+			if err != nil {
+				return err
+			}
+		}
 		tag_value = p.Query[i].Tagvalue
+
 		if p.Query[i].Measurement != "" {
 			meas_name = p.Query[i].Measurement
 		} else {

plugins/inputs/postgresql_extensible/postgresql_extensible_test.go

@@ -145,7 +145,7 @@ func TestPostgresqlFieldOutput(t *testing.T) {
 	}
 
 	acc := queryRunner(t, query{{
-		Sqlquery:   "select * from pg_stat_database",
+		Script:     "select * from pg_stat_database",
 		Version:    901,
 		Withdbname: false,
 		Tagvalue:   "",
@@ -201,6 +201,30 @@ func TestPostgresqlFieldOutput(t *testing.T) {
 	}
 }
 
+func TestPostgresqlSqlScript(t *testing.T) {
+	q := query{{
+		Script:     "testdata/test.sql",
+		Version:    901,
+		Withdbname: false,
+		Tagvalue:   "",
+	}}
+	p := &Postgresql{
+		Service: postgresql.Service{
+			Address: fmt.Sprintf(
+				"host=%s user=postgres sslmode=disable",
+				testutil.GetLocalHost(),
+			),
+			IsPgBouncer: false,
+		},
+		Databases: []string{"postgres"},
+		Query:     q,
+	}
+	var acc testutil.Accumulator
+	p.Start(&acc)
+
+	require.NoError(t, acc.GatherError(p.Gather))
+}
+
 func TestPostgresqlIgnoresUnwantedColumns(t *testing.T) {
 	if testing.Short() {
 		t.Skip("Skipping integration test in short mode")

plugins/inputs/postgresql_extensible/testdata/test.sql

@@ -0,0 +1 @@
+select * from pg_stat_database