summaryrefslogtreecommitdiff
path: root/golang/pkg/database
diff options
context:
space:
mode:
Diffstat (limited to 'golang/pkg/database')
-rw-r--r--golang/pkg/database/markets.go42
-rw-r--r--golang/pkg/database/sql/001_trading.sql2
2 files changed, 42 insertions, 2 deletions
diff --git a/golang/pkg/database/markets.go b/golang/pkg/database/markets.go
new file mode 100644
index 0000000..11498d2
--- /dev/null
+++ b/golang/pkg/database/markets.go
@@ -0,0 +1,42 @@
+package database
+
+import (
+ "database/sql"
+ "encoding/json"
+ "fmt"
+ "time"
+
+ "git.adyxax.org/adyxax/spacetraders/golang/pkg/model"
+)
+
+func (db *DB) LoadMarket(symbol string) (*model.Market, error) {
+ var buf []byte
+ if err := db.QueryRow(`SELECT data FROM markets WHERE data->>'symbol' = ?;`, symbol).Scan(&buf); err != nil {
+ return nil, fmt.Errorf("failed to query row: %w", err)
+ }
+ var market model.Market
+ if err := json.Unmarshal(buf, &market); err != nil {
+ return nil, fmt.Errorf("failed to unmarshal market: %w", err)
+ }
+ return &market, nil
+}
+
+func (db *DB) SaveMarket(market *model.Market) error {
+ data, err := json.Marshal(market)
+ if err != nil {
+ return fmt.Errorf("failed to marshal market: %w", err)
+ }
+ _, err = db.Exec(
+ `INSERT INTO markets(data, updated)
+ VALUES (json(:data), :updated)
+ ON CONFLICT DO UPDATE SET data = :data, updated = :updated
+ WHERE data->>'symbol' = :symbol;`,
+ sql.Named("data", data),
+ sql.Named("symbol", market.Symbol),
+ sql.Named("updated", time.Now()),
+ )
+ if err != nil {
+ return fmt.Errorf("failed to exec: %w", err)
+ }
+ return nil
+}
diff --git a/golang/pkg/database/sql/001_trading.sql b/golang/pkg/database/sql/001_trading.sql
index 89dd77c..e218962 100644
--- a/golang/pkg/database/sql/001_trading.sql
+++ b/golang/pkg/database/sql/001_trading.sql
@@ -4,11 +4,9 @@ CREATE TABLE agents (
);
CREATE TABLE markets (
id INTEGER PRIMARY KEY,
- systemSymbol TEXT NOT NULL,
data JSON NOT NULL,
updated DATE NOT NULL
);
-CREATE INDEX markets_systemSymbol on markets (systemSymbol);
CREATE UNIQUE INDEX markets_data_symbol on markets(json_extract(data, '$.symbol'));
CREATE TABLE shipyards (
id INTEGER PRIMARY KEY,