futriis/pkg/config/config.go

// /futriis/pkg/config/config.go
// Пакет config предоставляет функциональность для работы с конфигурацией

package config

import (
	"sync/atomic"

	"github.com/BurntSushi/toml"
	"futriis/pkg/utils"
)

// ClusterConfig конфигурация кластера
type ClusterConfig struct {
	Name               string `toml:"name"`
	CoordinatorAddress string `toml:"coordinator_address"`
	ReplicationFactor  int    `toml:"replication_factor"`
	SyncReplication    bool   `toml:"sync_replication"`
	AutoRebalance      bool   `toml:"auto_rebalance"`
}

// NodeConfig конфигурация узла
type NodeConfig struct {
	ID        string `toml:"id"`
	Address   string `toml:"address"`
	DataDir   string `toml:"data_dir"`
	AOFEnabled bool   `toml:"aof_enabled"`
	AOFFile   string `toml:"aof_file"`
}

// StorageConfig конфигурация хранилища
type StorageConfig struct {
	PageSize       int    `toml:"page_size"`
	MaxMemory      string `toml:"max_memory"`
	EvictionPolicy string `toml:"eviction_policy"`
}

// ReplicationConfig конфигурация репликации
type ReplicationConfig struct {
	Enabled          bool `toml:"enabled"`
	SyncMode         string `toml:"sync_mode"`
	HeartbeatInterval int   `toml:"heartbeat_interval"`
	Timeout          int   `toml:"timeout"`
}

// LuaConfig конфигурация Lua плагинов
type LuaConfig struct {
	Enabled    bool   `toml:"enabled"`
	PluginsDir string `toml:"plugins_dir"`
	MaxMemory  string `toml:"max_memory"`
}

// Config основная структура конфигурации
type Config struct {
	Cluster     ClusterConfig     `toml:"cluster"`
	Node        NodeConfig        `toml:"node"`
	Storage     StorageConfig     `toml:"storage"`
	Replication ReplicationConfig `toml:"replication"`
	Lua         LuaConfig         `toml:"lua"`
}

var globalConfig atomic.Value

// Load загружает конфигурацию из файла
func Load(path string) (*Config, error) {
	var config Config
	
	if _, err := toml.DecodeFile(path, &config); err != nil {
		return nil, err
	}
	
	// Устанавливаем значения по умолчанию, если не указаны
	if config.Cluster.CoordinatorAddress == "" {
		config.Cluster.CoordinatorAddress = "127.0.0.1:7379"
	}
	
	if config.Node.Address == "" {
		config.Node.Address = "127.0.0.1:7380"
	}
	
	if config.Node.DataDir == "" {
		config.Node.DataDir = "./data"
	}
	
	if config.Node.AOFFile == "" {
		config.Node.AOFFile = "./data/futriis.aof"
	}
	
	if config.Storage.PageSize == 0 {
		config.Storage.PageSize = 4096
	}
	
	if config.Replication.HeartbeatInterval == 0 {
		config.Replication.HeartbeatInterval = 5
	}
	
	if config.Replication.Timeout == 0 {
		config.Replication.Timeout = 30
	}
	
	if config.Lua.PluginsDir == "" {
		config.Lua.PluginsDir = "./plugins"
	}
	
	globalConfig.Store(&config)
	
	utils.PrintSuccess("Конфигурация загружена из " + path)
	return &config, nil
}

// Get возвращает глобальную конфигурацию
func Get() *Config {
	if cfg := globalConfig.Load(); cfg != nil {
		return cfg.(*Config)
	}
	return nil
}

// GetClusterConfig возвращает конфигурацию кластера
func GetClusterConfig() *ClusterConfig {
	if cfg := Get(); cfg != nil {
		return &cfg.Cluster
	}
	return nil
}

// GetNodeConfig возвращает конфигурацию узла
func GetNodeConfig() *NodeConfig {
	if cfg := Get(); cfg != nil {
		return &cfg.Node
	}
	return nil
}