package main

import (
	"context"
	"fmt"

	influxdb2 "github.com/influxdata/influxdb-client-go/v2"
	"github.com/influxdata/influxdb-client-go/v2/api"
	"github.com/influxdata/influxdb-client-go/v2/api/write"
)

type InfluxWriter struct {
	client    influxdb2.Client
	write_api api.WriteAPIBlocking
	channel   chan *write.Point
}

func (w *InfluxWriter) Write(point *write.Point) {
	w.channel <- point
}

func (w *InfluxWriter) Init(
	url string,
	token string,
	org string,
	bucket string,
) {
	w.channel = make(chan *write.Point)
	w.client = influxdb2.NewClient(url, token)
	w.write_api = w.client.WriteAPIBlocking(org, bucket)

	go func() {
		for point := range w.channel {
			err := w.write_api.WritePoint(context.Background(), point)
			if err != nil {
				fmt.Printf("Error writing to InfluxDB: %s", err)
			}
		}
	}()
}