Connect MySQL and Databricks in our serverless environment

class MysqlSourceSelect {
    async init() {
        // TODO: Create your mysql credential
        // More info at https://yepcode.io/docs/integrations/mysql/#credential-configuration
        this.mysql = yepcode.integration.mysql("your-mysql-credential-name");
    }

    async fetch(publish, done) {
        this.mysql
            // TODO: Customize your sql query
            .query(`SELECT * FROM your_table_name`)
            .stream()
            .on("data", async (row) => {
                // TODO: map your row to an item
                const item = {
                    ...row
                };

                await publish(item);
            })
            .on("error", console.error)
            .on("end", done);
    }

    async close() {
        await this.mysql.end();
    }
}

class DatabricksTargetInsert {
    async init() {
        // TODO: Create your databricks credential
        // More info at https://yepcode.io/docs/integrations/databricks/#credential-configuration
        this.databricks = yepcode.integration.databricks(
            "your-databricks-credential-name"
        );
        this.session = await this.databricks.openSession();
    }

    async consume(item) {
        // TODO: Customize your sql query
        // More info at: https://github.com/databricks/databricks-sql-nodejs/blob/main/tests/e2e/data_types.test.js#L151
        const operation = await this.session.executeStatement(
            `INSERT INTO yourTable VALUES(${item.id}, ${item.name})`
        );
        const result = await operation.fetchAll();
        await operation.close();
        return result;
    }

    async close() {
        await this.session.close();
        await this.databricks.close();
    }
}