DatabricksのIdentity列を試してみる

こちらで紹介されている新機能、Identity列を試してみました。

Databricks SQLでテーブルを作成して、データをインサートしていきます。

Databricks SQLにアクセス

Databricksワークスペースにログインし、サイドメニューのペルソナスイッチャーでSQLを選択します。

テーブルの作成

1. サイドメニューからクエリを選択します。
   

2. 画面右上のクエリーを作成をクリックします。
   

3. クエリーエディタが表示されます。
   

4. 以下のクエリーを入力します。USEで指定するデータベースは適宜変更してください。

   SQL

   USE `20210826_workshop_takaakiyayoidatabrickscom`;
   
   CREATE OR REPLACE TABLE demo (
     id BIGINT GENERATED ALWAYS AS IDENTITY,
     product_type STRING,
     sales BIGINT
   );
   

5. すべてを実行をクリックしてテーブルを作成します。クエリーが成功するとdemoテーブルが作成されます。
   

6. この時点ではテーブルは空です。
   

行のインサート

1. ダミーの行をインサートします。

   SQL

   USE `20210826_workshop_takaakiyayoidatabrickscom`;
   INSERT INTO demo (product_type, sales)
   VALUES ("Batteries", 150000);
   INSERT INTO demo (product_type, sales)
   VALUES ("Chargers", 200000);
   

2. テーブルを確認します。

   SQL

   SELECT * FROM `20210826_workshop_takaakiyayoidatabrickscom`.demo ORDER BY id ASC;
   

   id列にインサートされた順の連番が振られていることがわかります。
   

3. さらにダミーの行をインサートします。

   SQL

   USE `20210826_workshop_takaakiyayoidatabrickscom`;
   INSERT INTO demo (product_type, sales)
   VALUES ("Papers", 50000);
   INSERT INTO demo (product_type, sales)
   VALUES ("Repair", 1200000);
   

4. テーブルを確認します。

   SQL

   SELECT * FROM `20210826_workshop_takaakiyayoidatabrickscom`.demo ORDER BY id ASC;
   

   

クリーンアップするにはテーブルをドロップしてください。

SQL

DROP TABLE `20210826_workshop_takaakiyayoidatabrickscom`.demo

このようにIdentity列を使うことで、データベースでは当たり前の主キー・外部キーによるテーブル結合がデータレイク上のデータでも容易に行えるようになります。

Databricks 無料トライアル

Databricks 無料トライアル