From e59be5879b9bc79cadbad5987be28a2df4226fa9 Mon Sep 17 00:00:00 2001
From: XiZi <549384715@qq.com>
Date: Fri, 4 Oct 2024 13:21:57 +0800
Subject: [PATCH] add: getComputeUnitPrice

---
 src/lib/transaction.ts | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/src/lib/transaction.ts b/src/lib/transaction.ts
index 3fdd01c..b557604 100644
--- a/src/lib/transaction.ts
+++ b/src/lib/transaction.ts
@@ -54,4 +54,23 @@ export const getSimulationComputeUnits = async (
 
   getErrorFromRPCResponse(rpcResponse);
   return rpcResponse.value.unitsConsumed || null;
-};
\ No newline at end of file
+};
+
+export async function getComputeUnitPrice(connection: Connection, ignoreZeroMembers = 1 / 2) {
+  let prioritizationFees = await connection.getRecentPrioritizationFees();
+  let length = prioritizationFees.length;
+  let prioritizationZeros = 0, prioritizationTotal = 0;
+  for (let i = 0; i < length; i++) {
+    if (prioritizationFees[i].prioritizationFee === 0) {
+      prioritizationZeros++;
+    } else {
+      prioritizationTotal += prioritizationFees[i].prioritizationFee;
+    }
+  }
+  if (prioritizationZeros >= length * ignoreZeroMembers) {
+    return 0;
+  }
+  return Math.ceil(
+    prioritizationTotal / (length - prioritizationZeros),
+  );
+}