Skip to content
GitLab
Select Git revision
  • master default
Find file BlameHistoryPermalink
config.go 4.74 KiB
1 
package config
2
3 
import (
4 
	"context"
5 
	"fmt"
6 
	"time"
7
8 
	"github.com/go-playground/validator/v10"
9 
	"github.com/sethvargo/go-envconfig"
10 
)
11
12 
// Config объединяет все необходимые секции конфигурации для вашего проекта.
13 
type Config struct {
14 
	Server   ServerConfig   `env:",prefix=SERVER_"`
15 
	DB       DBConfig       `env:",prefix=DATABASE_"`
16 
	LLM      LLMConfig      `env:",prefix=LLM_"`
17 
	Kafka    KafkaConfig    `env:",prefix=KAFKA_"`
18 
	OTEL     OTelConfig     `env:",prefix=OTEL_"`
19 
	Telegram TelegramConfig `env:",prefix=TELEGRAM_"`
20 
}
21
22 
// ServerConfig содержит параметры HTTP-сервера.
23 
type ServerConfig struct {
24 
	AppName string `env:"APP_NAME,default=ProductAssistant" validate:"required"`
25 
	Port    string `env:"PORT,default=8080" validate:"required"`
26 
	Secret  string `env:"SECRET,default=secret" validate:"required"`
27 
}
28
29 
type DBConfig struct {
30 
	Hostname string `env:"HOSTNAME, default=localhost"`
31 
	Port     int    `env:"PORT, default=5435"`
32 
	User     string `env:"USER, default=postgres"`
33 
	Password string `env:"PASSWORD, default=secret"`
34 
	Dbname   string `env:"DBNAME, default=products_db"`
35 
}
36
37 
func (c *DBConfig) ConnectionString() string {
38 
	return fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%d sslmode=disable", c.Hostname, c.User, c.Password, c.Dbname, c.Port)
39 
}
40
41 
// LLMConfig содержит настройки для работы с LLM (модель, API-ключ, endpoint).
42 
type LLMConfig struct {
43 
	Embedder  string `env:"EMBEDDER,default=bge-m3" validate:"required"`
44 
	ModelName string `env:"MODEL_NAME,default=mistral-large-latest" validate:"required"`
45 
	APIKey    string `env:"API_KEY,default=ZdDk5S3QxaEytpf9DWXM2nEb6GVHuF9y" validate:"required"`
46 
	Endpoint  string `env:"ENDPOINT, default=http://localhost:11434" validate:"required,url"`
47 
}
48
49 
// KafkaConfig содержит общие настройки для Kafka.
50 
type KafkaConfig struct {
51 
	BootstrapServers []string       `env:"BOOTSTRAP_SERVERS, default=localhost:9092" validate:"required,dive,hostname_port"`
52 
	SSLCertPath      string         `env:"SSL_CERT_PATH"`
53 
	SSLKeyPath       string         `env:"SSL_KEY_PATH"`
54 
	SSLCaCert        string         `env:"SSL_CA_CERT"`
55 
	VerifyCert       bool           `env:"VERIFY_CERT,default=false"`
56 
	Producer         ProducerConfig `env:",prefix=PRODUCER_"`
57 
	Consumer         ConsumerConfig `env:",prefix=CONSUMER_"`
58 
}
59
60 
// ProducerConfig содержит настройки для продюсера Kafka.
61 
type ProducerConfig struct {
62 
	Topic                  string `env:"TOPIC, default=price_changes" validate:"required"`
63 
	Async                  bool   `env:"ASYNC, default=false"`
64 
	AllowAutoTopicCreation bool   `env:"ALLOW_AUTO_TOPIC_CREATION, default=true"`
65 
}
66
67 
// ConsumerConfig содержит настройки для консьюмера Kafka.
68 
type ConsumerConfig struct {
69 
	Topic   string `env:"TOPIC, default=price_changes" validate:"required"`
70 
	GroupID string `env:"GROUP_ID, default=notifications_group" validate:"required"`
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117
} // OTelConfig содержит настройки для OpenTelemetry. type OTelConfig struct { OTLP OTLPConfig `env:",prefix=OTLP_"` } // OTLPConfig задаёт параметры приёма трейсов. type OTLPConfig struct { Insecure bool `env:"INSECURE,default=true"` Endpoint string `env:"ENDPOINT,default=localhost:4317" validate:"required,url"` Timeout time.Duration `env:"TIMEOUT,default=10s"` HTTP bool `env:"HTTP,default=false"` Retry RetryConfig `env:",prefix=RETRY_"` } // RetryConfig содержит настройки повторных попыток для OTLP. type RetryConfig struct { Enabled bool `env:"ENABLED,default=true"` InitialInterval time.Duration `env:"INITIAL_INTERVAL,default=5s"` MaxInterval time.Duration `env:"MAX_INTERVAL,default=30s"` MaxElapsedTime time.Duration `env:"MAX_ELAPSED_TIME,default=1m"` } type TelegramConfig struct { BotToken string `env:"BOT_TOKEN,default=7665408380:AAFKbUgbdQb3iX6X-29H3spv4KGG1FXs_IM" validate:"required"` JWTSecret string `env:"JWT_SECRET,default=secret" validate:"required"` } // NewConfig загружает конфигурацию из переменных окружения и проводит валидацию. func NewConfig(ctx context.Context) (*Config, error) { var cfg Config if err := envconfig.Process(ctx, &cfg); err != nil { return nil, fmt.Errorf("failed to process env config: %w", err) } if err := cfg.Validate(); err != nil { return nil, fmt.Errorf("config validation error: %w", err) } return &cfg, nil } // Validate проверяет корректность настроек с использованием пакета validator. func (c *Config) Validate() error { validate := validator.New() return validate.Struct(c) }