From f87836ed81f80aacd385637f2e4517a2aeffc2a0 Mon Sep 17 00:00:00 2001
From: Mohamad Fadhil <sdil@users.noreply.github.com>
Date: Tue, 17 Dec 2024 18:01:25 +0800
Subject: [PATCH] feat: add durable statestore example (#102)

---
 example/durablestate/main.go        | 145 ++++++++++++++++++++++++++++
 example/{ => eventssourced}/main.go |   0
 2 files changed, 145 insertions(+)
 create mode 100644 example/durablestate/main.go
 rename example/{ => eventssourced}/main.go (100%)

diff --git a/example/durablestate/main.go b/example/durablestate/main.go
new file mode 100644
index 0000000..f48a30c
--- /dev/null
+++ b/example/durablestate/main.go
@@ -0,0 +1,145 @@
+/*
+ * MIT License
+ *
+ * Copyright (c) 2022-2024 Tochemey
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all
+ * copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ */
+
+package main
+
+import (
+	"context"
+	"errors"
+	"log"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
+	"github.com/google/uuid"
+	"google.golang.org/protobuf/proto"
+
+	"github.com/tochemey/ego/v3"
+	samplepb "github.com/tochemey/ego/v3/example/pbs/sample/pb/v1"
+	"github.com/tochemey/ego/v3/plugins/statestore/memory"
+)
+
+func main() {
+	// create the go context
+	ctx := context.Background()
+	// create the event store
+	durableStore := memory.NewStateStore()
+	// connect the event store
+	_ = durableStore.Connect(ctx)
+	// create the ego engine
+	engine := ego.NewEngine("Sample", nil, ego.WithStateStore(durableStore))
+	// start ego engine
+	_ = engine.Start(ctx)
+	// create a persistence id
+	entityID := uuid.NewString()
+	// create an entity behavior with a given id
+	behavior := NewAccountBehavior(entityID)
+	// create an entity
+	_ = engine.DurableStateEntity(ctx, behavior)
+
+	// send some commands to the pid
+	var command proto.Message
+	// create an account
+	command = &samplepb.CreateAccount{
+		AccountId:      entityID,
+		AccountBalance: 500.00,
+	}
+	// send the command to the actor. Please don't ignore the error in production grid code
+	reply, _, _ := engine.SendCommand(ctx, entityID, command, time.Minute)
+	account := reply.(*samplepb.Account)
+	log.Printf("current balance on opening: %v", account.GetAccountBalance())
+
+	// send another command to credit the balance
+	command = &samplepb.CreditAccount{
+		AccountId: entityID,
+		Balance:   250,
+	}
+
+	reply, _, _ = engine.SendCommand(ctx, entityID, command, time.Minute)
+	account = reply.(*samplepb.Account)
+	log.Printf("current balance after a credit of 250: %v", account.GetAccountBalance())
+
+	// capture ctrl+c
+	interruptSignal := make(chan os.Signal, 1)
+	signal.Notify(interruptSignal, os.Interrupt, syscall.SIGINT, syscall.SIGTERM)
+	<-interruptSignal
+
+	// disconnect the event store
+	_ = durableStore.Disconnect(ctx)
+	// stop the actor system
+	_ = engine.Stop(ctx)
+	os.Exit(0)
+}
+
+// AccountBehavior implements EventSourcedBehavior
+type AccountBehavior struct {
+	id string
+}
+
+// make sure that AccountBehavior is a true persistence behavior
+var _ ego.DurableStateBehavior = &AccountBehavior{}
+
+// NewAccountBehavior creates an instance of AccountBehavior
+func NewAccountBehavior(id string) *AccountBehavior {
+	return &AccountBehavior{id: id}
+}
+
+// ID returns the id
+func (a *AccountBehavior) ID() string {
+	return a.id
+}
+
+// InitialState returns the initial state
+func (a *AccountBehavior) InitialState() ego.State {
+	return ego.State(new(samplepb.Account))
+}
+
+// HandleCommand handles every command that is sent to the persistent behavior
+func (a *AccountBehavior) HandleCommand(_ context.Context, command ego.Command, priorVersion uint64, priorState ego.State) (event ego.State, newVersion uint64, err error) {
+	switch cmd := command.(type) {
+	case *samplepb.CreateAccount:
+		// TODO in production grid app validate the command using the prior state
+		return &samplepb.Account{
+			AccountId:      a.id,
+			AccountBalance: cmd.GetAccountBalance(),
+		}, priorVersion + 1, nil
+
+	case *samplepb.CreditAccount:
+		if cmd.GetAccountId() == a.id {
+			// TODO in production grid app validate the command using the prior state
+			account := priorState.(*samplepb.Account)
+			bal := account.GetAccountBalance() + cmd.GetBalance()
+
+			return &samplepb.Account{
+				AccountId:      a.id,
+				AccountBalance: bal,
+			}, priorVersion + 1, nil
+		}
+		return nil, 0, errors.New("command sent to the wrong entity")
+
+	default:
+		return nil, 0, errors.New("unhandled command")
+	}
+}
diff --git a/example/main.go b/example/eventssourced/main.go
similarity index 100%
rename from example/main.go
rename to example/eventssourced/main.go