TiDB Cloudと dbt を統合する
データ構築ツール(dbt)は、分析エンジニアが SQL ステートメントを使用してウェアハウス内のデータを変換するのに役立つ、人気のあるオープンソース データ変換ツールです。 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を使用して CSV ファイルにTiDB Cloudデータをエクスポートするできます。jaffle_shop
プロジェクトでは、これらの CSV ファイルが処理対象の生データとして使用されます。
ステップ 3: プロジェクトを構成する
プロジェクトを構成するには、次の手順を実行します。
グローバル構成を完了します。
プロフィールフィールドの説明を参照し、デフォルトのグローバル プロファイル
~/.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 clustersserver
、port
、およびusername
の値は、クラスターの接続ダイアログから取得できます。このダイアログを開くには、プロジェクトのクラスターページに移動し、ターゲット クラスターの名前をクリックして概要ページに移動し、右上隅の[接続]をクリックします。プロジェクトの構成を完了します。
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構成を確認します。
次のコマンドを実行して、データベースとプロジェクトの構成が正しいかどうかを確認します。
dbt debug
ステップ 4: (オプション) CSV ファイルをロードする
注記:
このステップはオプションです。処理対象のデータがターゲット データベースにすでに存在する場合は、この手順をスキップできます。
プロジェクトの作成と構成が正常に完了したので、次は CSV データをロードし、その CSV をターゲット データベース内のテーブルとして実体化します。
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]結果からわかるように、シード ファイルが開始され、3 つのテーブル (
analytics.raw_customers
、analytics.raw_orders
、およびanalytics.raw_payments
) にロードされました。TiDB Cloudで結果を確認します。
show databases
コマンドは、dbt が作成したanalytics
データベースをリストします。show tables
コマンドは、作成したテーブルに対応するテーブルがanalytics
データベースに 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: データを変換する
これで、構成されたプロジェクトを実行し、データ変換を完了する準備が整いました。
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.customers
とanalytics.orders
) と 3 つのビュー (analytics.stg_customers
、analytics.stg_orders
、およびanalytics.stg_payments
) が正常に作成されたことを示しています。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 つのテーブルまたはビューが追加され、テーブルまたはビューのデータが変換されたことが示されています。この例では、customer テーブルのデータの一部のみが示されています。
ステップ 6: ドキュメントを生成する
dbt を使用すると、プロジェクトの全体的な構造を表示し、すべてのテーブルとビューを説明するビジュアル ドキュメントを生成できます。
ビジュアルドキュメントを生成するには、次の手順を実行します。
ドキュメントを生成します。
dbt docs generateサーバーを起動します。
dbt docs serveブラウザからドキュメントにアクセスするには、 http://ローカルホスト:8080に進みます。
プロフィールフィールドの説明
オプション | 説明 | 必須? | 例 |
---|---|---|---|
type | 使用する特定のアダプター | 必須 | tidb |
server | 接続先のTiDB Cloudクラスターのエンドポイント | 必須 | gateway01.ap-southeast-1.prod.aws.tidbcloud.com |
port | 使用するポート | 必須 | 4000 |
schema | データを正規化するためのスキーマ (データベース) | 必須 | analytics |
username | TiDB Cloudクラスターへの接続に使用するユーザー名 | 必須 | xxxxxxxxxxx.root |
password | TiDB Cloudクラスターへの認証に使用するパスワード | 必須 | "your_password" |
retries | TiDB Cloudクラスターへの接続の再試行回数 (デフォルトでは 1 回) | オプション | 2 |
サポートされている関数
次の関数をdbt-tidb で直接使用できます。使用方法については、 dbt-utilを参照してください。
次の関数がサポートされています。
bool_or
cast_bool_to_text
dateadd
datediff
。datediff
は dbt-util とは少し異なることに注意してください。切り上げではなく切り下げます。date_trunc
hash
safe_cast
split_part
last_day
cast_bool_to_text
concat
escape_single_quotes
except
intersect
length
position
replace
right