Lakebase Postgres を Databricks ワークスペースの SQL エディターで操作する手順

概要

Lakebase に対して、Databricks Lakehouse の SQL エディター経由で SQL を実行する方法を紹介します。下記ドキュメントで紹介されている機能を、実際に試した手順です。

• ワークスペースの SQL エディターからのクエリ - Azure Databricks | Microsoft Learn

Lakebase Postgres アプリには SQL エディター機能が用意されていますが、従来の Databricks 開発体験とは異なります。

出所: Lakebase SQL エディターからのクエリ - Azure Databricks | Microsoft Learn

本記事の手順を実施することで、Lakebase に対しても Databricks の SQL エディターで開発できるようになります。

実施手順

1. Lakehouse アプリの SQL エディター で SQL ファイルを作成

2. 右上のコンピュート選択アイコンから その他 を選択

3. Lakebase Postgres にチェックを入れ、対象のプロジェクトを選択したうえで アタッチ を選択

4. SQL を記述して実行

-- 1. スキーマ作成
CREATE SCHEMA IF NOT EXISTS qiita_01;
-- 2. テーブル作成
CREATE TABLE IF NOT EXISTS qiita_01.test_table_01 (
    id SERIAL PRIMARY KEY,
    name TEXT NOT NULL,
    value REAL,
    created_at TIMESTAMPTZ NOT NULL DEFAULT CURRENT_TIMESTAMP
  );
-- 3. サンプルデータ投入
INSERT INTO qiita_01.test_table_01 (name, value)
  SELECT
    LEFT(md5(i::TEXT), 10),
    random()
  FROM
    generate_series(1, 10) AS s(i);
-- 4. データ確認
SELECT
  id,
  name,
  value,
  created_at
FROM
  qiita_01.test_table_01
ORDER BY
  id;

5. Lakebase Postgres アプリにてテーブルが作成されていることを確認