init

internal/state/store.go

@@ -0,0 +1,109 @@
+package state
+
+import (
+	"bufio"
+	"encoding/json"
+	"errors"
+	"os"
+	"path/filepath"
+	"sort"
+	"sync"
+
+	"github.com/example/glpi-ai-agent/internal/model"
+)
+
+type Store struct {
+	mu        sync.RWMutex
+	path      string
+	processed map[string]struct{}
+	runs      []model.RunRecord
+	maxRuns   int
+}
+
+func Open(dir string, maxRuns int) (*Store, error) {
+	if err := os.MkdirAll(dir, 0o750); err != nil {
+		return nil, err
+	}
+	s := &Store{path: filepath.Join(dir, "runs.jsonl"), processed: map[string]struct{}{}, maxRuns: maxRuns}
+	if err := s.load(); err != nil {
+		return nil, err
+	}
+	return s, nil
+}
+func key(id int64, version string) string { return fmtKey(id, version) }
+func fmtKey(id int64, version string) string {
+	b, _ := json.Marshal([]any{id, version})
+	return string(b)
+}
+func (s *Store) Seen(id int64, version string) bool {
+	s.mu.RLock()
+	defer s.mu.RUnlock()
+	_, ok := s.processed[key(id, version)]
+	return ok
+}
+func (s *Store) Append(r model.RunRecord) error {
+	s.mu.Lock()
+	defer s.mu.Unlock()
+	f, err := os.OpenFile(s.path, os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0o640)
+	if err != nil {
+		return err
+	}
+	enc := json.NewEncoder(f)
+	if err = enc.Encode(r); err == nil {
+		err = f.Sync()
+	}
+	cerr := f.Close()
+	if err == nil {
+		err = cerr
+	}
+	if err != nil {
+		return err
+	}
+	if r.Outcome != "error" && r.Error == "" {
+		s.processed[key(r.TicketID, r.SourceVersion)] = struct{}{}
+	}
+	s.runs = append(s.runs, r)
+	if len(s.runs) > s.maxRuns {
+		s.runs = s.runs[len(s.runs)-s.maxRuns:]
+	}
+	return nil
+}
+func (s *Store) Recent(limit int) []model.RunRecord {
+	s.mu.RLock()
+	defer s.mu.RUnlock()
+	if limit <= 0 || limit > len(s.runs) {
+		limit = len(s.runs)
+	}
+	out := make([]model.RunRecord, limit)
+	for i := 0; i < limit; i++ {
+		out[i] = s.runs[len(s.runs)-1-i]
+	}
+	return out
+}
+func (s *Store) load() error {
+	f, err := os.Open(s.path)
+	if errors.Is(err, os.ErrNotExist) {
+		return nil
+	}
+	if err != nil {
+		return err
+	}
+	defer f.Close()
+	sc := bufio.NewScanner(f)
+	buf := make([]byte, 64*1024)
+	sc.Buffer(buf, 2*1024*1024)
+	for sc.Scan() {
+		var r model.RunRecord
+		if json.Unmarshal(sc.Bytes(), &r) == nil {
+			s.runs = append(s.runs, r)
+			if r.Outcome != "error" && r.Error == "" {
+				s.processed[key(r.TicketID, r.SourceVersion)] = struct{}{}
+			}
+		}
+	}
+	if len(s.runs) > s.maxRuns {
+		s.runs = s.runs[len(s.runs)-s.maxRuns:]
+	}
+	sort.SliceStable(s.runs, func(i, j int) bool { return s.runs[i].FinishedAt.Before(s.runs[j].FinishedAt) })
+	return sc.Err()
+}

internal/state/store_test.go

@@ -0,0 +1,51 @@
+package state
+
+import (
+	"github.com/example/glpi-ai-agent/internal/model"
+	"testing"
+	"time"
+)
+
+func TestStoreRoundTrip(t *testing.T) {
+	dir := t.TempDir()
+	s, err := Open(dir, 10)
+	if err != nil {
+		t.Fatal(err)
+	}
+	r := model.RunRecord{TicketID: 7, SourceVersion: "v1", FinishedAt: time.Now(), Outcome: "processed"}
+	if err := s.Append(r); err != nil {
+		t.Fatal(err)
+	}
+	if !s.Seen(7, "v1") {
+		t.Fatal("not seen")
+	}
+	s2, err := Open(dir, 10)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !s2.Seen(7, "v1") {
+		t.Fatal("not persisted")
+	}
+}
+
+func TestErrorRunIsRetriedAfterRestart(t *testing.T) {
+	dir := t.TempDir()
+	s, err := Open(dir, 10)
+	if err != nil {
+		t.Fatal(err)
+	}
+	r := model.RunRecord{TicketID: 8, SourceVersion: "v1", FinishedAt: time.Now(), Outcome: "error", Error: "temporary outage"}
+	if err := s.Append(r); err != nil {
+		t.Fatal(err)
+	}
+	if s.Seen(8, "v1") {
+		t.Fatal("failed run must remain retryable")
+	}
+	s2, err := Open(dir, 10)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if s2.Seen(8, "v1") {
+		t.Fatal("failed run became seen after restart")
+	}
+}