-
Notifications
You must be signed in to change notification settings - Fork 7
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Abstract DB so we can switch between MongoDB and SQLite.
SQLite is only used for local development.
- Loading branch information
1 parent
7ffc6c9
commit d54f0f0
Showing
9 changed files
with
263 additions
and
103 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,194 @@ | ||
| package db | ||
|
|
||
| import ( | ||
| "context" | ||
| _ "embed" | ||
| "fmt" | ||
| "time" | ||
|
|
||
| "github.com/jmoiron/sqlx" | ||
| "github.com/temporalio/reference-app-orders-go/app/config" | ||
| "go.mongodb.org/mongo-driver/bson" | ||
| "go.mongodb.org/mongo-driver/mongo" | ||
| mongodb "go.mongodb.org/mongo-driver/mongo" | ||
| "go.mongodb.org/mongo-driver/mongo/options" | ||
| ) | ||
|
|
||
| // OrdersCollection is the name of the MongoDB collection to use for Orders. | ||
| const OrdersCollection = "orders" | ||
|
|
||
| // ShipmentCollection is the name of the MongoDB collection to use for Shipment data. | ||
| const ShipmentCollection = "shipments" | ||
|
|
||
| // DB is an interface that defines the methods that a database driver must implement | ||
| type DB interface { | ||
| Connect(ctx context.Context) error | ||
| Setup() error | ||
| Close() error | ||
| InsertOrder(context.Context, interface{}) error | ||
| UpdateOrderStatus(context.Context, string, string) error | ||
| GetOrders(context.Context, interface{}) error | ||
| UpdateShipmentStatus(context.Context, string, string) error | ||
| GetShipments(context.Context, interface{}) error | ||
| } | ||
|
|
||
| // CreateDB creates a new DB instance based on the configuration | ||
| func CreateDB(config config.AppConfig) DB { | ||
| if config.MongoURL != "" { | ||
| return &MongoDB{uri: config.MongoURL} | ||
| } | ||
|
|
||
| return &SQLiteDB{path: "./api-data.db"} | ||
| } | ||
|
|
||
| // MongoDB is a struct that implements the DB interface for MongoDB | ||
| type MongoDB struct { | ||
| uri string | ||
| client *mongo.Client | ||
| db *mongo.Database | ||
| } | ||
|
|
||
| // Connect connects to a MongoDB instance | ||
| func (m *MongoDB) Connect(ctx context.Context) error { | ||
| client, err := mongo.Connect(ctx, options.Client().ApplyURI(m.uri)) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| m.client = client | ||
| m.db = client.Database("orders") | ||
| return nil | ||
| } | ||
|
|
||
| // Setup sets up the MongoDB instance | ||
| func (m *MongoDB) Setup() error { | ||
| orders := m.db.Collection(OrdersCollection) | ||
| _, err := orders.Indexes().CreateOne(context.TODO(), mongodb.IndexModel{ | ||
| Keys: map[string]interface{}{"received_at": 1}, | ||
| }) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to create orders index: %w", err) | ||
| } | ||
|
|
||
| shipments := m.db.Collection(ShipmentCollection) | ||
| _, err = shipments.Indexes().CreateOne(context.TODO(), mongodb.IndexModel{ | ||
| Keys: map[string]interface{}{"booked_at": 1}, | ||
| }) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to create shipment index: %w", err) | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| // InsertOrder inserts an Order into the MongoDB instance | ||
| func (m *MongoDB) InsertOrder(ctx context.Context, order interface{}) error { | ||
| _, err := m.db.Collection(OrdersCollection).InsertOne(ctx, order) | ||
| return err | ||
| } | ||
|
|
||
| // UpdateOrderStatus updates an Order in the MongoDB instance | ||
| func (m *MongoDB) UpdateOrderStatus(ctx context.Context, id string, status string) error { | ||
| _, err := m.db.Collection(OrdersCollection).UpdateOne(ctx, bson.M{"id": id}, bson.M{"$set": bson.M{"status": status}}) | ||
| return err | ||
| } | ||
|
|
||
| // GetOrders returns a list of Orders from the MongoDB instance | ||
| func (m *MongoDB) GetOrders(ctx context.Context, result interface{}) error { | ||
| res, err := m.db.Collection(OrdersCollection).Find(ctx, bson.M{}, &options.FindOptions{ | ||
| Sort: bson.M{"received_at": 1}, | ||
| }) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| return res.All(ctx, result) | ||
| } | ||
|
|
||
| // UpdateShipmentStatus updates a Shipment in the MongoDB instance | ||
| func (m *MongoDB) UpdateShipmentStatus(ctx context.Context, id string, status string) error { | ||
| _, err := m.db.Collection(ShipmentCollection).UpdateOne( | ||
| ctx, | ||
| bson.M{"id": id}, | ||
| bson.M{ | ||
| "$set": bson.M{"status": status}, | ||
| "$setOnInsert": bson.M{"booked_at": time.Now().UTC()}, | ||
| }, | ||
| ) | ||
| return err | ||
| } | ||
|
|
||
| // GetShipments returns a list of Shipments from the MongoDB instance | ||
| func (m *MongoDB) GetShipments(ctx context.Context, result interface{}) error { | ||
| res, err := m.db.Collection(ShipmentCollection).Find(ctx, bson.M{}, &options.FindOptions{ | ||
| Sort: bson.M{"booked_at": 1}, | ||
| }) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| return res.All(ctx, result) | ||
| } | ||
|
|
||
| // Close closes the connection to the MongoDB instance | ||
| func (m *MongoDB) Close() error { | ||
| return m.client.Disconnect(context.Background()) | ||
| } | ||
|
|
||
| // SQLiteDB is a struct that implements the DB interface for SQLite | ||
| type SQLiteDB struct { | ||
| path string | ||
| db *sqlx.DB | ||
| } | ||
|
|
||
| //go:embed schema.sql | ||
| var sqliteSchema string | ||
|
|
||
| // Connect connects to a SQLite instance | ||
| func (s *SQLiteDB) Connect(_ context.Context) error { | ||
| db, err := sqlx.Connect("sqlite3", s.path) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| s.db = db | ||
| db.SetMaxOpenConns(1) // SQLite does not support concurrent writes | ||
| return nil | ||
| } | ||
|
|
||
| // Setup sets up the SQLite instance | ||
| func (s *SQLiteDB) Setup() error { | ||
| _, err := s.db.Exec(sqliteSchema) | ||
| return err | ||
| } | ||
|
|
||
| // Close closes the connection to the SQLite instance | ||
| func (s *SQLiteDB) Close() error { | ||
| return s.db.Close() | ||
| } | ||
|
|
||
| // InsertOrder inserts an Order into the SQLite instance | ||
| func (s *SQLiteDB) InsertOrder(ctx context.Context, order interface{}) error { | ||
| _, err := s.db.NamedExecContext(ctx, "INSERT INTO orders (id, received_at, status) VALUES (:id, :received_at, :status)", order) | ||
| return err | ||
| } | ||
|
|
||
| // UpdateOrderStatus updates an Order in the SQLite instance | ||
| func (s *SQLiteDB) UpdateOrderStatus(ctx context.Context, id string, status string) error { | ||
| _, err := s.db.ExecContext(ctx, "UPDATE orders SET status = ? WHERE id = ?", status, id) | ||
| return err | ||
| } | ||
|
|
||
| // GetOrders returns a list of Orders from the SQLite instance | ||
| func (s *SQLiteDB) GetOrders(ctx context.Context, result interface{}) error { | ||
| return s.db.SelectContext(ctx, result, "SELECT * FROM orders ORDER BY received_at") | ||
| } | ||
|
|
||
| // UpdateShipmentStatus updates a Shipment in the SQLite instance | ||
| func (s *SQLiteDB) UpdateShipmentStatus(ctx context.Context, id string, status string) error { | ||
| _, err := s.db.ExecContext(ctx, "INSERT INTO shipments (id, booked_at, status) VALUES (?, ?, ?) ON CONFLICT(id) DO UPDATE SET status = ?", id, time.Now().UTC(), status, status) | ||
| return err | ||
| } | ||
|
|
||
| // GetShipments returns a list of Shipments from the SQLite instance | ||
| func (s *SQLiteDB) GetShipments(ctx context.Context, result interface{}) error { | ||
| return s.db.SelectContext(ctx, result, "SELECT * FROM shipments ORDER BY booked_at") | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,16 @@ | ||
| CREATE TABLE IF NOT EXISTS orders ( | ||
| id TEXT PRIMARY KEY, | ||
| customer_id TEXT NOT NULL, | ||
| received_at TIMESTAMP NOT NULL, | ||
| status TEXT NOT NULL | ||
| ); | ||
|
|
||
| CREATE INDEX IF NOT EXISTS orders_received_at ON orders(received_at DESC); | ||
|
|
||
| CREATE TABLE IF NOT EXISTS shipments ( | ||
| id TEXT PRIMARY KEY, | ||
| status TEXT NOT NULL, | ||
| booked_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP | ||
| ); | ||
|
|
||
| CREATE INDEX IF NOT EXISTS shipments_booked_at ON shipments (booked_at DESC); |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.