ucpr/mongo-streamer

package config
 
import (
    "context"
 
    "github.com/google/wire"
    envconfig "github.com/sethvargo/go-envconfig"
)
 
// Set is a Wire provider set that provides configuration.
var Set = wire.NewSet(
    NewMongoDB,
    NewPubSub,
    NewMetrics,
)
 
const (
    mongoDBPrefix = "MONGO_DB_"
    pubSubPrefix  = "PUBSUB_"
    mrtricsPrefix = "METRICS_"
)
 
// PublishFormat is the format of the message to publish.
const (
    // PublishFormatJSON is the JSON format.
    PubSubPublishFormatJSON = "json"
    // PublishFormatAvro is the Avro format.
    PubSubPublishFormatAvro = "avro"
)
 
type MongoDB struct {
    URI        string `env:"URI, required"`
    Password   string `env:"PASSWORD"`
    User       string `env:"USER"`
    Database   string `env:"DATABASE, required"`
    Collection string `env:"COLLECTION, required"`
}
 
type PubSub struct {
    // ProjectID is the id of the Google Cloud project to publish messages to.
    ProjectID string `env:"PROJECT_ID"`
    // TopicID is the id of the topic to publish messages to.
    TopicID string `env:"TOPIC_ID"`
    // PublishFormat is the format of the message to publish.
    // Supported format are: json, avro.
    PublishFormat string `env:"PUBLISH_FORMAT, default=json"`
}
 
type Metrics struct {
    Addr string `env:"ADDR, default=:8080"`
}
 
Similar blocks of code found in 3 locations. Consider refactoring.
func NewMongoDB(ctx context.Context) (*MongoDB, error) {
    conf := &MongoDB{}
    pl := envconfig.PrefixLookuper(mongoDBPrefix, envconfig.OsLookuper())
    if err := envconfig.ProcessWith(ctx, &envconfig.Config{
        Target:   conf,
        Lookuper: pl,
    }); err != nil {
        return nil, err
    }
 
    return conf, nil
}
 
Similar blocks of code found in 3 locations. Consider refactoring.
func NewPubSub(ctx context.Context) (*PubSub, error) {
    conf := &PubSub{}
    pl := envconfig.PrefixLookuper(pubSubPrefix, envconfig.OsLookuper())
    if err := envconfig.ProcessWith(ctx, &envconfig.Config{
        Target:   conf,
        Lookuper: pl,
    }); err != nil {
        return nil, err
    }
 
    return conf, nil
}
 
Similar blocks of code found in 3 locations. Consider refactoring.
func NewMetrics(ctx context.Context) (*Metrics, error) {
    conf := &Metrics{}
    pl := envconfig.PrefixLookuper(mrtricsPrefix, envconfig.OsLookuper())
    if err := envconfig.ProcessWith(ctx, &envconfig.Config{
        Target:   conf,
        Lookuper: pl,
    }); err != nil {
        return nil, err
    }
 
    return conf, nil
}