@certik/skynet 0.24.0 → 0.25.0

package/src/databricks.ts

@@ -1,82 +0,0 @@
-import { DBSQLClient, DBSQLParameter } from "@databricks/sql";
-import { getEnvOrThrow } from "./env.js";
-import { exponentialRetry } from "./availability.js";
-import type IDBSQLSession from "@databricks/sql/dist/contracts/IDBSQLSession.js";
-import type { DBSQLParameterValue } from "@databricks/sql/dist/DBSQLParameter.js";
-
-let client: DBSQLClient | null = null;
-
-let session: IDBSQLSession | null = null;
-
-async function initSession() {
-  if (!client) {
-    client = new DBSQLClient();
-    await client.connect({
-      authType: "access-token",
-      token: getEnvOrThrow("SKYNET_DATABRICKS_TOKEN"),
-      host: getEnvOrThrow("SKYNET_DATABRICKS_SERVER_HOSTNAME"),
-      path: getEnvOrThrow("SKYNET_DATABRICKS_HTTP_PATH"),
-    });
-  }
-  if (!session) {
-    session = await client.openSession();
-  }
-}
-
-async function executeSql<T>(sql: string, bindings: Record<string, DBSQLParameterValue>) {
-  // retry sql query to avoid socket hang up error
-  const results = await exponentialRetry(
-    async () => {
-      let queryOperation;
-      try {
-        await initSession();
-        if (!session) throw new Error("Session is not initialized");
-        queryOperation = await session.executeStatement(sql, {
-          runAsync: true,
-          namedParameters: objToParameter(bindings),
-        });
-
-        const result = await queryOperation.fetchAll();
-        return result;
-      } catch (err) {
-        // If the error is not a Databricks error, throw it
-        if (!checkIsDatabricksError(err)) {
-          throw err;
-        }
-        return err;
-      } finally {
-        await queryOperation?.close();
-      }
-    },
-    {
-      maxRetry: 3,
-      initialDuration: 500,
-      test: (result) => {
-        return !checkIsDatabricksError(result);
-      },
-    },
-  );
-
-  if (checkIsDatabricksError(results)) {
-    // still error after retry, throw it
-    throw results;
-  }
-  return results as T[];
-}
-
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-function checkIsDatabricksError(err: any) {
-  return err?.errno === "ECONNRESET" || err?.message?.includes("java.lang.NullPointerException");
-}
-
-function objToParameter(obj: Record<string, DBSQLParameterValue>) {
-  return Object.entries(obj).reduce(
-    (acc, [key, value]) => {
-      acc[key] = new DBSQLParameter({ value });
-      return acc;
-    },
-    {} as Record<string, DBSQLParameter>,
-  );
-}
-
-export { executeSql };