89 lines
2.3 KiB
Go
89 lines
2.3 KiB
Go
package keeper
|
|
|
|
import (
|
|
"context"
|
|
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
|
|
"github.com/cosmos/cosmos-sdk/store/prefix"
|
|
"github.com/cosmos/cosmos-sdk/types/query"
|
|
|
|
proto "github.com/gogo/protobuf/proto"
|
|
|
|
codectypes "github.com/cosmos/cosmos-sdk/codec/types"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/evidence/types"
|
|
)
|
|
|
|
var _ types.QueryServer = Keeper{}
|
|
|
|
// Evidence implements the Query/Evidence gRPC method
|
|
func (k Keeper) Evidence(c context.Context, req *types.QueryEvidenceRequest) (*types.QueryEvidenceResponse, error) {
|
|
if req == nil {
|
|
return nil, status.Errorf(codes.InvalidArgument, "empty request")
|
|
}
|
|
|
|
if req.EvidenceHash == nil {
|
|
return nil, status.Errorf(codes.InvalidArgument, "invalid hash")
|
|
}
|
|
|
|
ctx := sdk.UnwrapSDKContext(c)
|
|
|
|
evidence, _ := k.GetEvidence(ctx, req.EvidenceHash)
|
|
if evidence == nil {
|
|
return nil, status.Errorf(codes.NotFound, "evidence %s not found", req.EvidenceHash)
|
|
}
|
|
|
|
msg, ok := evidence.(proto.Message)
|
|
if !ok {
|
|
return nil, status.Errorf(codes.Internal, "can't protomarshal %T", msg)
|
|
}
|
|
|
|
evidenceAny, err := codectypes.NewAnyWithValue(msg)
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Internal, err.Error())
|
|
}
|
|
|
|
return &types.QueryEvidenceResponse{Evidence: evidenceAny}, nil
|
|
}
|
|
|
|
// AllEvidence implements the Query/AllEvidence gRPC method
|
|
func (k Keeper) AllEvidence(c context.Context, req *types.QueryAllEvidenceRequest) (*types.QueryAllEvidenceResponse, error) {
|
|
if req == nil {
|
|
return nil, status.Errorf(codes.InvalidArgument, "empty request")
|
|
}
|
|
ctx := sdk.UnwrapSDKContext(c)
|
|
|
|
k.GetAllEvidence(ctx)
|
|
|
|
var evidence []*codectypes.Any
|
|
store := ctx.KVStore(k.storeKey)
|
|
evidenceStore := prefix.NewStore(store, types.KeyPrefixEvidence)
|
|
|
|
pageRes, err := query.Paginate(evidenceStore, req.Pagination, func(key []byte, value []byte) error {
|
|
result, err := k.UnmarshalEvidence(value)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
msg, ok := result.(proto.Message)
|
|
if !ok {
|
|
return status.Errorf(codes.Internal, "can't protomarshal %T", msg)
|
|
}
|
|
|
|
evidenceAny, err := codectypes.NewAnyWithValue(msg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
evidence = append(evidence, evidenceAny)
|
|
return nil
|
|
})
|
|
|
|
if err != nil {
|
|
return &types.QueryAllEvidenceResponse{}, err
|
|
}
|
|
|
|
return &types.QueryAllEvidenceResponse{Evidence: evidence, Pagination: pageRes}, nil
|
|
}
|