// Copyright (c) Mainflux // SPDX-License-Identifier: Apache-2.0 // Package main contains mongodb-reader main function to start the mongodb-reader service. package main import ( "context" "fmt" "log" "os" chclient "github.com/mainflux/callhome/pkg/client" "github.com/mainflux/mainflux" "github.com/mainflux/mainflux/internal" authclient "github.com/mainflux/mainflux/internal/clients/grpc/auth" mongoclient "github.com/mainflux/mainflux/internal/clients/mongo" "github.com/mainflux/mainflux/internal/env" "github.com/mainflux/mainflux/internal/server" httpserver "github.com/mainflux/mainflux/internal/server/http" mflog "github.com/mainflux/mainflux/logger" "github.com/mainflux/mainflux/pkg/uuid" "github.com/mainflux/mainflux/readers" "github.com/mainflux/mainflux/readers/api" "github.com/mainflux/mainflux/readers/mongodb" "go.mongodb.org/mongo-driver/mongo" "golang.org/x/sync/errgroup" ) const ( svcName = "mongodb-reader" envPrefixDB = "MF_MONGO_" envPrefixHTTP = "MF_MONGO_READER_HTTP_" defSvcHTTPPort = "9007" ) type config struct { LogLevel string `env:"MF_MONGO_READER_LOG_LEVEL" envDefault:"info"` SendTelemetry bool `env:"MF_SEND_TELEMETRY" envDefault:"true"` InstanceID string `env:"MF_MONGO_READER_INSTANCE_ID" envDefault:""` } func main() { ctx, cancel := context.WithCancel(context.Background()) g, ctx := errgroup.WithContext(ctx) cfg := config{} if err := env.Parse(&cfg); err != nil { log.Fatalf("failed to load %s configuration : %s", svcName, err) } logger, err := mflog.New(os.Stdout, cfg.LogLevel) if err != nil { log.Fatalf("failed to init logger: %s", err) } var exitCode int defer mflog.ExitWithError(&exitCode) if cfg.InstanceID == "" { if cfg.InstanceID, err = uuid.New().ID(); err != nil { logger.Error(fmt.Sprintf("failed to generate instanceID: %s", err)) exitCode = 1 return } } db, err := mongoclient.Setup(envPrefixDB) if err != nil { logger.Error(fmt.Sprintf("failed to setup mongo database : %s", err)) exitCode = 1 return } repo := newService(db, logger) ac, acHandler, err := authclient.Setup(svcName) if err != nil { logger.Fatal(err.Error()) exitCode = 1 return } defer acHandler.Close() logger.Info("Successfully connected to auth grpc server " + acHandler.Secure()) tc, tcHandler, err := authclient.SetupAuthz(svcName) if err != nil { logger.Error(err.Error()) exitCode = 1 return } defer tcHandler.Close() logger.Info("Successfully connected to things grpc server " + tcHandler.Secure()) httpServerConfig := server.Config{Port: defSvcHTTPPort} if err := env.Parse(&httpServerConfig, env.Options{Prefix: envPrefixHTTP}); err != nil { logger.Error(fmt.Sprintf("failed to load %s HTTP server configuration : %s", svcName, err)) exitCode = 1 return } hs := httpserver.New(ctx, cancel, svcName, httpServerConfig, api.MakeHandler(repo, ac, tc, svcName, cfg.InstanceID), logger) if cfg.SendTelemetry { chc := chclient.New(svcName, mainflux.Version, logger, cancel) go chc.CallHome(ctx) } g.Go(func() error { return hs.Start() }) g.Go(func() error { return server.StopSignalHandler(ctx, cancel, logger, svcName, hs) }) if err := g.Wait(); err != nil { logger.Error(fmt.Sprintf("MongoDB reader service terminated: %s", err)) } } func newService(db *mongo.Database, logger mflog.Logger) readers.MessageRepository { repo := mongodb.New(db) repo = api.LoggingMiddleware(repo, logger) counter, latency := internal.MakeMetrics("mongodb", "message_reader") repo = api.MetricsMiddleware(repo, counter, latency) return repo }