split GetBtClient() from chained functions

bluzelle · Aug 12, 2021 · eba36ca · eba36ca
1 parent c3c0be1
commit eba36ca
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/x/nft/handler.go b/x/nft/handler.go
@@ -47,7 +47,8 @@ func handleMsgCreateNft(goCtx sdk.Context, k keeper.Keeper, msg *types.MsgCreate
 
 
 	if _, err := os.Stat(k.HomeDir+"/nft/"+msg.Hash); err == nil {
-		metainfo, err := k.GetBtClient().TorrentFromFile(msg.Hash)
+		btClient := k.GetBtClient()
+		metainfo, err := btClient.TorrentFromFile(msg.Hash)
 		if err != nil {
 			return nil, sdkerrors.New("nft", 2, fmt.Sprintf("unable to create torrent for file", msg.Hash))
 		}
@@ -83,7 +84,10 @@ func handleMsgPublishFile(ctx sdk.Context, k Keeper, msg *types.MsgPublishFile)
 	k.Logger(ctx).Debug("Publish file message received", "id", msg.Id)
 	var metainfo metainfo.MetaInfo
 	bencode.DecodeBytes(msg.Metainfo, &metainfo)
-	k.GetBtClient().RetrieveFile(&metainfo)
+
+	btClient := k.GetBtClient()
+
+	btClient.RetrieveFile(&metainfo)
 	k.EnsureNftDirExists()
 
 
@@ -122,7 +126,8 @@ func handleMsgRegisterPeer(ctx sdk.Context, k Keeper, msg *types.MsgRegisterPeer
 	store.Set([]byte(msg.Id), k.Cdc.MustMarshalBinaryBare(&peer))
 
 	if k.GetMyNodeId(ctx) != msg.Id {
-		k.GetBtClient().AddPeer(msg.Id, msg.Address, int(msg.Port))
+		btClient := k.GetBtClient()
+		btClient.AddPeer(msg.Id, msg.Address, int(msg.Port))
 	}
 
 	return &sdk.Result{}, nil