[PoC] Activation Service builds and publishes ATXs with Node Service help

Makefile

@@ -6,6 +6,7 @@ GOLANGCI_LINT_VERSION := v1.61.0
 GOTESTSUM_VERSION := v1.12.0
 GOSCALE_VERSION := v1.2.0
 MOCKGEN_VERSION := v0.5.0
+OAPI_CODEGEN_VERSION := v2.4.0
 
 # Add an indicator to the branch name if dirty and use commithash if running in detached mode
 ifeq ($(BRANCH),HEAD)
@@ -68,6 +69,7 @@ install:
 	go install github.com/spacemeshos/go-scale/scalegen@$(GOSCALE_VERSION)
 	go install go.uber.org/mock/mockgen@$(MOCKGEN_VERSION)
 	go install gotest.tools/gotestsum@$(GOTESTSUM_VERSION)
+	go install github.com/oapi-codegen/oapi-codegen/v2/cmd/oapi-codegen@$(OAPI_CODEGEN_VERSION)
 .PHONY: install
 
 build: go-spacemesh get-profiler get-postrs-service

activation/interface.go

@@ -200,6 +200,10 @@ type certifierService interface {
 	DeleteCertificate(id types.NodeID, pubkey []byte) error
 }
 
+type PoetDbStorer interface {
+	StorePoetProof(ctx context.Context, proofMessage *types.PoetProofMessage) error
+}
+
 type poetDbAPI interface {
 	Proof(types.PoetProofRef) (*types.PoetProof, *types.Hash32, error)
 	ProofForRound(poetID []byte, roundID string) (*types.PoetProof, error)

activation/poetdb.go

@@ -23,7 +23,8 @@
 
 // PoetDbOptions are options for PoetDb.
 type PoetDbOptions struct {
-	cacheSize int
+	cacheSize    int
+	remoteStorer PoetDbStorer
 }
 
 type PoetDbOption func(*PoetDbOptions)
@@ -35,12 +36,19 @@
 	}
 }
 
+func WithRemotePoetStorer(storer PoetDbStorer) PoetDbOption {
+	return func(opts *PoetDbOptions) {
+		opts.remoteStorer = storer
+	}
+}
+
 // PoetDb is a database for PoET proofs.
 type PoetDb struct {
 	sqlDB               sql.StateDatabase
 	poetProofsDbRequest singleflight.Group
 	poetProofsLru       *lru.Cache[types.PoetProofRef, *types.PoetProofMessage]
 	logger              *zap.Logger
+	remoteStorer        PoetDbStorer
 }
 
 // NewPoetDb returns a new PoET handler.
@@ -66,6 +74,7 @@
 		sqlDB:         db,
 		poetProofsLru: poetProofsLru,
 		logger:        log,
+		remoteStorer:  options.remoteStorer,
 	}, nil
 }
 
@@ -99,6 +108,13 @@
 		return err
 	}
 
+	if db.remoteStorer != nil {
+		err := db.remoteStorer.StorePoetProof(ctx, proofMessage)
+		if err != nil {
+			db.logger.Warn("failed to store the poet proof in remote store", zap.Error(err))
+		}
+	}
+
 	return db.StoreProof(ctx, ref, proofMessage)
 }
 
@@ -178,13 +194,17 @@
 }
 
 // GetProofMessage returns the originally received PoET proof message.
-func (db *PoetDb) GetProofMessage(proofRef types.PoetProofRef) ([]byte, error) {
+func (db *PoetDb) ProofMessage(proofRef types.PoetProofRef) (*types.PoetProofMessage, error) {
 	proof, err := poets.Get(db.sqlDB, proofRef)
 	if err != nil {
-		return proof, fmt.Errorf("get proof from store: %w", err)
+		return nil, fmt.Errorf("get proof from store: %w", err)
+	}
+	var proofMessage types.PoetProofMessage
+	if err := codec.Decode(proof, &proofMessage); err != nil {
+		return nil, fmt.Errorf("failed to unmarshal poet proof for ref %x: %w", proofRef, err)
 	}
 
-	return proof, nil
+	return &proofMessage, nil
 }
 
 // Proof returns full proof.
@@ -194,16 +214,12 @@
 		if ok && cachedProof != nil {
 			return cachedProof, nil
 		}
-		proofMessageBytes, err := db.GetProofMessage(proofRef)
+		proofMessage, err := db.ProofMessage(proofRef)
 		if err != nil {
 			return nil, fmt.Errorf("could not fetch poet proof for ref %x: %w", proofRef, err)
 		}
-		var proofMessage types.PoetProofMessage
-		if err := codec.Decode(proofMessageBytes, &proofMessage); err != nil {
-			return nil, fmt.Errorf("failed to unmarshal poet proof for ref %x: %w", proofRef, err)
-		}
-		db.poetProofsLru.Add(proofRef, &proofMessage)
-		return &proofMessage, nil
+		db.poetProofsLru.Add(proofRef, proofMessage)
+		return proofMessage, nil
 	})
 	if err != nil {
 		return nil, nil, err

activation_service_poc/config.standalone.client.json

@@ -0,0 +1,31 @@
+{
+    "preset": "standalone",
+    "api": {
+        "grpc-public-listener": "0.0.0.0:9082",
+        "grpc-private-listener": "0.0.0.0:9083"
+    },
+    "genesis": {
+        "genesis-time": "2024-09-25T13:00:00.000Z"
+    },
+    "logging": {
+        "trtl": "WARN",
+        "beacon": "ERROR",
+        "proposalBuilder": "ERROR",
+        "atxBuilder": "DEBUG"
+    },
+    "main": {
+        "node-service-address": "http://0.0.0.0:9099",
+        "data-folder": "/tmp/spacemesh-client",
+        "filelock": "/tmp/spacemesh-client/node.lock",
+        "poet-servers": [
+            {
+                "address": "http://127.0.0.1:10011"
+            }
+        ]
+    },
+    "smeshing": {
+        "smeshing-opts": {
+            "smeshing-opts-datadir": "/tmp/spacemesh-client/post-data"
+        }
+    }
+}

activation_service_poc/config.standalone.node-service.json

@@ -0,0 +1,23 @@
+{
+    "preset": "standalone",
+    "api": {
+        "grpc-public-listener": "0.0.0.0:9092",
+        "grpc-private-listener": "0.0.0.0:9093",
+        "node-service-listener": "0.0.0.0:9099"
+    },
+    "genesis": {
+        "genesis-time": "2024-09-25T13:00:00.000Z"
+    },
+    "logging": {
+        "trtl": "WARN",
+        "beacon": "ERROR",
+        "proposalBuilder": "ERROR"
+    },
+    "main": {
+        "data-folder": "/tmp/spacemesh-node-service",
+        "filelock": "/tmp/spacemesh-node-service/node.lock"
+    },
+    "smeshing": {
+        "smeshing-start": false
+    }
+}

activation_service_poc/docker-compose.yml

@@ -0,0 +1,24 @@
+version: '3'
+
+services:
+  activation-service:
+    image: spacemeshos/go-spacemesh-dev:activation-service-poc.0
+    command: ["-c", "/config.json", "--node-service-address", "http://node-service:9099"]
+    volumes:
+      - /tmp/spacemesh-client:/tmp/spacemesh-client
+      - ./config.standalone.client.json:/config.json
+    networks:
+      - spacemesh-net
+
+  node-service:
+    image: spacemeshos/go-spacemesh-dev:activation-service-poc.0
+    command: ["-c", "/config.json"]
+    volumes:
+      - /tmp/spacemesh-node-service:/tmp/spacemesh-node-service
+      - ./config.standalone.node-service.json:/config.json
+    networks:
+      - spacemesh-net
+
+networks:
+  spacemesh-net:
+    driver: bridge

api/grpcserver/config.go

@@ -22,6 +22,8 @@ type Config struct {
 	JSONListener           string    `mapstructure:"grpc-json-listener"`
 	JSONCorsAllowedOrigins []string  `mapstructure:"grpc-cors-allowed-origins"`
 
+	NodeServiceListener string `mapstructure:"node-service-listener"`
+
 	SmesherStreamInterval time.Duration `mapstructure:"smesherstreaminterval"`
 
 	DatabaseConnections int `mapstructure:"db-connections"`