initial commit
This commit is contained in:
74
internal/infrastructure/session/file_storage.go
Normal file
74
internal/infrastructure/session/file_storage.go
Normal file
@@ -0,0 +1,74 @@
|
||||
package session
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
|
||||
"github.com/paramah/ai_devs4/s01e03/internal/domain"
|
||||
)
|
||||
|
||||
// FileStorage implements domain.SessionStorage using JSON files
|
||||
type FileStorage struct {
|
||||
baseDir string
|
||||
}
|
||||
|
||||
// NewFileStorage creates a new file-based session storage
|
||||
func NewFileStorage(baseDir string) *FileStorage {
|
||||
return &FileStorage{
|
||||
baseDir: baseDir,
|
||||
}
|
||||
}
|
||||
|
||||
// Get retrieves a session from file storage
|
||||
func (s *FileStorage) Get(ctx context.Context, sessionID string) (*domain.Session, error) {
|
||||
filePath := s.getFilePath(sessionID)
|
||||
|
||||
// Check if file exists
|
||||
if _, err := os.Stat(filePath); os.IsNotExist(err) {
|
||||
// Return empty session if file doesn't exist
|
||||
return &domain.Session{
|
||||
ID: sessionID,
|
||||
Messages: []domain.Message{},
|
||||
}, nil
|
||||
}
|
||||
|
||||
data, err := os.ReadFile(filePath)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("reading session file: %w", err)
|
||||
}
|
||||
|
||||
var session domain.Session
|
||||
if err := json.Unmarshal(data, &session); err != nil {
|
||||
return nil, fmt.Errorf("unmarshaling session: %w", err)
|
||||
}
|
||||
|
||||
return &session, nil
|
||||
}
|
||||
|
||||
// Save saves a session to file storage
|
||||
func (s *FileStorage) Save(ctx context.Context, session *domain.Session) error {
|
||||
filePath := s.getFilePath(session.ID)
|
||||
|
||||
// Ensure directory exists
|
||||
if err := os.MkdirAll(s.baseDir, 0755); err != nil {
|
||||
return fmt.Errorf("creating cache directory: %w", err)
|
||||
}
|
||||
|
||||
data, err := json.MarshalIndent(session, "", " ")
|
||||
if err != nil {
|
||||
return fmt.Errorf("marshaling session: %w", err)
|
||||
}
|
||||
|
||||
if err := os.WriteFile(filePath, data, 0644); err != nil {
|
||||
return fmt.Errorf("writing session file: %w", err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *FileStorage) getFilePath(sessionID string) string {
|
||||
return filepath.Join(s.baseDir, fmt.Sprintf("%s.json", sessionID))
|
||||
}
|
||||
Reference in New Issue
Block a user