Cross Program Invocation

Zaktualizuj instrukcję Update

Najpierw program potrzebuje prostego mechanizmu "pay-to-update" poprzez zmianę Update struktury i funkcji update.

Rozpocznij od aktualizacji pliku lib.rs, aby uwzględnić elementy z modułu system_program.

use anchor_lang::system_program::{transfer, Transfer};

Następnie zaktualizuj strukturę Update, aby uwzględnić nowe konto o nazwie vault_account. To konto, kontrolowane przez program, otrzymuje SOL od użytkownika, gdy aktualizuje on swoje konto wiadomości.

#[account(
    mut,
    seeds = [b"vault", user.key().as_ref()],
    bump,
)]
pub vault_account: SystemAccount<'info>,

Następnie dodaj logikę CPI w instrukcji update, aby przelać 0,001 SOL z konta użytkownika na konto skarbca.

let transfer_accounts = Transfer {
    from: ctx.accounts.user.to_account_info(),
    to: ctx.accounts.vault_account.to_account_info(),
};

let cpi_context = CpiContext::new(
    ctx.accounts.system_program.to_account_info(),
    transfer_accounts,
);

transfer(cpi_context, 1_000_000)?;

Przebuduj program.

$ 
buildCopy

Zaktualizuj instrukcję usuwania

Dodaj teraz mechanizm "zwrotu przy usunięciu" poprzez zmianę struktury Delete i funkcji delete.

Najpierw zaktualizuj strukturę Delete, aby uwzględniała vault_account. To pozwala na transfer dowolnych SOL w skarbcu z powrotem do użytkownika, gdy zamykają swoje konto wiadomości.

#[account(
    mut,
    seeds = [b"vault", user.key().as_ref()],
    bump,
)]
pub vault_account: SystemAccount<'info>,

Dodaj również system_program, ponieważ CPI dla transferu wymaga wywołania System Programu.

pub system_program: Program<'info, System>,

Następnie dodaj logikę CPI w instrukcji delete, aby przelać SOL z konta skarbca z powrotem na konto użytkownika.

let user_key = ctx.accounts.user.key();
let signer_seeds: &[&[&[u8]]] =
    &[&[b"vault", user_key.as_ref(), &[ctx.bumps.vault_account]]];

let transfer_accounts = Transfer {
    from: ctx.accounts.vault_account.to_account_info(),
    to: ctx.accounts.user.to_account_info(),
};
let cpi_context = CpiContext::new(
    ctx.accounts.system_program.to_account_info(),
    transfer_accounts,
).with_signer(signer_seeds);
transfer(cpi_context, ctx.accounts.vault_account.lamports())?;

Zauważ, że _ctx: Context<Delete> zmienia się na ctx: Context<Delete>, aby użyć kontekstu w ciele funkcji.

Przebuduj program.

$ 
buildCopy

Ponowne wdrożenie programu

Po wprowadzeniu tych zmian ponownie wdroż zaktualizowany program. To zapewnia, że zmodyfikowany program będzie dostępny do testowania. Na Solanie aktualizacja programu wymaga jedynie wdrożenia programu pod tym samym ID programu.

$ 
deploy
Copy
[show output]

Zaktualizuj plik testowy

Następnie zaktualizuj plik anchor.test.ts, aby uwzględnić nowe konto skarbca w instrukcjach. Wymaga to wyprowadzenia PDA skarbca i uwzględnienia go w wywołaniach instrukcji aktualizacji i usunięcia.

Wyprowadź Vault PDA

Najpierw dodaj wyprowadzenie Vault PDA:

const [vaultPda, vaultBump] = PublicKey.findProgramAddressSync(
  [Buffer.from("vault"), wallet.publicKey.toBuffer()],
  program.programId
);

Zmień test aktualizacji

const transactionSignature = await program.methods
  .update(message)
  .accounts({
    messageAccount: messagePda,
    vaultAccount: vaultPda
  })
  .rpc({ commitment: "confirmed" });

Zmień test usuwania

const transactionSignature = await program.methods
  .delete()
  .accounts({
    messageAccount: messagePda,
    vaultAccount: vaultPda
  })
  .rpc({ commitment: "confirmed" });

Uruchom test ponownie

Po wprowadzeniu tych zmian uruchom testy, aby upewnić się, że wszystko działa zgodnie z oczekiwaniami:

$ 
test
Copy
[show output]

Możesz następnie sprawdzić linki SolanaFM, aby zobaczyć szczegóły transakcji, gdzie znajdziesz CPI dla instrukcji transferu w ramach instrukcji aktualizacji i usuwania.

Aktualizacja CPI

Usunięcie CPI

Jeśli napotkasz jakiekolwiek błędy, możesz odwołać się do ostatecznego kodu.