【活用事例あり】BigQueryからCloudSQLのデータ参照

ADKとは、Googleが提供するAIエージェントを構築するためのオープンソースフレームワークです。特徴としては、Googleサービスと連携するためのToolsが豊富に用意されていることです。

google.adk.tools.google_api_tool.google_api_tool_set を使うことで、LLMを使って簡単にBigQueryのテーブルを参照することができます。

一方、CloudSQLのデータは現状google_api_tool_setには用意されていないこともあり、BigQueryを介して、LLMを使ってCloudSQLのデータを参照するという実装にしました。機能としては近いうちに導入していく予定なので、実装の際の苦労等は別途ブログにまとめていきます。

設定方法

ここからは設定方法の解説をします。以下公式ページにもやり方の記載はありますが、ややわかりにくい部分もあるので解説となります。

Cloud SQL に接続する  |  BigQuery  |  Google Cloud

cloud.google.com

Cloud SQL 接続の作成

1. Google Cloud コンソールで [BigQuery] ページに移動します。

 2. 左側のナビゲーションメニューから 添付のを選択し、[データを追加] をクリックします。

 3. [データソースを検索] で、使用しているCloudSQLデータベースに併せて「mysql」または「postgresql」と入力します。

4. [特徴量データソース]を選択します。

5. [外部データに関する BigQuery テーブルの設定（フェデレーション）]を選択します。

6. 接続情報を入力して、[接続を作成]します

7. 添付の流れで接続情報ができていることを確認できます。権限設定でサービスアカウントIDを使用するのでコピーしておきましょう。

必要権限の付与

大きく①BigQueryがCloudSQLにアクセスするための権限と、②作成した外部接続をユーザが使うための権限 の2つを設定する必要があります。

①については、[IAM]より先ほど、コピーしたサービスアカウントIDをプリンシパルとし、「Cloud SQL クライアント」という事前定義ロールを付与します。

②については添付の導線より、[共有]をクリックします。

そうすると、以下添付の通り、IAM設定できるのでBigQuery Connection Userの事前定義ロールを対象のユーザーに付与します。（リソース単位ではなく、プロジェクト単位でIAM付与してももちろん問題ないです。）

クエリの実行

SELECT
*
FROM
EXTERNAL_QUERY(
"プロジェクトID.リージョン.接続ID",
"SELECT * FROM データベース名.テーブル名"
);

上記の形式でSQLを書くと実行できます。

最後に

いかがだったでしょうか。本日BigQueryからCloudSQLのデータを参照する方法を解説しました。参考になればと思います。