From fa941174c2221484d62a5752081391bd84687184 Mon Sep 17 00:00:00 2001
From: wei <1415121722@qq.com>
Date: Wed, 18 Jun 2025 00:20:13 +0800
Subject: [PATCH] =?UTF-8?q?feat=EF=BC=9Acheck=20if=20the=20account=20exist?=
 =?UTF-8?q?s=20before=20selling?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/trading/protocols/pumpfun.rs | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/src/trading/protocols/pumpfun.rs b/src/trading/protocols/pumpfun.rs
index cce81ba..27320f2 100644
--- a/src/trading/protocols/pumpfun.rs
+++ b/src/trading/protocols/pumpfun.rs
@@ -125,6 +125,16 @@ impl InstructionBuilder for PumpFunInstructionBuilder {
         let creator_vault_pda = get_creator_vault_pda(&params.creator).unwrap();
         let ata = get_associated_token_address(&params.payer.pubkey(), &params.mint);
 
+        // 检查账户是否存在
+        if let Some(rpc) = &params.rpc {
+            let account_info = rpc.get_account(&ata).await;
+            if account_info.is_err() {
+                return Err(anyhow!("account not exists"));
+            }
+        } else {
+            return Err(anyhow!("RPC client is required to check account existence"));
+        }
+
         // 获取代币余额
         let balance_u64 = if let Some(rpc) = &params.rpc {
             let balance = rpc.get_token_account_balance(&ata).await?;