Use atomic.bool for flag AllowedToSend

Author: konradkonrad

rolling-shutter/keyper/fx/messagesender.go

@@ -7,6 +7,7 @@ import (
 	"encoding/base64"
 	"encoding/binary"
 	"fmt"
+	"sync/atomic"
 
 	"github.com/pkg/errors"
 	"github.com/rs/zerolog/log"
@@ -49,7 +50,7 @@ type RPCMessageSender struct {
 	rpcclient     client.Client
 	chainID       string
 	signingKey    *ecdsa.PrivateKey
-	AllowedToSend bool
+	AllowedToSend *atomic.Bool
 }
 
 var _ MessageSender = &RPCMessageSender{}
@@ -65,17 +66,18 @@ var mockMessageSenderBufferSize = 0x10000
 
 // NewRPCMessageSender creates a new RPCMessageSender.
 func NewRPCMessageSender(cl client.Client, signingKey *ecdsa.PrivateKey) RPCMessageSender {
-	return RPCMessageSender{
-		rpcclient:     cl,
-		chainID:       "",
-		signingKey:    signingKey,
-		AllowedToSend: false,
+	ms := RPCMessageSender{
+		rpcclient:  cl,
+		chainID:    "",
+		signingKey: signingKey,
 	}
+	ms.AllowedToSend.Store(false)
+	return ms
 }
 
 // SendMessage signs the given shmsg.Message and sends the message to shuttermint.
 func (ms *RPCMessageSender) SendMessage(ctx context.Context, msg *shmsg.Message) error {
-	if !ms.AllowedToSend {
+	if !ms.AllowedToSend.Load() {
 		log.Info().Str("msg", msg.String()).Msg("not allowed to send")
 		return nil
 	}

rolling-shutter/keyper/keyper.go

@@ -358,7 +358,7 @@ func (kpr *KeyperCore) operateShuttermint(ctx context.Context, _ service.Runner)
 		if err != nil {
 			return err
 		}
-		if !kpr.messageSender.AllowedToSend {
+		if !kpr.messageSender.AllowedToSend.Load() {
 			allowSendIfInKeyperSet(ctx, database.New(kpr.dbpool), syncBlockNumber, kpr)
 		}
 		err = fx.SendShutterMessages(ctx, database.New(kpr.dbpool), &kpr.messageSender)
@@ -384,6 +384,6 @@ func allowSendIfInKeyperSet(ctx context.Context, queries *database.Queries, sync
 		log.Err(err).Msg("could not query if in keyper set")
 	}
 	if count > 0 {
-		kpr.messageSender.AllowedToSend = true
+		kpr.messageSender.AllowedToSend.Store(true)
 	}
 }