TiDB Cloud をdbt と統合する

データ構築ツール (dbt)は、分析エンジニアが SQL ステートメントを使用してウェアハウス内のデータを変換するのに役立つ、人気の高いオープンソースのデータ変換ツールです。2 dbt-tidbインを使用すると、 TiDB Cloudで作業する分析エンジニアは、テーブルやビューの作成プロセスを考えることなく、SQL を介してフォームを直接作成し、データを照合できます。

このドキュメントでは、dbt プロジェクトを例に、 TiDB Cloudで dbt を使用する方法を紹介します。

ステップ1: dbtとdbt-tidbをインストールする

dbt と dbt-tidb は 1 つのコマンドだけでインストールできます。次のコマンドでは、dbt-tidb をインストールするときに dbt が依存関係としてインストールされます。

pip install dbt-tidb

dbt を別途インストールすることもできます。dbt ドキュメントのdbtのインストール方法参照してください。

ステップ2: デモプロジェクトを作成する

dbt 関数を試すには、dbt-lab が提供するデモ プロジェクトジャッフルショップ使用できます。プロジェクトは GitHub から直接クローンできます。

git clone https://github.com/dbt-labs/jaffle_shop && \ cd jaffle_shop

jaffle_shopディレクトリ内のすべてのファイルは次のように構造化されています。

. ├── LICENSE ├── README.md ├── dbt_project.yml ├── etc │ ├── dbdiagram_definition.txt │ └── jaffle_shop_erd.png ├── models │ ├── customers.sql │ ├── docs.md │ ├── orders.sql │ ├── overview.md │ ├── schema.yml │ └── staging │ ├── schema.yml │ ├── stg_customers.sql │ ├── stg_orders.sql │ └── stg_payments.sql └── seeds ├── raw_customers.csv ├── raw_orders.csv └── raw_payments.csv

このディレクトリ内:

  • dbt_project.ymlは dbt プロジェクト構成ファイルであり、プロジェクト名とデータベース構成ファイル情報を保持します。

  • modelsディレクトリには、プロジェクトの SQL モデルとテーブル スキーマが含まれています。このセクションはデータ アナリストが記述することに注意してください。モデルの詳細については、 SQL モデル参照してください。

  • seedsディレクトリには、データベース エクスポート ツールによってダンプされた CSV ファイルが格納されます。たとえば、 Dumplingを使用してTiDB Cloudデータをエクスポートする CSV ファイルにすることができます。5 jaffle_shopでは、これらの CSV ファイルが処理される生データとして使用されます。

ステップ3: プロジェクトを構成する

プロジェクトを構成するには、次の手順を実行します。

  1. グローバル構成を完了します。

    プロフィールフィールドの説明を参照してデフォルトのグローバル プロファイル~/.dbt/profiles.ymlを編集し、 TiDB Cloudとの接続を構成できます。

    sudo vi ~/.dbt/profiles.yml

    エディターで次の構成を追加します。

    jaffle_shop_tidb: # Project name target: dev # Target outputs: dev: type: tidb # The specific adapter to use server: gateway01.ap-southeast-1.prod.aws.tidbcloud.com # The TiDB Cloud clusters' endpoint to connect to port: 4000 # The port to use schema: analytics # Specify the schema (database) to normalize data into username: xxxxxxxxxxx.root # The username to use to connect to the TiDB Cloud clusters password: "your_password" # The password to use for authenticating to the TiDB Cloud clusters

    server username値はportクラスターの接続ダイアログから取得できます。このダイアログを開くには、プロジェクトのクラスターページに移動し、ターゲット クラスターの名前をクリックして概要ページに移動し、右上隅の[接続]をクリックします。

  2. プロジェクトの構成を完了します。

    jaffle_shop プロジェクト ディレクトリで、プロジェクト構成ファイルdbt_project.ymlを編集し、 profileフィールドをjaffle_shop_tidbに変更します。この構成により、プロジェクトは~/.dbt/profiles.ymlファイルで指定されたとおりにデータベースからクエリを実行できるようになります。

    vi dbt_project.yml

    エディターで、次のように構成を更新します。

    name: 'jaffle_shop' config-version: 2 version: '0.1' profile: 'jaffle_shop_tidb' # note the modification here model-paths: ["models"] # model path seed-paths: ["seeds"] # seed path test-paths: ["tests"] analysis-paths: ["analysis"] macro-paths: ["macros"] target-path: "target" clean-targets: - "target" - "dbt_modules" - "logs" require-dbt-version: [">=1.0.0", "<2.0.0"] models: jaffle_shop: materialized: table # *.sql which in models/ would be materialized to table staging: materialized: view # *.sql which in models/staging/ would bt materialized to view
  3. 構成を確認します。

    次のコマンドを実行して、データベースとプロジェクトの構成が正しいかどうかを確認します。

    dbt debug

ステップ4: (オプション) CSVファイルを読み込む

注記:

この手順はオプションです。処理対象のデータがすでにターゲット データベースにある場合は、この手順をスキップできます。

プロジェクトの作成と構成が正常に完了したので、CSV データをロードし、CSV をターゲット データベースのテーブルとして実現します。

  1. CSV データをロードし、CSV をターゲット データベース内のテーブルとして実現します。

    dbt seed

    出力例は次のとおりです。

    Running with dbt=1.0.1 Partial parse save file not found. Starting full parse. Found 5 models, 20 tests, 0 snapshots, 0 analyses, 172 macros, 0 operations, 3 seed files, 0 sources, 0 exposures, 0 metrics Concurrency: 1 threads (target='dev') 1 of 3 START seed file analytics.raw_customers.................................. [RUN] 1 of 3 OK loaded seed file analytics.raw_customers.............................. [INSERT 100 in 0.19s] 2 of 3 START seed file analytics.raw_orders..................................... [RUN] 2 of 3 OK loaded seed file analytics.raw_orders................................. [INSERT 99 in 0.14s] 3 of 3 START seed file analytics.raw_payments................................... [RUN] 3 of 3 OK loaded seed file analytics.raw_payments............................... [INSERT 113 in 0.24s]

    結果からわかるように、シード ファイルが開始され、 analytics.raw_customersanalytics.raw_ordersanalytics.raw_paymentsの 3 つのテーブルにロードされました。

  2. TiDB Cloudで結果を確認します。

    show databasesコマンドは、dbt が作成した新しいanalyticsデータベースを一覧表示します。5 コマンドはanalytics show tablesに、作成したテーブルに対応する 3 つのテーブルがあることを示します。

    mysql> SHOW DATABASES; +--------------------+ | Database | +--------------------+ | INFORMATION_SCHEMA | | METRICS_SCHEMA | | PERFORMANCE_SCHEMA | | analytics | | io_replicate | | mysql | | test | +--------------------+ 7 rows in set (0.00 sec) mysql> USE ANALYTICS; mysql> SHOW TABLES; +---------------------+ | Tables_in_analytics | +---------------------+ | raw_customers | | raw_orders | | raw_payments | +---------------------+ 3 rows in set (0.00 sec) mysql> SELECT * FROM raw_customers LIMIT 10; +------+------------+-----------+ | id | first_name | last_name | +------+------------+-----------+ | 1 | Michael | P. | | 2 | Shawn | M. | | 3 | Kathleen | P. | | 4 | Jimmy | C. | | 5 | Katherine | R. | | 6 | Sarah | R. | | 7 | Martin | M. | | 8 | Frank | R. | | 9 | Jennifer | F. | | 10 | Henry | W. | +------+------------+-----------+ 10 rows in set (0.10 sec)

ステップ5: データの変換

これで、構成されたプロジェクトを実行し、データ変換を完了する準備が整いました。

  1. dbt プロジェクトを実行してデータ変換を完了します。

    dbt run

    出力例は次のとおりです。

    Running with dbt=1.0.1 Found 5 models, 20 tests, 0 snapshots, 0 analyses, 170 macros, 0 operations, 3 seed files, 0 sources, 0 exposures, 0 metrics Concurrency: 1 threads (target='dev') 1 of 5 START view model analytics.stg_customers................................. [RUN] 1 of 5 OK created view model analytics.stg_customers............................ [SUCCESS 0 in 0.31s] 2 of 5 START view model analytics.stg_orders.................................... [RUN] 2 of 5 OK created view model analytics.stg_orders............................... [SUCCESS 0 in 0.23s] 3 of 5 START view model analytics.stg_payments.................................. [RUN] 3 of 5 OK created view model analytics.stg_payments............................. [SUCCESS 0 in 0.29s] 4 of 5 START table model analytics.customers.................................... [RUN] 4 of 5 OK created table model analytics.customers............................... [SUCCESS 0 in 0.76s] 5 of 5 START table model analytics.orders....................................... [RUN] 5 of 5 OK created table model analytics.orders.................................. [SUCCESS 0 in 0.63s] Finished running 3 view models, 2 table models in 2.27s. Completed successfully Done. PASS=5 WARN=0 ERROR=0 SKIP=0 TOTAL=5

    結果は、2 つのテーブル ( analytics.customersanalytics.orders ) と 3 つのビュー ( analytics.stg_customersanalytics.stg_orders 、およびanalytics.stg_payments ) が正常に作成されたことを示しています。

  2. TiDB Cloudにアクセスして、変換が成功したことを確認します。

    mysql> USE ANALYTICS; mysql> SHOW TABLES; +---------------------+ | Tables_in_analytics | +---------------------+ | customers | | orders | | raw_customers | | raw_orders | | raw_payments | | stg_customers | | stg_orders | | stg_payments | +---------------------+ 8 rows in set (0.00 sec) mysql> SELECT * FROM customers LIMIT 10; +-------------+------------+-----------+-------------+-------------------+------------------+-------------------------+ | customer_id | first_name | last_name | first_order | most_recent_order | number_of_orders | customer_lifetime_value | +-------------+------------+-----------+-------------+-------------------+------------------+-------------------------+ | 1 | Michael | P. | 2018-01-01 | 2018-02-10 | 2 | 33.0000 | | 2 | Shawn | M. | 2018-01-11 | 2018-01-11 | 1 | 23.0000 | | 3 | Kathleen | P. | 2018-01-02 | 2018-03-11 | 3 | 65.0000 | | 4 | Jimmy | C. | NULL | NULL | NULL | NULL | | 5 | Katherine | R. | NULL | NULL | NULL | NULL | | 6 | Sarah | R. | 2018-02-19 | 2018-02-19 | 1 | 8.0000 | | 7 | Martin | M. | 2018-01-14 | 2018-01-14 | 1 | 26.0000 | | 8 | Frank | R. | 2018-01-29 | 2018-03-12 | 2 | 45.0000 | | 9 | Jennifer | F. | 2018-03-17 | 2018-03-17 | 1 | 30.0000 | | 10 | Henry | W. | NULL | NULL | NULL | NULL | +-------------+------------+-----------+-------------+-------------------+------------------+-------------------------+ 10 rows in set (0.00 sec)

    出力には、さらに 5 つのテーブルまたはビューが追加され、テーブルまたはビュー内のデータが変換されたことが示されています。この例では、顧客テーブルのデータの一部のみが表示されています。

ステップ6: ドキュメントを生成する

dbt を使用すると、プロジェクトの全体的な構造を表示し、すべてのテーブルとビューを説明するビジュアル ドキュメントを生成できます。

ビジュアルドキュメントを生成するには、次の手順を実行します。

  1. ドキュメントを生成します:

    dbt docs generate
  2. サーバーを起動します:

    dbt docs serve
  3. ブラウザからドキュメントにアクセスするには、 http://localhost:8080に進みます。

プロフィールフィールドの説明

オプション説明必須?
type使用する特定のアダプタ必須tidb
serverTiDB Cloudクラスターの接続エンドポイント必須gateway01.ap-southeast-1.prod.aws.tidbcloud.com
port使用するポート必須4000
schemaデータを正規化するスキーマ(データベース)必須analytics
usernameTiDB Cloudクラスターに接続するために使用するユーザー名必須xxxxxxxxxxx.root
passwordTiDB Cloudクラスターへの認証に使用するパスワード必須"your_password"
retriesTiDB Cloudクラスターへの接続の再試行回数 (デフォルトは 1)オプション2

サポートされている関数

dbt-tidb では以下の関数を直接使用できます。使用方法についてはdbt-util参照してください。

以下の関数がサポートされています:

  • bool_or
  • cast_bool_to_text
  • dateadd
  • datediffは dbt-util とは少しdatediffことに注意してください。切り上げではなく切り捨てになります。
  • date_trunc
  • hash
  • safe_cast
  • split_part
  • last_day
  • cast_bool_to_text
  • concat
  • escape_single_quotes
  • except
  • intersect
  • length
  • position
  • replace
  • right

このページは役に立ちましたか?