fix: tx msgServer fixes

cheqd · Feb 21, 2024 · 44874df · 44874df
1 parent 23aca74
commit 44874df
Show file tree

Hide file tree

Showing 5 changed files with 17 additions and 2 deletions.
diff --git a/app/app.go b/app/app.go
@@ -277,6 +277,7 @@ func New(
 	bApp.SetVersion(version.Version)
 	bApp.SetProtocolVersion(ProtocolVersion)
 	bApp.SetInterfaceRegistry(interfaceRegistry)
+	bApp.SetTxEncoder(txConfig.TxEncoder())
 
 	keys := sdk.NewKVStoreKeys(
 		authtypes.StoreKey,
@@ -355,7 +356,6 @@ func New(
 		BlockedAddresses(),
 		authtypes.NewModuleAddress(govtypes.ModuleName).String(),
 	)
-
 	app.AuthzKeeper = authzkeeper.NewKeeper(
 		keys[authzkeeper.StoreKey],
 		appCodec,

diff --git a/x/did/keeper/msg_server.go b/x/did/keeper/msg_server.go
@@ -10,6 +10,13 @@ type MsgServer struct {
 	Keeper
 }
 
+// NewMsgServerImpl returns an implementation of the x/auth MsgServer interface.
+func NewMsgServerImpl(keeper Keeper) types.MsgServer {
+	return &MsgServer{
+		keeper,
+	}
+}
+
 // NewMsgServer returns an implementation of the MsgServer interface for the provided Keeper.
 func NewMsgServer(keeper Keeper) types.MsgServer {
 	return &MsgServer{Keeper: keeper}

diff --git a/x/did/module.go b/x/did/module.go
@@ -136,6 +136,7 @@ func (am AppModule) Name() string {
 // module-specific GRPC queries.
 func (am AppModule) RegisterServices(cfg module.Configurator) {
 	types.RegisterQueryServer(cfg.QueryServer(), am.keeper)
+	types.RegisterMsgServer(cfg.MsgServer(), keeper.NewMsgServerImpl(am.keeper))
 }
 
 // RegisterInvariants registers the cheqd module's invariants.

diff --git a/x/resource/keeper/msg_server.go b/x/resource/keeper/msg_server.go
@@ -10,6 +10,13 @@ type msgServer struct {
 	didKeeper didkeeper.Keeper
 }
 
+// NewMsgServerImpl returns an implementation of the x/auth MsgServer interface.
+func NewMsgServerImpl(keeper Keeper) types.MsgServer {
+	return &msgServer{
+		Keeper: keeper,
+	}
+}
+
 // NewMsgServer returns an implementation of the MsgServer interface for the provided Keeper.
 func NewMsgServer(keeper Keeper, cheqdKeeper didkeeper.Keeper) types.MsgServer {
 	return &msgServer{

diff --git a/x/resource/module.go b/x/resource/module.go
@@ -139,7 +139,7 @@ func (am AppModule) Name() string {
 // module-specific GRPC queries.
 func (am AppModule) RegisterServices(cfg module.Configurator) {
 	types.RegisterQueryServer(cfg.QueryServer(), keeper.NewQueryServer(am.keeper, am.didKeeper))
-
+	types.RegisterMsgServer(cfg.MsgServer(), keeper.NewMsgServerImpl(am.keeper))
 	// Register for migration from consensus version 2 -> 3
 	migratorV3 := migrationV3.NewMigrator(am.keeper)
 	if err := cfg.RegisterMigration(types.ModuleName, 2, migratorV3.Migrate2to3); err != nil {