add params size to quote

massalabs · Oct 18, 2024 · 5bb0912 · 5bb0912
1 parent 147c349
commit 5bb0912
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/assembly/env/env.ts b/assembly/env/env.ts
@@ -231,6 +231,7 @@ export namespace env {
     ascPeriod: u64,
     ascThread: u8,
     maxGas: u64,
+    paramsSize: u64,
   ): u64;
 
   @external("massa", "assembly_script_deferred_call_register")

diff --git a/assembly/std/deferredCalls.ts b/assembly/std/deferredCalls.ts
@@ -14,8 +14,8 @@ import { Slot } from './context';
  * @returns Mas amount
  *
  */
-export function deferredCallQuote(targetSlot: Slot, maxGas: u64): u64 {
-  return env.deferredCallQuote(targetSlot.period, targetSlot.thread, maxGas);
+export function deferredCallQuote(targetSlot: Slot, maxGas: u64, paramsSize: u64): u64 {
+  return env.deferredCallQuote(targetSlot.period, targetSlot.thread, maxGas, paramsSize);
 }
 
 /**
@@ -84,17 +84,19 @@ export function findCheapestSlot(
   startPeriod: u64,
   endPeriod: u64,
   maxGas: u64,
+  paramsSize: u64,
 ): Slot {
   let cheapestSlotPeriod: u64 = startPeriod;
   let cheapestSlotThread: u8 = 0;
   let cheapestSlotPrice: u64 = deferredCallQuote(
     new Slot(startPeriod, 0),
     maxGas,
+    paramsSize
   );
 
   for (let period = startPeriod; period <= endPeriod; period++) {
     for (let thread: u8 = 1; thread < 32; thread++) {
-      const price = deferredCallQuote(new Slot(period, thread), maxGas);
+      const price = deferredCallQuote(new Slot(period, thread), maxGas, paramsSize);
       if (price < cheapestSlotPrice) {
         cheapestSlotPrice = price;
         cheapestSlotPeriod = period;