Skip to content

Commit

Permalink
move torrent client creation right before broadcast
Browse files Browse the repository at this point in the history
  • Loading branch information
avendauz committed Aug 12, 2021
1 parent eba36ca commit 78153de
Showing 1 changed file with 14 additions and 11 deletions.
25 changes: 14 additions & 11 deletions x/nft/module.go
Original file line number Diff line number Diff line change
Expand Up @@ -145,28 +145,31 @@ func (am AppModule) ExportGenesis(ctx sdk.Context) json.RawMessage {
func (am AppModule) BeginBlock(_ sdk.Context, req abci.RequestBeginBlock) {
}

var once sync.Once
var registerPeerOnce sync.Once
// EndBlock returns the end blocker for the nft module. It returns no validator
// updates.
func (am AppModule) EndBlock(ctx sdk.Context, _ abci.RequestEndBlock) []abci.ValidatorUpdate {

once.Do(func () {
btClient, err := torrentClient.NewTorrentClient(am.btDirectory, am.btPort)

if err != nil {
am.keeper.Logger(ctx).Error("Error creating btClient", "btClient", err)
}

am.keeper.SetBtClient(btClient)
})

err := am.keeper.CheckNftUserExists(ctx, am.keeper.KeyringReader, am.accKeeper)
fmt.Println("Checking nft user exists", err)
if err != nil {
am.keeper.Logger(ctx).Error("nft user does not exist in keyring", "nft", err)
} else {
registerPeerOnce.Do(func () {

btClient, err := torrentClient.NewTorrentClient(am.btDirectory, am.btPort)
fmt.Println("Torrent client created", *btClient)
if err != nil {
am.keeper.Logger(ctx).Error("Error creating btClient", "btClient", err)
}

fmt.Println("Bt client before setting", *am.keeper.GetBtClient(), am.keeper.GetBtClient())

am.keeper.SetBtClient(btClient)

fmt.Println("Bt client after setting", *am.keeper.GetBtClient(), am.keeper.GetBtClient())


fmt.Println("Doing broadcast register peer")
go func() {
err := am.keeper.BroadcastRegisterBtPeer(ctx)
Expand Down

0 comments on commit 78153de

Please sign in to comment.