package checkpoint

import "sync"

// MemoryStore 是进程内检查点存储，线程安全。用于单元测试与零依赖本地运行。
type MemoryStore struct {
	mu         sync.RWMutex
	logOffsets map[string]int64
	ackedSeq   map[string]int64
}

// NewMemoryStore 创建空的进程内检查点存储。
func NewMemoryStore() *MemoryStore {
	return &MemoryStore{
		logOffsets: map[string]int64{},
		ackedSeq:   map[string]int64{},
	}
}

func (s *MemoryStore) GetLogOffset(file string) (int64, error) {
	s.mu.RLock()
	defer s.mu.RUnlock()
	return s.logOffsets[file], nil
}

func (s *MemoryStore) SetLogOffset(file string, offset int64) error {
	s.mu.Lock()
	defer s.mu.Unlock()
	s.logOffsets[file] = offset
	return nil
}

func (s *MemoryStore) GetAckedSeq(hostID string) (int64, error) {
	s.mu.RLock()
	defer s.mu.RUnlock()
	return s.ackedSeq[hostID], nil
}

func (s *MemoryStore) SetAckedSeq(hostID string, seq int64) error {
	s.mu.Lock()
	defer s.mu.Unlock()
	s.ackedSeq[hostID] = seq
	return nil
}

func (s *MemoryStore) Close() error { return nil }