package engine

import "time"

// EngineConfig 存储引擎配置接口
type EngineConfig interface {
	// 获取引擎类型
	Type() string
}

// BaseConfig 基础配置，所有引擎共享
type BaseConfig struct {
	// 数据保留时间，超过此时间的数据将被清理
	RetentionPeriod time.Duration

	// 是否启用同步写入，启用后每次写入都会立即持久化
	SyncWrites bool

	// 写入缓冲区大小，0表示不使用缓冲区
	WriteBufferSize int

	// 是否启用压缩
	EnableCompaction bool

	// 压缩间隔
	CompactionInterval time.Duration
}

// MemoryEngineConfig 内存引擎配置
type MemoryEngineConfig struct {
	BaseConfig

	// 每个数据点保留的历史值数量（环形队列大小）
	MaxHistoryValues int

	// 最大数据点数量，超过此数量将拒绝写入，0表示不限制
	MaxPoints int

	// 是否启用持久化，启用后将定期将数据写入磁盘
	EnablePersistence bool

	// 持久化文件路径，仅在EnablePersistence为true时有效
	PersistencePath string

	// 持久化间隔，仅在EnablePersistence为true时有效
	PersistenceInterval time.Duration
}

// 实现EngineConfig接口
func (c *MemoryEngineConfig) Type() string {
	return "memory"
}

// MemoryConfig 返回内存引擎配置
func (c *MemoryEngineConfig) MemoryConfig() *MemoryEngineConfig {
	return c
}

// BoltConfig 返回Bolt引擎配置
func (c *MemoryEngineConfig) BoltConfig() *BoltEngineConfig {
	return nil
}

// BoltEngineConfig Bolt引擎配置
type BoltEngineConfig struct {
	BaseConfig

	// 数据库文件路径
	Path string

	// 存储桶名称
	BucketName string

	// 是否启用WAL日志
	EnableWAL bool

	// WAL日志目录，仅在EnableWAL为true时有效
	WALDir string

	// 是否启用批量写入
	EnableBatch bool

	// 批量写入大小，仅在EnableBatch为true时有效
	BatchSize int

	// 批量写入超时，仅在EnableBatch为true时有效
	BatchTimeout time.Duration
}

// 实现EngineConfig接口
func (c *BoltEngineConfig) Type() string {
	return "bolt"
}

// MemoryConfig 返回内存引擎配置
func (c *BoltEngineConfig) MemoryConfig() *MemoryEngineConfig {
	return nil
}

// BoltConfig 返回Bolt引擎配置
func (c *BoltEngineConfig) BoltConfig() *BoltEngineConfig {
	return c
}

// NewMemoryEngineConfig 创建默认的内存引擎配置
func NewMemoryEngineConfig() *MemoryEngineConfig {
	return &MemoryEngineConfig{
		BaseConfig: BaseConfig{
			RetentionPeriod:    24 * time.Hour, // 默认保留24小时
			SyncWrites:         false,
			WriteBufferSize:    1000,
			EnableCompaction:   true,
			CompactionInterval: 1 * time.Hour,
		},
		MaxHistoryValues:    30, // 默认保留30个历史值
		MaxPoints:           0,  // 默认不限制
		EnablePersistence:   false,
		PersistencePath:     "",
		PersistenceInterval: 10 * time.Minute,
	}
}

// NewBoltEngineConfig 创建默认的Bolt引擎配置
func NewBoltEngineConfig(path string) *BoltEngineConfig {
	return &BoltEngineConfig{
		BaseConfig: BaseConfig{
			RetentionPeriod:    24 * time.Hour, // 默认保留24小时
			SyncWrites:         false,
			WriteBufferSize:    1000,
			EnableCompaction:   true,
			CompactionInterval: 1 * time.Hour,
		},
		Path:         path,
		BucketName:   "gotidb",
		EnableWAL:    true,
		WALDir:       path + "_wal",
		EnableBatch:  true,
		BatchSize:    1000,
		BatchTimeout: 1 * time.Second,
	}
}