update with Marius Poke's suggestions

decentrio · Jan 5, 2024 · e1dd2ed · e1dd2ed
1 parent d9ef37b
commit e1dd2ed
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/...improvements/1503-query-key-assignment.md → ...nts/provider/1503-query-key-assignment.md b/...improvements/1503-query-key-assignment.md → ...nts/provider/1503-query-key-assignment.md
diff --git a/x/ccv/provider/keeper/grpc_query.go b/x/ccv/provider/keeper/grpc_query.go
@@ -198,13 +198,15 @@ func (k Keeper) QueryAllPairsValConAddrByConsumerChainID(goCtx context.Context,
 	ctx := sdk.UnwrapSDKContext(goCtx)
 	validatorConsumerAddrs := k.GetAllValidatorsByConsumerAddr(ctx, &req.ChainId)
 	for _, data := range validatorConsumerAddrs {
-		pubKey, _, _ := k.GetKeyAssignmentReplacement(ctx, req.ChainId, types.ProviderConsAddress{Address: data.ProviderAddr})
+		providerAddr := types.NewProviderConsAddress(data.ProviderAddr)
+		pubKey, _ := k.GetValidatorConsumerPubKey(ctx, req.ChainId, providerAddr)
 		pairValConAddrs = append(pairValConAddrs, &types.PairValConAddrProviderAndConsumer{
 			ProviderAddress: string(data.ProviderAddr),
 			ConsumerAddress: string(data.ConsumerAddr),
 			ConsumerKey:     &pubKey,
 		})
 	}
+
 	return &types.QueryAllPairsValConAddrByConsumerChainIDResponse{
 		PairValConAddr: pairValConAddrs,
 	}, nil