Persistance and recovery

src/client/store/db.sql

@@ -0,0 +1,16 @@
+CREATE TABLE IF NOT EXISTS qfixdpl_messages (
+    id          UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+    quote_req_id TEXT NOT NULL,
+    j_message   JSONB NOT NULL,
+    created_at  TIMESTAMPTZ NOT NULL DEFAULT NOW()
+);
+CREATE INDEX IF NOT EXISTS idx_messages_quote_req_id ON qfixdpl_messages(quote_req_id);
+CREATE INDEX IF NOT EXISTS idx_messages_created_at ON qfixdpl_messages(created_at);
+
+CREATE TABLE IF NOT EXISTS qfixdpl_logs (
+    id          UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+    quote_req_id TEXT NOT NULL UNIQUE,
+    raw_msg     TEXT NOT NULL,
+    created_at  TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+    updated_at  TIMESTAMPTZ NOT NULL DEFAULT NOW()
+);

src/client/store/manager.go

@@ -2,7 +2,9 @@
 package store
 
 import (
+	_ "embed"
 	"log/slog"
+	"strings"
 	"time"
 
 	"quantex.com.ar/multidb"
@@ -11,6 +13,9 @@ import (
 	"quantex.com/qfixdpl/src/common/tracerr"
 )
 
+//go:embed db.sql
+var schemaSQL string
+
 const dbPingSeconds = 30
 
 type Store struct {
@@ -45,9 +50,31 @@ func New(config Config) (*Store, error) {
 
 	go s.db.PeriodicDBPing(time.Second * dbPingSeconds)
 
+	if err := s.ensureTables(); err != nil {
+		return nil, tracerr.Errorf("error ensuring tables: %w", err)
+	}
+
 	return s, nil
 }
 
+func (p *Store) ensureTables() error {
+	statements := strings.Split(schemaSQL, ";")
+	for _, stmt := range statements {
+		stmt = strings.TrimSpace(stmt)
+		if stmt == "" {
+			continue
+		}
+
+		if _, err := p.db.Exec(stmt); err != nil {
+			return tracerr.Errorf("error executing schema statement: %w", err)
+		}
+	}
+
+	slog.Info("database tables ensured")
+
+	return nil
+}
+
 func (p *Store) CloseDB() {
 	p.db.Close()
 	slog.Info("closing database connection.")

src/client/store/persistence.go

@@ -0,0 +1,105 @@
+package store
+
+import (
+	"encoding/json"
+	"strings"
+	"time"
+
+	"quantex.com/qfixdpl/src/common/tracerr"
+	"quantex.com/qfixdpl/src/domain"
+)
+
+func (p *Store) SaveMessage(msg domain.TradeMessage) error {
+	jsonBytes, err := json.Marshal(msg.JMessage)
+	if err != nil {
+		return tracerr.Errorf("error marshaling j_message: %w", err)
+	}
+
+	_, err = p.db.Exec(
+		"INSERT INTO qfixdpl_messages (quote_req_id, j_message) VALUES ($1, $2)",
+		msg.QuoteReqID, string(jsonBytes),
+	)
+	if err != nil {
+		return tracerr.Errorf("error inserting message: %w", err)
+	}
+
+	return nil
+}
+
+func (p *Store) SaveLog(entry domain.LogEntry) error {
+	upsertStmt := `INSERT INTO qfixdpl_logs (quote_req_id, raw_msg)
+		 VALUES ($1, $2)
+		 ON CONFLICT (quote_req_id) DO UPDATE
+		 SET raw_msg = qfixdpl_logs.raw_msg || E'\n' || EXCLUDED.raw_msg,
+		     updated_at = NOW()`
+
+	_, err := p.db.Exec(upsertStmt, entry.QuoteReqID, entry.RawMsg)
+	if err != nil {
+		return tracerr.Errorf("error upserting log: %w", err)
+	}
+
+	return nil
+}
+
+func (p *Store) GetTodayMessages() ([]domain.TradeMessage, error) {
+	rows, err := p.db.Query(
+		"SELECT id, quote_req_id, j_message, created_at FROM qfixdpl_messages WHERE created_at >= current_date ORDER BY created_at ASC",
+	)
+	if err != nil {
+		return nil, tracerr.Errorf("error querying today messages: %w", err)
+	}
+	defer rows.Close()
+
+	var messages []domain.TradeMessage
+
+	for rows.Next() {
+		var (
+			id, quoteReqID string
+			jMessageRaw    []byte
+			createdAt      time.Time
+		)
+
+		if err := rows.Scan(&id, &quoteReqID, &jMessageRaw, &createdAt); err != nil {
+			return nil, tracerr.Errorf("error scanning message row: %w", err)
+		}
+
+		var jMessage domain.FixMessageJSON
+		if err := json.Unmarshal(jMessageRaw, &jMessage); err != nil {
+			return nil, tracerr.Errorf("error unmarshaling j_message: %w", err)
+		}
+
+		messages = append(messages, domain.TradeMessage{
+			ID:         id,
+			QuoteReqID: quoteReqID,
+			JMessage:   jMessage,
+			CreatedAt:  createdAt,
+		})
+	}
+
+	if err := rows.Err(); err != nil {
+		return nil, tracerr.Errorf("error iterating message rows: %w", err)
+	}
+
+	return messages, nil
+}
+
+func (p *Store) GetLogsByQuoteReqID(quoteReqID string) (domain.Logs, error) {
+	selectStmt := "SELECT raw_msg FROM qfixdpl_logs WHERE quote_req_id = '" + quoteReqID + "';"
+
+	response, err := p.db.Query(selectStmt)
+	if err != nil {
+		return domain.Logs{}, tracerr.Errorf("error querying logs: %w", err)
+	}
+	defer response.Close()
+
+	if !response.Next() {
+		return domain.Logs{}, nil
+	}
+
+	var rawMsg string
+	if err := response.Scan(&rawMsg); err != nil {
+		return domain.Logs{}, tracerr.Errorf("error scanning log row: %w", err)
+	}
+
+	return domain.Logs{Entries: strings.Split(rawMsg, "\n")}, nil
+}