remove default bucket

This commit is contained in:
2023-04-09 19:03:57 -07:00
parent 1eac67d09a
commit ac5869e63b

29
arma.go
View File

@@ -54,7 +54,6 @@ type influxSettings struct {
Host string `json:"host"`
Token string `json:"token"`
Org string `json:"org"`
DefaultBucket string `json:"defaultBucket"`
}
type arma3Settings struct {
@@ -69,13 +68,13 @@ type settingsJson struct {
func connectToInflux() string {
if influxConnectionSettings.Host == "" {
logLine("connectToInflux", `["influxConnectionSettings.Host is empty", "ERROR"]`)
return "ERROR"
return "Error connecting to Influx"
}
DB_CLIENT = influxdb2.NewClientWithOptions(influxConnectionSettings.Host, influxConnectionSettings.Token, influxdb2.DefaultOptions().SetBatchSize(500).SetFlushInterval(2000))
logLine("connectToInflux", `["DB_CLIENT created", "INFO"]`)
return "CONNECTED"
return "Connected to Influx successfully"
}
func deinitialize() {
@@ -83,6 +82,20 @@ func deinitialize() {
DB_CLIENT.Close()
}
// sanitize line protocol for influx
func sanitizeLineProtocol(line string) string {
// replace all spaces with underscores
// line = strings.ReplaceAll(line, ` `, `\ `)
// replace all commas with underscores
// line = strings.ReplaceAll(line, `,`, `\,`)
// replace all equals with underscores
// line = strings.ReplaceAll(line, "=", "_")
// replace all quotes with underscores
// line = strings.ReplaceAll(line, "\"", "_")
return line
}
func getDir() string {
dir, err := os.Getwd()
if err != nil {
@@ -116,7 +129,6 @@ func loadSettings() (dir string, result string, host string) {
Host: "http://localhost:8086",
Token: "my-token",
Org: "my-org",
DefaultBucket: "my-bucket",
}
a3Set := arma3Settings{
RefreshRateMs: 1000,
@@ -253,11 +265,7 @@ func sendToInflux(a3DataRaw *[]string) string {
}
// use custom bucket or default
var customBucket string = trimQuotes(string(a3Data[0]))
bucket := influxConnectionSettings.DefaultBucket
if customBucket != "" {
bucket = customBucket
}
var bucket string = trimQuotes(string(a3Data[0]))
WRITE_API := DB_CLIENT.WriteAPI(influxConnectionSettings.Org, bucket)
@@ -321,11 +329,10 @@ func goRVExtension(output *C.char, outputsize C.size_t, input *C.char) {
if result != "" {
logLine("goRVExtension", result)
temp = fmt.Sprintf(
`["%s", "%s", "%s", "%s", "%d"]`,
`["%s", "%s", "%s", "%d"]`,
EXTENSION_VERSION,
influxConnectionSettings.Host,
influxConnectionSettings.Org,
influxConnectionSettings.DefaultBucket,
a3Settings.RefreshRateMs,
)
}