Skip to content

Commit eb5db38

Browse files
committed
Use balance from Geyser update to for real-time VM deposit flow
1 parent a526198 commit eb5db38

File tree

2 files changed

+9
-1
lines changed

2 files changed

+9
-1
lines changed

pkg/code/async/geyser/external_deposit.go

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -79,6 +79,14 @@ func maybeInitiateExternalDepositIntoVm(ctx context.Context, data code_data.Prov
7979
if balance == 0 {
8080
return nil
8181
}
82+
return initiateExternalDepositIntoVm(ctx, data, vmIndexerClient, userAuthority, balance)
83+
}
84+
85+
func initiateExternalDepositIntoVm(ctx context.Context, data code_data.Provider, vmIndexerClient indexerpb.IndexerClient, userAuthority *common.Account, balance uint64) error {
86+
vmDepositAccounts, err := userAuthority.GetVmDepositAccounts(common.CodeVmAccount, common.CoreMintAccount)
87+
if err != nil {
88+
return errors.Wrap(err, "error getting vm deposit ata")
89+
}
8290

8391
memoryAccount, memoryIndex, err := getVirtualTimelockAccountLocationInMemory(ctx, vmIndexerClient, common.CodeVmAccount, userAuthority)
8492
if err != nil {

pkg/code/async/geyser/handler.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ func (h *TokenProgramAccountHandler) Handle(ctx context.Context, update *geyserp
102102
}
103103

104104
if unmarshalled.Amount > 0 {
105-
err = maybeInitiateExternalDepositIntoVm(ctx, h.data, h.vmIndexerClient, userAuthorityAccount)
105+
err = initiateExternalDepositIntoVm(ctx, h.data, h.vmIndexerClient, userAuthorityAccount, unmarshalled.Amount)
106106
if err != nil {
107107
return errors.Wrap(err, "error depositing into the vm")
108108
}

0 commit comments

Comments
 (0)