github.com/diadata-org/diadata@v1.4.593/pkg/dia/helpers/db/influx.go (about)

     1  package db
     2  
     3  import (
     4  	"github.com/diadata-org/diadata/pkg/utils"
     5  	clientInfluxdb "github.com/influxdata/influxdb1-client/v2"
     6  	"github.com/sirupsen/logrus"
     7  )
     8  
     9  var log *logrus.Logger
    10  
    11  func init() {
    12  	log = logrus.New()
    13  }
    14  
    15  // GetInfluxClient returns an influx client connecting through the
    16  // URL given in the environment variable INFLUXURL.
    17  // If INFLUXURL is not set, it connects to @url per default.
    18  func GetInfluxClient(url string) clientInfluxdb.Client {
    19  	var influxClient clientInfluxdb.Client
    20  	var err error
    21  
    22  	address := utils.Getenv("INFLUXURL", url)
    23  	log.Info("INFLUXURL: ", address)
    24  	username := utils.Getenv("INFLUXUSER", "")
    25  	password := utils.Getenv("INFLUXPASSWORD", "")
    26  	influxClient, err = clientInfluxdb.NewHTTPClient(clientInfluxdb.HTTPConfig{
    27  		Addr:     address,
    28  		Username: username,
    29  		Password: password,
    30  	})
    31  
    32  	if err != nil {
    33  		log.Error("NewDataStore influxdb", err)
    34  	}
    35  
    36  	return influxClient
    37  }