Code cleanup

Author: anhthii

cmd/mpcium/main.go

@@ -184,7 +184,7 @@ func runNode(ctx context.Context, c *cli.Command) error {
 	if err := peerRegistry.Ready(); err != nil {
 		logger.Error("Failed to mark peer registry as ready", err)
 	}
-
+	logger.Info("[READY] Node is ready", "nodeID", nodeID)
 	appContext, cancel := context.WithCancel(context.Background())
 	// Setup signal handling to cancel context on termination signals.
 	go func() {

pkg/eventconsumer/event_consumer.go

@@ -127,6 +127,7 @@ func (ec *eventConsumer) handleKeyGenEvent(natMsg *nats.Msg) {
 		ec.handleKeygenSessionError(msg.WalletID, err, "Failed to unmarshal keygen message")
 		return
 	}
+
 	if err := ec.identityStore.VerifyInitiatorMessage(&msg); err != nil {
 		logger.Error("Failed to verify initiator message", err)
 		ec.handleKeygenSessionError(msg.WalletID, err, "Failed to verify initiator message")
@@ -216,7 +217,6 @@ func (ec *eventConsumer) handleKeyGenEvent(natMsg *nats.Msg) {
 		return
 	}
 
-	logger.Info("Closing session successfully!", "event", successEvent)
 	payload, err := json.Marshal(successEvent)
 	if err != nil {
 		logger.Error("Failed to marshal keygen success event", err)