// influxdb
package main

import (
	"fmt"
	//	"time"

	influxdb "github.com/influxdata/influxdb/client/v2"
)

func writeStats(mimp *MongoImp) {
	if opts.Debug {
		fmt.Printf("writing to influxdb server: %s", opts.Influxdb)
	}

	c, err := influxdb.NewHTTPClient(influxdb.HTTPConfig{
		Addr: opts.Influxdb,
	})
	if err != nil {
		fmt.Printf("Error: %+v\n", err)
		return
	}
	defer c.Close()

	bp, err := influxdb.NewBatchPoints(influxdb.BatchPointsConfig{
		Database:  "dovecot",
		Precision: "s",
	})
	if err != nil {
		fmt.Printf("Error: %+v\n", err)
		return
	}

	tags := map[string]string{"server": mimp.Host}
	fields := map[string]interface{}{
		"banner_legit":   mimp.Banner.Legit,
		"banner_spam":    mimp.Banner.Spam,
		"banner_virus":   mimp.Banner.Virus,
		"auth_legit":     mimp.Auth.Legit,
		"auth_spam":      mimp.Auth.Spam,
		"auth_virus":     mimp.Auth.Virus,
		"content_legit":  mimp.Content.Legit,
		"content_spam":   mimp.Content.Spam,
		"content_virus":  mimp.Content.Virus,
		"data_legit":     mimp.Data.Legit,
		"data_spam":      mimp.Data.Spam,
		"data_virus":     mimp.Data.Virus,
		"mailfrom_legit": mimp.Mailfrom.Legit,
		"mailfrom_spam":  mimp.Mailfrom.Spam,
		"mailfrom_virus": mimp.Mailfrom.Virus,
		"rcptto_legit":   mimp.Rcptto.Legit,
		"rcptto_spam":    mimp.Rcptto.Spam,
		"rcptto_virus":   mimp.Rcptto.Virus,
		"rate_content":   mimp.Rate.Contentp,
		"rate_smtp":      mimp.Rate.Smtpp,
		"rate_total":     mimp.Rate.Totalp,
	}
	if opts.Out {
		pt, err := influxdb.NewPoint("cloudmark_out", tags, fields, mimp.Date)
		if err != nil {
			fmt.Printf("Error: %+v\n", err)
			return
		}

		bp.AddPoint(pt)
	} else {
		pt, err := influxdb.NewPoint("cloudmark", tags, fields, mimp.Date)
		if err != nil {
			fmt.Printf("Error: %+v\n", err)
			return
		}

		bp.AddPoint(pt)
	}

	// Write the batch
	c.Write(bp)
}