Fix dependency errors
This commit is contained in:
@@ -1,6 +1,7 @@
|
|||||||
package csv
|
package csv
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"context"
|
||||||
"encoding/csv"
|
"encoding/csv"
|
||||||
"os"
|
"os"
|
||||||
|
|
||||||
@@ -9,7 +10,6 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type Helper struct {
|
type Helper struct {
|
||||||
fx.In
|
|
||||||
csvFile *os.File
|
csvFile *os.File
|
||||||
csvWriter *csv.Writer
|
csvWriter *csv.Writer
|
||||||
logger *zap.Logger
|
logger *zap.Logger
|
||||||
@@ -23,11 +23,19 @@ func NewHelper(lc fx.Lifecycle, logger *zap.Logger) *Helper {
|
|||||||
|
|
||||||
csvWriter := csv.NewWriter(csvFile)
|
csvWriter := csv.NewWriter(csvFile)
|
||||||
|
|
||||||
return &Helper{
|
helper := &Helper{
|
||||||
csvFile: csvFile,
|
csvFile: csvFile,
|
||||||
csvWriter: csvWriter,
|
csvWriter: csvWriter,
|
||||||
logger: logger,
|
logger: logger,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
lc.Append(fx.Hook{
|
||||||
|
OnStop: func(context.Context) error {
|
||||||
|
return helper.Close()
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
return helper
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *Helper) AddEntry(date string, topic string, message string) error {
|
func (h *Helper) AddEntry(date string, topic string, message string) error {
|
||||||
|
|||||||
@@ -35,6 +35,11 @@ func NewClient(logger *zap.Logger, config *config.Config, csvHelper *csv.Helper)
|
|||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
|
if config.Username != "" && config.Password != "" {
|
||||||
|
opts.SetUsername(config.Username)
|
||||||
|
opts.SetPassword(config.Password)
|
||||||
|
}
|
||||||
|
|
||||||
c := mqtt.NewClient(opts)
|
c := mqtt.NewClient(opts)
|
||||||
if token := c.Connect(); token.Wait() && token.Error() != nil {
|
if token := c.Connect(); token.Wait() && token.Error() != nil {
|
||||||
logger.Fatal("error connecting to MQTT broker", zap.Error(token.Error()))
|
logger.Fatal("error connecting to MQTT broker", zap.Error(token.Error()))
|
||||||
|
|||||||
Reference in New Issue
Block a user