Add preloading of keys.

Signed-off-by: Cody Littley <[email protected]>

relay/auth/authenticator.go

@@ -20,6 +20,9 @@ type RequestAuthenticator interface {
 		origin string,
 		request *pb.GetChunksRequest,
 		now time.Time) error
+
+	// PreloadCache preloads the key cache with the public keys of all operators that are currently indexed.
+	PreloadCache() error
 }
 
 // authenticationTimeout is used to track the expiration of an auth.
@@ -71,6 +74,23 @@ func NewRequestAuthenticator(
 	}, nil
 }
 
+func (a *requestAuthenticator) PreloadCache() error {
+	blockNumber, err := a.ics.GetCurrentBlockNumber()
+	if err != nil {
+		return fmt.Errorf("failed to get current block number: %w", err)
+	}
+	operators, err := a.ics.GetIndexedOperators(context.Background(), blockNumber)
+	if err != nil {
+		return fmt.Errorf("failed to get operators: %w", err)
+	}
+
+	for operatorID, operator := range operators {
+		a.keyCache.Add(operatorID, operator.PubkeyG2)
+	}
+
+	return nil
+}
+
 func (a *requestAuthenticator) AuthenticateGetChunksRequest(
 	origin string,
 	request *pb.GetChunksRequest,

relay/server.go

@@ -155,6 +155,11 @@ func NewServer(
 		if err != nil {
 			return nil, fmt.Errorf("error creating authenticator: %w", err)
 		}
+
+		err = authenticator.PreloadCache()
+		if err != nil {
+			return nil, fmt.Errorf("error preloading operator key cache: %w", err)
+		}
 	}
 
 	return &Server{