setup basic spanner as persistence layer

exchange.go

@@ -26,15 +26,15 @@ func NewExchange() *Exchange {
 
 func (ex *Exchange) Run(ctx context.Context) {
 	go ex.PollMixinMessages(ctx)
-	ex.PollMixinNetwork(ctx)
+	go ex.PollMixinNetwork(ctx)
+	ex.PollOrderActions(ctx)
 }
 
-func (ex *Exchange) PollMixinMessages(ctx context.Context) {
-	bot.Loop(ctx, ex, config.ClientId, config.SessionId, config.SessionKey)
+func (ex *Exchange) PollOrderActions(ctx context.Context) {
 }
 
 func (ex *Exchange) PollMixinNetwork(ctx context.Context) {
-	checkpoint, limit := persistence.ReadLatestAction(ctx).UTC(), 500
+	checkpoint, limit := persistence.ReadActionCheckpoint(ctx).UTC(), 500
 	for {
 		snapshots, err := ex.requestMixinNetwork(ctx, checkpoint, limit)
 		if err != nil {
@@ -58,6 +58,10 @@ func (ex *Exchange) PollMixinNetwork(ctx context.Context) {
 	}
 }
 
+func (ex *Exchange) PollMixinMessages(ctx context.Context) {
+	bot.Loop(ctx, ex, config.ClientId, config.SessionId, config.SessionKey)
+}
+
 func (ex *Exchange) OnMessage(ctx context.Context, mc *bot.MessageContext, msg bot.MessageView, userId string) error {
 	log.Println(msg, userId)
 	return nil

main.go

@@ -1,8 +1,26 @@
 package main
 
-import "context"
+import (
+	"context"
+	"log"
+	"time"
+
+	"cloud.google.com/go/spanner"
+	"github.com/MixinMessenger/ocean.one/config"
+	"github.com/MixinMessenger/ocean.one/persistence"
+)
 
 func main() {
-	ex := NewExchange()
-	ex.Run(context.Background())
+	ctx := context.Background()
+	client, err := spanner.NewClientWithConfig(ctx, config.GoogleCloudSpanner, spanner.ClientConfig{NumChannels: 4,
+		SessionPoolConfig: spanner.SessionPoolConfig{
+			HealthCheckInterval: 5 * time.Second,
+		},
+	})
+	if err != nil {
+		log.Panicln(err)
+	}
+
+	ctx = persistence.SetupSpanner(ctx, client)
+	NewExchange().Run(ctx)
 }

mixin.go

@@ -62,7 +62,7 @@ func (ex *Exchange) processSnapshot(ctx context.Context, s *Snapshot) error {
 		return ex.refundSnapshot(ctx, s)
 	}
 	if action.O.String() != uuid.Nil.String() {
-		return persistence.CancelOrder(ctx, action.O.String())
+		return persistence.CancelOrder(ctx, action.O.String(), s.CreatedAt)
 	}
 
 	amount := number.FromString(s.Amount).RoundFloor(8)

persistence/action.go

@@ -0,0 +1,20 @@
+package persistence
+
+import (
+	"context"
+	"time"
+
+	"github.com/MixinMessenger/go-number"
+)
+
+func ReadActionCheckpoint(ctx context.Context) time.Time {
+	return time.Now()
+}
+
+func CreateOrder(ctx context.Context, userId, traceId string, side, quote, base string, amount, price number.Decimal, createdAt time.Time) error {
+	return nil
+}
+
+func CancelOrder(ctx context.Context, orderId string, createdAt time.Time) error {
+	return nil
+}

persistence/context.go

@@ -0,0 +1,22 @@
+package persistence
+
+import (
+	"context"
+
+	"cloud.google.com/go/spanner"
+)
+
+type contextValueKey int
+
+const (
+	keySpanner contextValueKey = 1
+)
+
+func SetupSpanner(ctx context.Context, client *spanner.Client) context.Context {
+	return context.WithValue(ctx, keySpanner, client)
+}
+
+func Spanner(ctx context.Context) *spanner.Client {
+	v, _ := ctx.Value(keySpanner).(*spanner.Client)
+	return v
+}

persistence/order.go

@@ -0,0 +1 @@
+package persistence

persistence/trade.go

@@ -0,0 +1 @@
+package persistence