Cross Program Invocation

Cập nhật lệnh Update

Đầu tiên, chương trình cần một cơ chế "trả tiền để cập nhật" đơn giản bằng cách thay đổi cấu trúc Update và hàm update.

Bắt đầu bằng việc cập nhật tệp lib.rs để đưa vào phạm vi các mục từ module system_program.

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

Tiếp theo, cập nhật cấu trúc Update để bao gồm một tài khoản mới gọi là vault_account. Tài khoản này, được kiểm soát bởi program, nhận SOL từ người dùng khi họ cập nhật tài khoản tin nhắn của mình.

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

Tiếp theo, thêm logic CPI trong lệnh update để chuyển 0.001 SOL từ tài khoản của người dùng đến tài khoản kho.

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)?;

Xây dựng lại chương trình.

$ 
buildCopy

Cập nhật lệnh xóa

Bây giờ hãy thêm cơ chế "hoàn tiền khi xóa" bằng cách thay đổi cấu trúc Delete và hàm delete.

Đầu tiên, cập nhật cấu trúc Delete để bao gồm vault_account. Điều này cho phép chuyển bất kỳ SOL nào trong vault trở lại cho người dùng khi họ đóng tài khoản tin nhắn của họ.

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

Đồng thời thêm system_program vì CPI cho việc chuyển tiền yêu cầu gọi System Program.

pub system_program: Program<'info, System>,

Tiếp theo, thêm logic CPI trong lệnh delete để chuyển SOL từ tài khoản vault trở lại tài khoản của người dùng.

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())?;

Lưu ý rằng _ctx: Context<Delete> thay đổi thành ctx: Context<Delete> để sử dụng context trong phần thân của hàm.

Xây dựng lại chương trình.

$ 
buildCopy

Triển khai lại chương trình

Sau khi thực hiện những thay đổi này, hãy triển khai lại chương trình đã cập nhật. Điều này đảm bảo chương trình đã sửa đổi sẽ khả dụng để kiểm tra. Trên Solana, việc cập nhật chương trình chỉ đơn giản là triển khai chương trình với cùng một program ID.

$ 
deploy
Copy
[show output]

Cập nhật tệp kiểm tra

Tiếp theo, cập nhật tệp anchor.test.ts để bao gồm tài khoản vault mới trong các lệnh. Điều này yêu cầu phải tạo vault PDA và bao gồm nó trong các lệnh gọi cập nhật và xóa.

Tạo Vault PDA

Đầu tiên, thêm phần tạo vault PDA:

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

Thay đổi Update Test

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

Thay đổi Delete Test

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

Chạy lại Test

Sau khi thực hiện những thay đổi này, chạy các bài test để đảm bảo mọi thứ hoạt động như mong đợi:

$ 
test
Copy
[show output]

Sau đó bạn có thể kiểm tra các liên kết SolanaFM để xem chi tiết giao dịch, nơi bạn sẽ tìm thấy các CPI cho các lệnh chuyển tiền trong các lệnh cập nhật và xóa.

CPI Cập nhật

CPI Xóa

Nếu bạn gặp bất kỳ lỗi nào, bạn có thể tham khảo mã cuối cùng.