Add support for batch operator addresses to IDs translation (Layr-Lab…

…s#1160)

core/chainio.go

@@ -62,6 +62,9 @@ type Reader interface {
 	// BatchOperatorIDToAddress returns the addresses of the operators from the operator id.
 	BatchOperatorIDToAddress(ctx context.Context, operatorIds []OperatorID) ([]gethcommon.Address, error)
 
+	// BatchOperatorAddressToID returns the operator IDs for the given operator addresses.
+	BatchOperatorAddressToID(ctx context.Context, addresses []gethcommon.Address) ([]OperatorID, error)
+
 	// GetCurrentQuorumBitmapByOperatorId returns the current quorum bitmap for the operator.
 	GetCurrentQuorumBitmapByOperatorId(ctx context.Context, operatorId OperatorID) (*big.Int, error)
 

core/eth/reader.go

@@ -527,6 +527,21 @@ func (t *Reader) BatchOperatorIDToAddress(ctx context.Context, operatorIds []cor
 	return addresses, nil
 }
 
+func (t *Reader) BatchOperatorAddressToID(ctx context.Context, addresses []gethcommon.Address) ([]core.OperatorID, error) {
+	ids, err := t.bindings.OpStateRetriever.GetBatchOperatorId(&bind.CallOpts{
+		Context: ctx,
+	}, t.bindings.RegCoordinatorAddr, addresses)
+	if err != nil {
+		t.logger.Error("Failed to get operator IDs in batch", "err", err)
+		return nil, err
+	}
+	operatorIds := make([]core.OperatorID, len(ids))
+	for i, id := range ids {
+		operatorIds[i] = core.OperatorID(id)
+	}
+	return operatorIds, nil
+}
+
 func (t *Reader) GetCurrentQuorumBitmapByOperatorId(ctx context.Context, operatorId core.OperatorID) (*big.Int, error) {
 	return t.bindings.RegistryCoordinator.GetCurrentQuorumBitmap(&bind.CallOpts{
 		Context: ctx,

core/mock/writer.go

@@ -131,6 +131,12 @@ func (t *MockWriter) BatchOperatorIDToAddress(ctx context.Context, operatorIds [
 	return result.([]gethcommon.Address), args.Error(1)
 }
 
+func (t *MockWriter) BatchOperatorAddressToID(ctx context.Context, addresses []gethcommon.Address) ([]core.OperatorID, error) {
+	args := t.Called()
+	result := args.Get(0)
+	return result.([]core.OperatorID), args.Error(1)
+}
+
 func (t *MockWriter) GetQuorumBitmapForOperatorsAtBlockNumber(ctx context.Context, operatorIds []core.OperatorID, blockNumber uint32) ([]*big.Int, error) {
 	args := t.Called()
 	result := args.Get(0)