jobs.mjs

// Retrieve Job-defined env vars
const { CLOUD_RUN_TASK_INDEX = 0, CLOUD_RUN_TASK_ATTEMPT = 0 } = process.env;
// Retrieve User-defined env vars
const { JOB_NAME, CONTINUOUS_JOB } = process.env;

import { JOBS } from "./variables.mjs";
import { cacheCoingeckoTickersJob } from "./jobs/cache_coingecko_tickers_job.mjs";
import { cacheHydrationWebStatsJob } from "./jobs/cache_hydration-web_stats_job.mjs";
import { cacheHydradxUiStatsTvlJob } from "./jobs/cache_hydradx-ui_stats_tvl_job.mjs";
import { cacheHydradxUiV2StatsVolumeJob } from "./jobs/cache_hydradx-ui_v2_stats_volume_job.mjs";
import { cacheCoinmarketcapSummaryJob } from "./jobs/cache_coinmarketcap_summary_job.mjs";
import { newSqlClient } from "./clients/sql.mjs";
import { newRedisClient } from "./clients/redis.mjs";

const main = async () => {
  console.log(
    `Cloud Run Job ${JOB_NAME} started (task index #${CLOUD_RUN_TASK_INDEX}, attempt #${CLOUD_RUN_TASK_ATTEMPT})`
  );
  console.log(`CONTINUOUS_JOB: ${isContinuousJob()}`);

  await executeJob(JOB_NAME);

  console.log(
    `Cloud Run Job ${JOB_NAME} completed (task index #${CLOUD_RUN_TASK_INDEX})`
  );

  process.exit();
};

async function executeJob(job_name) {
  console.log(`Executing ${job_name}..`);

  const sqlClient = await newSqlClient();
  const redisClient = await newRedisClient();

  switch (job_name) {
    case JOBS["cacheCoingeckoTickersJob"]: {
      await cacheCoingeckoTickersJob(sqlClient, redisClient);
      break;
    }
    case JOBS["cacheHydrationWebStatsJob"]: {
      await cacheHydrationWebStatsJob(sqlClient, redisClient);
      break;
    }
    case JOBS["cacheHydradxUiStatsTvlJob"]: {
      await cacheHydradxUiStatsTvlJob(sqlClient, redisClient);
      break;
    }
    case JOBS["cacheHydradxUiV2StatsVolumeJob"]: {
      await cacheHydradxUiV2StatsVolumeJob(sqlClient, redisClient);
      break;
    }
    case JOBS["cacheCoinmarketcapSummaryJob"]: {
      await cacheCoinmarketcapSummaryJob(sqlClient, redisClient);
      break;
    }
    default: {
      throw new Error(`Job not found: ${JOB_NAME}`);
    }
  }

  console.log(`Executed ${job_name}`);

  if (CONTINUOUS_JOB == "true") {
    return executeJob(job_name);
  } else {
    await sqlClient.release();
    await redisClient.disconnect();

    return true;
  }
}

function isContinuousJob() {
  return CONTINUOUS_JOB == "true";
}

// Start script
main().catch((err) => {
  console.error(err);
  process.exit(1);
});