[golang] implement shipyards visits

3 files changed, 72 insertions, 1 deletions

diff --git a/golang/pkg/database/shipyards.go b/golang/pkg/database/shipyards.go
new file mode 100644
index 0000000..8dd801e
--- /dev/null
+++ b/golang/pkg/database/shipyards.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) LoadShipyard(symbol string) (*model.Shipyard, error) {
+	var buf []byte
+	if err := db.QueryRow(`SELECT data FROM shipyards WHERE data->>'symbol' = ?;`, symbol).Scan(&buf); err != nil {
+		return nil, fmt.Errorf("failed to query shipyard: %w", err)
+	}
+	var shipyard model.Shipyard
+	if err := json.Unmarshal(buf, &shipyard); err != nil {
+		return nil, fmt.Errorf("failed to unmarshal shipyard: %w", err)
+	}
+	return &shipyard, nil
+}
+
+func (db *DB) SaveShipyard(shipyard *model.Shipyard) error {
+	data, err := json.Marshal(shipyard)
+	if err != nil {
+		return fmt.Errorf("failed to marshal shipyard: %w", err)
+	}
+	_, err = db.Exec(
+		`INSERT INTO shipyards(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", shipyard.Symbol),
+		sql.Named("updated", time.Now()),
+	)
+	if err != nil {
+		return fmt.Errorf("failed to append shipyard: %w", err)
+	}
+	return nil
+}
diff --git a/golang/pkg/database/sql/001_trading.sql b/golang/pkg/database/sql/001_trading.sql
index 68f2568..89dd77c 100644
--- a/golang/pkg/database/sql/001_trading.sql
+++ b/golang/pkg/database/sql/001_trading.sql
@@ -10,7 +10,12 @@ CREATE TABLE markets (
 );
 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,
+       data JSON NOT NULL,
+       updated DATE DEFAULT NULL
+);
+CREATE UNIQUE INDEX shipyards_data_symbol on shipyards (json_extract(data, '$.symbol'));
 CREATE TABLE systems (
   id INTEGER PRIMARY KEY,
   data JSON NOT NULL
diff --git a/golang/pkg/database/systems.go b/golang/pkg/database/systems.go
index 858c3dd..0af6aa7 100644
--- a/golang/pkg/database/systems.go
+++ b/golang/pkg/database/systems.go
@@ -45,6 +45,30 @@ func (db *DB) LoadWaypoint(symbol string) (*model.Waypoint, error) {
 	return &waypoint, nil
 }
 
+func (db *DB) LoadWaypointsInSystem(system *model.System) ([]model.Waypoint, error) {
+	rows, err := db.Query(`SELECT data FROM waypoints WHERE data->>'systemSymbol' = ?;`, system.Symbol)
+	if err != nil {
+		return nil, fmt.Errorf("failed to query waypoints: %w", err)
+	}
+	defer rows.Close()
+	waypoints := make([]model.Waypoint, 0)
+	for rows.Next() {
+		var buf []byte
+		if err := rows.Scan(&buf); err != nil {
+			return nil, fmt.Errorf("failed to load waypoint from row: %w", err)
+		}
+		var waypoint model.Waypoint
+		if err := json.Unmarshal(buf, &waypoint); err != nil {
+			return nil, fmt.Errorf("failed to unmarshal waypoint: %w", err)
+		}
+		waypoints = append(waypoints, waypoint)
+	}
+	if err := rows.Err(); err != nil {
+		return nil, fmt.Errorf("failed to load waypoints from rows: %w", err)
+	}
+	return waypoints, nil
+}
+
 func (db *DB) SaveWaypoint(waypoint *model.Waypoint) error {
 	data, err := json.Marshal(waypoint)
 	if err != nil {