108 lines
2.6 KiB
Go
108 lines
2.6 KiB
Go
package codec_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
protov2 "google.golang.org/protobuf/proto"
|
|
"google.golang.org/protobuf/types/dynamicpb"
|
|
|
|
bankv1beta1 "cosmossdk.io/api/cosmos/bank/v1beta1"
|
|
|
|
codectestutil "github.com/cosmos/cosmos-sdk/codec/testutil"
|
|
codectypes "github.com/cosmos/cosmos-sdk/codec/types"
|
|
"github.com/cosmos/cosmos-sdk/testutil/testdata"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
|
|
)
|
|
|
|
type bankSendWrapper struct {
|
|
*banktypes.MsgSend
|
|
}
|
|
|
|
func (msg bankSendWrapper) GetSigners() []sdk.AccAddress {
|
|
fromAddress, _ := sdk.AccAddressFromBech32(msg.FromAddress)
|
|
return []sdk.AccAddress{fromAddress}
|
|
}
|
|
|
|
func BenchmarkLegacyGetSigners(b *testing.B) {
|
|
_, _, addr := testdata.KeyTestPubAddr()
|
|
msg := bankSendWrapper{&banktypes.MsgSend{
|
|
FromAddress: addr.String(),
|
|
ToAddress: "",
|
|
Amount: nil,
|
|
}}
|
|
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
_ = msg.GetSigners()
|
|
}
|
|
}
|
|
|
|
func BenchmarkProtoreflectGetSigners(b *testing.B) {
|
|
cdc := codectestutil.CodecOptions{}.NewCodec()
|
|
signingCtx := cdc.InterfaceRegistry().SigningContext()
|
|
_, _, addr := testdata.KeyTestPubAddr()
|
|
// use a pulsar message
|
|
msg := &bankv1beta1.MsgSend{
|
|
FromAddress: addr.String(),
|
|
ToAddress: "",
|
|
Amount: nil,
|
|
}
|
|
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
_, err := signingCtx.GetSigners(msg)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkProtoreflectGetSignersWithUnmarshal(b *testing.B) {
|
|
cdc := codectestutil.CodecOptions{}.NewCodec()
|
|
_, _, addr := testdata.KeyTestPubAddr()
|
|
// start with a protoreflect message
|
|
msg := &banktypes.MsgSend{
|
|
FromAddress: addr.String(),
|
|
ToAddress: "",
|
|
Amount: nil,
|
|
}
|
|
// marshal to an any first because this is what we get from the wire
|
|
a, err := codectypes.NewAnyWithValue(msg)
|
|
require.NoError(b, err)
|
|
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
_, _, err := cdc.GetMsgAnySigners(a)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkProtoreflectGetSignersDynamicpb(b *testing.B) {
|
|
cdc := codectestutil.CodecOptions{}.NewCodec()
|
|
signingCtx := cdc.InterfaceRegistry().SigningContext()
|
|
_, _, addr := testdata.KeyTestPubAddr()
|
|
msg := &bankv1beta1.MsgSend{
|
|
FromAddress: addr.String(),
|
|
ToAddress: "",
|
|
Amount: nil,
|
|
}
|
|
bz, err := protov2.Marshal(msg)
|
|
require.NoError(b, err)
|
|
|
|
dynamicmsg := dynamicpb.NewMessage(msg.ProtoReflect().Descriptor())
|
|
err = protov2.Unmarshal(bz, dynamicmsg)
|
|
require.NoError(b, err)
|
|
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
_, err := signingCtx.GetSigners(dynamicmsg)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
}
|