chore: add logs for prover-proxy (#4)

* chore: add logs for prover-proxy * chore: add indent
kroma-network · May 7, 2024 · 7fb7317 · 7fb7317
1 parent 73d9e11
commit 7fb7317
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 0 deletions.
diff --git a/internal/ec2/controller.go b/internal/ec2/controller.go
@@ -99,6 +99,7 @@ func (c *Controller) StartIfNotRunning() error {
 	c.mu.Lock()
 	defer c.mu.Unlock()
 	if c.running {
+		log.Println("instance is already running")
 		return nil
 	}
 	for {
@@ -112,6 +113,7 @@ func (c *Controller) StartIfNotRunning() error {
 		time.Sleep(1 * time.Second)
 	}
 	_, err := c.client.StartInstances(&ec2.StartInstancesInput{InstanceIds: c.instanceIds()})
+	log.Printf("start instance (id: %s)", c.instanceId)
 	if err != nil {
 		log.Println(fmt.Errorf("failed to start ec2 instance %s: %w", c.instanceId, err))
 		return err
@@ -124,6 +126,7 @@ func (c *Controller) StopIfRunning() {
 	c.mu.Lock()
 	defer c.mu.Unlock()
 	if c.running {
+		log.Printf("stop instance (id: %s)", c.instanceId)
 		_, err := c.client.StopInstances(&ec2.StopInstancesInput{InstanceIds: c.instanceIds()})
 		if err == nil {
 			c.running = false

diff --git a/internal/proof/client.go b/internal/proof/client.go
@@ -55,10 +55,12 @@ func NewJsonRpcErrorFromErrorOrNil(err error) (rpcError *JsonRpcError) {
 func (j *JsonRpcError) Error() string { return fmt.Sprintf("[%d] %s", j.Code, j.Message) }
 
 func (d dialJsonRpcProverClient) Prove(traceString string) (*ProveResponse, error) {
+	log.Println("send request to generate proof to prover")
 	return send[ProveResponse](d.address, "prove", []any{traceString})
 }
 
 func (d dialJsonRpcProverClient) Spec() (*ProverSpecResponse, error) {
+	log.Println("send request of spec")
 	return send[ProverSpecResponse](d.address, "spec", nil)
 }
 

diff --git a/internal/proof/server.go b/internal/proof/server.go
@@ -4,6 +4,7 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
+	"log"
 	"net/http"
 )
 
@@ -76,13 +77,15 @@ func (s *Server) serveJsonRpc(writer http.ResponseWriter, httpRequest *http.Requ
 func (s *Server) callMethod(method string, params interface{}) (any, error) {
 	switch method {
 	case "prove":
+		log.Println("prove requested")
 		p := params.([]any)
 		traceString, traceStringOk := p[0].(string)
 		if !traceStringOk {
 			return nil, errors.New("failed to read traceString parameter")
 		}
 		return s.service.Prove(traceString)
 	case "spec":
+		log.Println("spec requested")
 		return s.service.Spec()
 	default:
 		return nil, fmt.Errorf("unsupported method %s", method)

diff --git a/internal/proof/service.go b/internal/proof/service.go
@@ -30,6 +30,7 @@ func NewService(disk *DiskRepository, ec2 *ec2.Controller) *Service {
 
 func (s *Service) Prove(traceString string) (*ProveResponse, error) {
 	id, blockNumber := computeId(traceString), readBlockNumber(traceString)
+	log.Printf("request prove for block number %s to prover", blockNumber)
 	if proof := s.disk.Find(id); proof != nil {
 		return newProofResponseFromFileProof(proof)
 	}
@@ -79,6 +80,7 @@ func (s *Service) Prove(traceString string) (*ProveResponse, error) {
 }
 
 func (s *Service) Spec() (*ProverSpecResponse, error) {
+	log.Println("request spec to prover")
 	return withClient(s, func(c ProverClient) (*ProverSpecResponse, error) { return c.Spec() })
 }
 
@@ -89,6 +91,7 @@ func (s *Service) Close() {
 func withClient[R interface{}](s *Service, callback func(c ProverClient) (*R, error)) (*R, error) {
 	defer func() {
 		if len(s.inProgressProof) == 0 {
+			log.Println("there is no proof in progress. shut down if it is running.")
 			s.ec2.StopIfRunning()
 		}
 	}()