Skip to content

Commit

Permalink
feat(API): Split part of cron-cache-gas-prices into cron-cache-gas-costs
Browse files Browse the repository at this point in the history
Updating the gas costs and gas prices in the same cron job makes the cache resets slower due to the memory constraints of a single file, and when updating the gas price every 5s, this makes a difference.

This PR splits the gas cost updates, which are 10s and 30s for l1 data fees and native gas costs, separate from the gas price updates so that the gas price updates can be closer to 5s. Currently its anywhere from 5s to 10s.

Additionally, this PR adds try-catches around the `await cache.set()` calls so that the a failure due to an upstream RPC call (e.g. I've seen this happen with linea_estimateGas calls a lot) doesn't take down the cron job for the full run.
  • Loading branch information
nicholaspai committed Jan 14, 2025
1 parent 49f146c commit 2a61bc9
Show file tree
Hide file tree
Showing 3 changed files with 232 additions and 99 deletions.
209 changes: 209 additions & 0 deletions api/cron-cache-gas-costs.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,209 @@
import { VercelResponse } from "@vercel/node";
import { TypedVercelRequest } from "./_types";
import {
HUB_POOL_CHAIN_ID,
getCachedNativeGasCost,
getCachedOpStackL1DataFee,
getLogger,
handleErrorCondition,
resolveVercelEndpoint,
} from "./_utils";
import { UnauthorizedError } from "./_errors";

import mainnetChains from "../src/data/chains_1.json";
import { utils, constants } from "@across-protocol/sdk";
import { DEFAULT_SIMULATED_RECIPIENT_ADDRESS } from "./_constants";
import axios from "axios";
import { ethers } from "ethers";

type Route = {
originChainId: number;
originToken: string;
destinationChainId: number;
destinationToken: string;
originTokenSymbol: string;
destinationTokenSymbol: string;
};

// Set lower than TTL in getCachedOpStackL1DataFee
// Set lower than the L1 block time so we can try to get as up to date L1 data fees based on L1 base fees as possible.
const updateL1DataFeeIntervalsSecPerChain = {
default: 10,
};

// Set lower than TTL in getCachedNativeGasCost. This should rarely change so we should just make sure
// we keep this cache warm.
const updateNativeGasCostIntervalsSecPerChain = {
default: 30,
};

// Force the cache update promises to stop 1s before the Vercel serverless function times out.
const maxDurationSec = 60;

const getDepositArgsForChainId = (chainId: number, tokenAddress: string) => {
return {
amount: ethers.BigNumber.from(100),
inputToken: constants.ZERO_ADDRESS,
outputToken: tokenAddress,
recipientAddress: DEFAULT_SIMULATED_RECIPIENT_ADDRESS,
originChainId: 0, // Shouldn't matter for simulation
destinationChainId: Number(chainId),
};
};

const handler = async (
request: TypedVercelRequest<Record<string, never>>,
response: VercelResponse
) => {
const logger = getLogger();
logger.debug({
at: "CronCacheGasPrices",
message: "Starting cron job...",
});
try {
const authHeader = request.headers?.["authorization"];
if (
!process.env.CRON_SECRET ||
authHeader !== `Bearer ${process.env.CRON_SECRET}`
) {
throw new UnauthorizedError();
}

// Skip cron job on testnet
if (HUB_POOL_CHAIN_ID !== 1) {
logger.info({
at: "CronCacheGasPrices",
message: "Skipping cron job on testnet",
});
return;
}

const availableRoutes = (
await axios(`${resolveVercelEndpoint()}/api/available-routes`)
).data as Array<Route>;

// This marks the timestamp when the function started
const functionStart = Date.now();

/**
* @notice Updates the L1 data fee gas cost cache every `updateL1DataFeeIntervalsSecPerChain` seconds
* up to `maxDurationSec` seconds.
* @param chainId Chain to estimate l1 data fee for
* @param outputTokenAddress This output token will be used to construct a fill transaction to simulate
* gas costs for.
*/
const updateL1DataFeePromise = async (
chainId: number,
outputTokenAddress: string
): Promise<void> => {
const secondsPerUpdate = updateL1DataFeeIntervalsSecPerChain.default;
const depositArgs = getDepositArgsForChainId(chainId, outputTokenAddress);
const gasCostCache = getCachedNativeGasCost(depositArgs);

while (true) {
const diff = Date.now() - functionStart;
// Stop after `maxDurationSec` seconds
if (diff >= maxDurationSec * 1000) {
break;
}
const gasCost = await gasCostCache.get();
if (utils.chainIsOPStack(chainId)) {
const cache = getCachedOpStackL1DataFee(depositArgs, gasCost);
try {
await cache.set();
} catch (err) {
logger.warn({
at: "CronCacheGasPrices#updateL1DataFeePromise",
message: `Failed to set l1 data fee cache for chain ${chainId}`,
depositArgs,
gasCost,
error: err,
});
}
}
await utils.delay(secondsPerUpdate);
}
};

/**
* @notice Updates the native gas cost cache every `updateNativeGasCostIntervalsSecPerChain` seconds
* up to `maxDurationSec` seconds.
* @param chainId Chain to estimate gas cost for
* @param outputTokenAddress This output token will be used to construct a fill transaction to simulate
* gas costs for.
*/
const updateNativeGasCostPromise = async (
chainId: number,
outputTokenAddress: string
): Promise<void> => {
const secondsPerUpdate = updateNativeGasCostIntervalsSecPerChain.default;
const depositArgs = getDepositArgsForChainId(chainId, outputTokenAddress);
const cache = getCachedNativeGasCost(depositArgs);

while (true) {
const diff = Date.now() - functionStart;
// Stop after `maxDurationSec` seconds
if (diff >= maxDurationSec * 1000) {
break;
}
try {
await cache.set();
} catch (err) {
logger.warn({
at: "CronCacheGasPrices#updateNativeGasCostPromise",
message: `Failed to set native gas cost cache for chain ${chainId}`,
depositArgs,
error: err,
});
}
await utils.delay(secondsPerUpdate);
}
};

// The minimum interval for Vercel Serverless Functions cron jobs is 1 minute.
// But we want to update gas data more frequently than that.
// To circumvent this, we run the function in a loop and update gas prices every
// `secondsPerUpdateForChain` seconds and stop after `maxDurationSec` seconds (1 minute).
const cacheUpdatePromises = Promise.all([
Promise.all(
mainnetChains.map(async (chain) => {
const routesToChain = availableRoutes.filter(
({ destinationChainId }) => destinationChainId === chain.chainId
);
const outputTokensForChain = routesToChain.map(
({ destinationToken }) => destinationToken
);
await Promise.all([
Promise.all(
outputTokensForChain.map((outputToken) =>
updateNativeGasCostPromise(chain.chainId, outputToken)
)
),
Promise.all(
outputTokensForChain.map((outputToken) =>
updateL1DataFeePromise(chain.chainId, outputToken)
)
),
]);
})
),
]);
await Promise.race([cacheUpdatePromises, utils.delay(maxDurationSec)]);

logger.debug({
at: "CronCacheGasPrices",
message: "Finished",
});
response.status(200);
response.send("OK");
} catch (error: unknown) {
return handleErrorCondition(
"cron-cache-gas-prices",
response,
logger,
error
);
}
};

export default handler;
115 changes: 16 additions & 99 deletions api/cron-cache-gas-prices.ts
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,6 @@ import { VercelResponse } from "@vercel/node";
import { TypedVercelRequest } from "./_types";
import {
HUB_POOL_CHAIN_ID,
getCachedNativeGasCost,
getCachedOpStackL1DataFee,
getLogger,
handleErrorCondition,
latestGasPriceCache,
Expand Down Expand Up @@ -31,18 +29,7 @@ const updateIntervalsSecPerChain = {
default: 5,
};

// Set lower than TTL in getCachedOpStackL1DataFee
// Set lower than the L1 block time so we can try to get as up to date L1 data fees based on L1 base fees as possible.
const updateL1DataFeeIntervalsSecPerChain = {
default: 10,
};

// Set lower than TTL in getCachedNativeGasCost. This should rarely change so we should just make sure
// we keep this cache warm.
const updateNativeGasCostIntervalsSecPerChain = {
default: 30,
};

// Force the cache update promises to stop 1s before the Vercel serverless function times out.
const maxDurationSec = 60;

const getDepositArgsForChainId = (chainId: number, tokenAddress: string) => {
Expand Down Expand Up @@ -101,77 +88,28 @@ const handler = async (
outputTokenAddress?: string
): Promise<void> => {
const secondsPerUpdateForChain = updateIntervalsSecPerChain.default;
const cache = latestGasPriceCache(
chainId,
outputTokenAddress
? getDepositArgsForChainId(chainId, outputTokenAddress)
: undefined
);

while (true) {
const diff = Date.now() - functionStart;
// Stop after `maxDurationSec` seconds
if (diff >= maxDurationSec * 1000) {
break;
}
await cache.set();
await utils.delay(secondsPerUpdateForChain);
}
};

/**
* @notice Updates the L1 data fee gas cost cache every `updateL1DataFeeIntervalsSecPerChain` seconds
* up to `maxDurationSec` seconds.
* @param chainId Chain to estimate l1 data fee for
* @param outputTokenAddress This output token will be used to construct a fill transaction to simulate
* gas costs for.
*/
const updateL1DataFeePromise = async (
chainId: number,
outputTokenAddress: string
): Promise<void> => {
const secondsPerUpdate = updateL1DataFeeIntervalsSecPerChain.default;
const depositArgs = getDepositArgsForChainId(chainId, outputTokenAddress);
const gasCostCache = getCachedNativeGasCost(depositArgs);
const depositArgs = outputTokenAddress
? getDepositArgsForChainId(chainId, outputTokenAddress)
: undefined;
const cache = latestGasPriceCache(chainId, depositArgs);

while (true) {
const diff = Date.now() - functionStart;
// Stop after `maxDurationSec` seconds
if (diff >= maxDurationSec * 1000) {
break;
}
const gasCost = await gasCostCache.get();
if (utils.chainIsOPStack(chainId)) {
const cache = getCachedOpStackL1DataFee(depositArgs, gasCost);
try {
await cache.set();
} catch (err) {
logger.warn({
at: "CronCacheGasPrices#updateGasPricePromise",
message: `Failed to set gas price cache for chain ${chainId}`,
depositArgs,
error: err,
});
}
await utils.delay(secondsPerUpdate);
}
};

/**
* @notice Updates the native gas cost cache every `updateNativeGasCostIntervalsSecPerChain` seconds
* up to `maxDurationSec` seconds.
* @param chainId Chain to estimate gas cost for
* @param outputTokenAddress This output token will be used to construct a fill transaction to simulate
* gas costs for.
*/
const updateNativeGasCostPromise = async (
chainId: number,
outputTokenAddress: string
): Promise<void> => {
const secondsPerUpdate = updateNativeGasCostIntervalsSecPerChain.default;
const depositArgs = getDepositArgsForChainId(chainId, outputTokenAddress);
const cache = getCachedNativeGasCost(depositArgs);

while (true) {
const diff = Date.now() - functionStart;
// Stop after `maxDurationSec` seconds
if (diff >= maxDurationSec * 1000) {
break;
}
await cache.set();
await utils.delay(secondsPerUpdate);
await utils.delay(secondsPerUpdateForChain);
}
};

Expand All @@ -182,7 +120,7 @@ const handler = async (
// But we want to update gas data more frequently than that.
// To circumvent this, we run the function in a loop and update gas prices every
// `secondsPerUpdateForChain` seconds and stop after `maxDurationSec` seconds (1 minute).
await Promise.all([
const cacheUpdatePromises = Promise.all([
// @dev Linea gas prices are dependent on the L2 calldata to be submitted so compute one gas price for each output token,
// so we compute one gas price per output token for Linea
Promise.all(
Expand All @@ -195,29 +133,8 @@ const handler = async (
updateGasPricePromise(CHAIN_IDs.LINEA, destinationToken)
)
),
Promise.all(
mainnetChains.map(async (chain) => {
const routesToChain = availableRoutes.filter(
({ destinationChainId }) => destinationChainId === chain.chainId
);
const outputTokensForChain = routesToChain.map(
({ destinationToken }) => destinationToken
);
await Promise.all([
Promise.all(
outputTokensForChain.map((outputToken) =>
updateNativeGasCostPromise(chain.chainId, outputToken)
)
),
Promise.all(
outputTokensForChain.map((outputToken) =>
updateL1DataFeePromise(chain.chainId, outputToken)
)
),
]);
})
),
]);
await Promise.race([cacheUpdatePromises, utils.delay(maxDurationSec)]);

logger.debug({
at: "CronCacheGasPrices",
Expand Down
7 changes: 7 additions & 0 deletions vercel.json
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,10 @@
"path": "/api/cron-cache-gas-prices",
"schedule": "* * * * *"
},
{
"path": "/api/cron-cache-gas-costs",
"schedule": "* * * * *"
},
{
"path": "/api/cron-ping-endpoints",
"schedule": "* * * * *"
Expand All @@ -18,6 +22,9 @@
"api/cron-cache-gas-prices.ts": {
"maxDuration": 90
},
"api/cron-cache-gas-costs.ts": {
"maxDuration": 90
},
"api/cron-ping-endpoints.ts": {
"maxDuration": 90
}
Expand Down

0 comments on commit 2a61bc9

Please sign in to comment.