Connect Discord messages and Databricks in our serverless environment

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

    async fetch(publish, done) {
        // TODO: Customize your discord channel id
        // See how to find channel id at:
        // https://support.discord.com/hc/en-us/articles/206346498-Where-can-I-find-my-User-Server-Message-ID-#:~:text=On%20Android%20press%20and%20hold,name%20and%20select%20Copy%20ID.
        const channelId = "your-discord-channel-id";
        const channel = await this.discordClient.channels.fetch(channelId);
        const messages = await channel.messages.fetch({
            limit: 10
        });

        for (const message of messages) {
            await publish(message);
        }
        done();
    }

    async close() {
        this.discordClient.destroy();
    }
}

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();
    }
}