mirror of
https://github.com/mainflux/mainflux.git
synced 2025-04-24 13:48:49 +08:00
3e7bac493c
* Initial Commit: Sync Env Veriables With Docker Deployment Signed-off-by: rodneyosodo <blackd0t@protonmail.com> * Sync Env Vars With Master Signed-off-by: rodneyosodo <blackd0t@protonmail.com> * Remove Altprefix Signed-off-by: rodneyosodo <blackd0t@protonmail.com> * Rename HttpPort to HTTPPort Signed-off-by: rodneyosodo <blackd0t@protonmail.com> * Fix envPrefixDB After Rebase Signed-off-by: rodneyosodo <blackd0t@protonmail.com> * Remove Server Parse Signed-off-by: rodneyosodo <blackd0t@protonmail.com> * Add Provision For TLS on CoAP Signed-off-by: rodneyosodo <blackd0t@protonmail.com> * Fix Exit After Defer Signed-off-by: rodneyosodo <blackd0t@protonmail.com> * Remove Unused Function Signed-off-by: rodneyosodo <blackd0t@protonmail.com> * Document Undocumentated Env Variables Signed-off-by: rodneyosodo <blackd0t@protonmail.com> --------- Signed-off-by: rodneyosodo <blackd0t@protonmail.com> Co-authored-by: Drasko DRASKOVIC <drasko.draskovic@gmail.com>
58 lines
2.5 KiB
Go
58 lines
2.5 KiB
Go
// Copyright (c) Mainflux
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package influxdb
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
influxdb2 "github.com/influxdata/influxdb-client-go/v2"
|
|
"github.com/mainflux/mainflux/internal/env"
|
|
"github.com/mainflux/mainflux/pkg/errors"
|
|
)
|
|
|
|
var (
|
|
errConnect = errors.New("failed to create InfluxDB client")
|
|
errConfig = errors.New("failed to load InfluxDB client configuration from environment variable")
|
|
)
|
|
|
|
type Config struct {
|
|
Protocol string `env:"PROTOCOL" envDefault:"http"`
|
|
Host string `env:"HOST" envDefault:"localhost"`
|
|
Port string `env:"PORT" envDefault:"8086"`
|
|
Username string `env:"ADMIN_USER" envDefault:"mainflux"`
|
|
Password string `env:"ADMIN_PASSWORD" envDefault:"mainflux"`
|
|
DbName string `env:"NAME" envDefault:"mainflux"`
|
|
Bucket string `env:"BUCKET" envDefault:"mainflux-bucket"`
|
|
Org string `env:"ORG" envDefault:"mainflux"`
|
|
Token string `env:"TOKEN" envDefault:"mainflux-token"`
|
|
DBUrl string `env:"DBURL" envDefault:""`
|
|
UserAgent string `env:"USER_AGENT" envDefault:"InfluxDBClient"`
|
|
Timeout time.Duration `env:"TIMEOUT"` // Influxdb client configuration by default has no timeout duration , this field will not have a fallback default timeout duration. Reference: https://pkg.go.dev/github.com/influxdata/influxdb@v1.10.0/client/v2#HTTPConfig
|
|
InsecureSkipVerify bool `env:"INSECURE_SKIP_VERIFY" envDefault:"false"`
|
|
}
|
|
|
|
// Setup load configuration from environment variable, create InfluxDB client and connect to InfluxDB server.
|
|
func Setup(ctx context.Context, envPrefix string) (influxdb2.Client, error) {
|
|
config := Config{}
|
|
if err := env.Parse(&config, env.Options{Prefix: envPrefix}); err != nil {
|
|
return nil, errors.Wrap(errConfig, err)
|
|
}
|
|
return Connect(ctx, config)
|
|
}
|
|
|
|
// Connect create InfluxDB client and connect to InfluxDB server.
|
|
func Connect(ctx context.Context, config Config) (influxdb2.Client, error) {
|
|
client := influxdb2.NewClientWithOptions(config.DBUrl, config.Token,
|
|
influxdb2.DefaultOptions().
|
|
SetUseGZip(true).
|
|
SetFlushInterval(100))
|
|
ctx, cancel := context.WithTimeout(ctx, config.Timeout)
|
|
defer cancel()
|
|
if _, err := client.Ready(ctx); err != nil {
|
|
return nil, errors.Wrap(errConnect, err)
|
|
}
|
|
return client, nil
|
|
}
|