使用Solana、Jupiter工具获取关联钱包代币余额失败排查

问题背景 #

基于Solana、Jupiter Aggregator及Jupiter APIs实现代币兑换流程时，已完成服务端API的报价获取、兑换指令生成、平台手续费划转等逻辑，但无法正确获取关联钱包的特定代币余额。使用@solana/web3.js的getTokenAccountBalance和getAssociatedTokenAddress方法实现，已确认函数调用及参数传递正常，但结果始终异常。

相关代码片段 #

服务端server.js的余额查询接口

app.get("/fetchBalance", async (req, res) => {
  const { address, walletAddress } = req.query;
  try {
    const tokenPublicKey = new PublicKey(address);
    const walletPublicKey = new PublicKey(walletAddress);

    const balance = await connection.getTokenAccountBalance(
      await connection.getAssociatedTokenAddress(tokenPublicKey, walletPublicKey)
    );

    return res.status(200).json({
      balance: balance.value.uiAmount,
      decimals: balance.value.decimals
    });
  } catch (error) {
    console.error("Error fetching token balance:", error);
    return res.status(500).json({ error: "Error fetching token balance" });
  }
});

前端swap.js的余额获取函数

const fetchTokenBalance = useCallback(async (tokenAddress) => {
  if (!publicKey) return;

  try {
    const response = await axios.get("http://localhost:3001/fetchBalance", {
      params: {
        address: tokenAddress,
        walletAddress: publicKey.toString(),
      },
    });
    const balance = response.data.balance;
    setBalances(prevBalances => ({ ...prevBalances, [tokenAddress]: balance }));
    updateTokenBalances({ ...balances, [tokenAddress]: balance });
  } catch (error) {
    console.error("Error fetching token balance:", error);
    messageApi.error("Failed to fetch token balance");
  }
}, [publicKey, updateTokenBalances, balances, messageApi]);

服务端server.js的价格查询接口

app.get("/tokenPrice", async (req, res) => {
  const { addressOne, addressTwo } = req.query;
  try {
    const response = await axios.get(`https://price.jup.ag/v6/price`, {
      params: { ids: addressOne, vsToken: addressTwo },
    });
    const priceData = response.data.data[addressOne];
    if (!priceData || !priceData.price) {
      throw new Error("Invalid response from price API");
    }
    const usdPrices = {
      tokenOne: priceData.price,
      tokenTwo: 1 / priceData.price,
      ratio: priceData.price,
    };
    return res.status(200).json(usdPrices);
  } catch (error) {
    console.error("Error fetching token prices:", error);
    return res.status(500).json({ error: "Error fetching token prices" });
  }
});

可能原因与修复方案 #

1. 关联代币账户未创建 #

Solana中，钱包首次接收某代币时才会自动创建关联代币账户，直接通过getAssociatedTokenAddress获取地址后查询余额，会因账户不存在报错或返回无效值。
修复方案：
先查询钱包下是否存在该代币的有效账户，不存在则返回0余额（或提示用户先接收代币）；也可预创建关联账户（需用户签名）。修改服务端接口如下：

app.get("/fetchBalance", async (req, res) => {
  const { address, walletAddress } = req.query;
  try {
    const tokenPublicKey = new PublicKey(address);
    const walletPublicKey = new PublicKey(walletAddress);

    // 查询钱包下所有该代币的账户
    const tokenAccounts = await connection.getTokenAccountsByOwner(
      walletPublicKey,
      { mint: tokenPublicKey }
    );

    if (tokenAccounts.value.length === 0) {
      // 查询代币本身的decimal值，返回更准确的初始状态
      const mintInfo = await connection.getAccountInfo(tokenPublicKey);
      const decimals = mintInfo ? Mint.fromAccountInfo(mintInfo).decimals : 0;
      return res.status(200).json({ balance: 0, decimals });
    }

    // 使用第一个有效账户查询余额
    const balance = await connection.getTokenAccountBalance(tokenAccounts.value[0].pubkey);
    return res.status(200).json({
      balance: balance.value.uiAmount,
      decimals: balance.value.decimals
    });
  } catch (error) {
    console.error("Error fetching token balance:", error);
    return res.status(500).json({ error: "Error fetching token balance" });
  }
});

2. RPC节点同步延迟或配置错误 #

使用的Solana RPC节点可能未同步到最新区块，或节点本身不可靠，导致余额查询结果不准确。
修复方案：
切换到官方或稳定的RPC节点（如主网https://api.mainnet-beta.solana.com），同时在连接时设置合理的超时和重试机制：

// 初始化connection时配置
const connection = new Connection("https://api.mainnet-beta.solana.com", {
  commitment: "confirmed",
  timeout: 10000,
});

3. 前端状态更新闭包问题 #

前端fetchTokenBalance中依赖balances状态更新时，因闭包特性可能获取到旧值，导致状态更新异常。
修复方案：
移除balances依赖，仅使用prevBalances更新状态：

const fetchTokenBalance = useCallback(async (tokenAddress) => {
  if (!publicKey) return;

  try {
    const response = await axios.get("http://localhost:3001/fetchBalance", {
      params: {
        address: tokenAddress,
        walletAddress: publicKey.toString(),
      },
    });
    const balance = response.data.balance;
    setBalances(prevBalances => ({ ...prevBalances, [tokenAddress]: balance }));
    // 全局状态更新也依赖prev值
    updateTokenBalances(prev => ({ ...prev, [tokenAddress]: balance }));
  } catch (error) {
    console.error("Error fetching token balance:", error);
    messageApi.error("Failed to fetch token balance");
  }
}, [publicKey, updateTokenBalances, messageApi]); // 移除balances依赖

4. 代币Decimal值处理缺失 #

若代币账户未创建，原逻辑返回的decimals为0，会导致后续金额格式化错误。
修复方案：
通过getMint查询代币合约的Decimal值，确保返回正确的精度信息（已包含在第一个修复方案的代码中）。

代币余额在兑换流程中的作用 #

• 校验交易可行性：确保用户钱包有足够的输入代币余额（包含Gas费），避免交易因余额不足失败。
• 限制兑换额度：根据用户可用余额，自动限制最大兑换数量，防止用户输入超出实际拥有的代币量。
• 用户资产展示：实时显示钱包内的代币数量，辅助用户决策兑换金额。
• 交易后状态同步：兑换完成后更新余额，展示最新的资产情况，提升用户体验。

内容的提问来源于stack exchange，提问作者Osman Ghani Granday