Hologres:GUC パラメータ

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLWarning;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class HologresQueryIdDemo {

    // NOTICE メッセージ内の "QueryID: " に一致します。
    private static final Pattern QUERY_ID_PATTERN = Pattern.compile(
            "query[_ ]?id\\s*(?:is\\b|[=:])?\\s*([0-9a-zA-Z_\\-]{6,})", Pattern.CASE_INSENSITIVE);

    public static void main(String[] args) throws Exception {
        // ご利用のネットワーク環境 (インターネットまたは VPC) に応じたエンドポイントを使用します。
        // コンソールのインスタンス詳細ページで確認できます。
        String url = "jdbc:postgresql://<ENDPOINT>:80/<DB_NAME>";
        String user = "<ACCESS_KEY_ID>";
        String password = "<ACCESS_KEY_SECRET>";

        try (Connection conn = DriverManager.getConnection(url, user, password);
             Statement stmt = conn.createStatement()) {

            // クエリ ID の表示を有効にします。このパラメーターはデフォルトでオフになっており、このセッションにのみ適用されます。
            stmt.execute("set hg_experimental_display_query_id = on;");

            String sql = "select count(*), sum(id) from holo_query_id_demo;";

            // 各クエリ ID が 1 つの SQL ステートメントに対応するように、前のステートメントによって残された警告をクリアします。
            stmt.clearWarnings();
            boolean hasResultSet = stmt.execute(sql);

            // 結果セットを読み取ります。クエリ ID はその一部ではありません。
            if (hasResultSet) {
                try (ResultSet rs = stmt.getResultSet()) {
                    while (rs.next()) {
                        System.out.println("rows      : (" + rs.getLong(1) + ", " + rs.getLong(2) + ")");
                    }
                }
            }

            // NOTICE メッセージは SQLWarning オブジェクトのチェーンとして届きます。チェーンをたどってクエリ ID を解析します。
            String queryId = null;
            List<String> notices = new ArrayList<>();
            for (SQLWarning w = stmt.getWarnings(); w != null; w = w.getNextWarning()) {
                notices.add(w.getMessage());
                Matcher m = QUERY_ID_PATTERN.matcher(w.getMessage());
                if (m.find()) {
                    queryId = m.group(1);
                }
            }

            System.out.println("query_id  : " + queryId);
            System.out.println("notices   : " + notices);
        }
    }
}

rows      : (1000, 500500)
query_id  : 1002002606817139331
notices   : [One or more columns in the following table(s) do not have statistics: holo_query_id_demo, QueryID: 1002002606817139331]

import re
import psycopg

NOTICES = []
QUERY_IDS = []

QUERY_ID_PATTERN = re.compile(
    r"query[_ ]?id\s*(?:is\b|[=:])?\s*([0-9a-zA-Z_\-]{6,})", re.IGNORECASE
)

def notice_handler(diag):
    """サーバーが送信するすべての NOTICE に対して一度呼び出されます。そこからクエリ ID を解析します。"""
    text = diag.message_primary or ""
    NOTICES.append(text)
    match = QUERY_ID_PATTERN.search(text)
    if match:
        QUERY_IDS.append(match.group(1))

conn = psycopg.connect(
    host="<ENDPOINT>",          # コンソールのインスタンス詳細ページで確認できます。
    port=80,
    dbname="<DB_NAME>",
    user="<ACCESS_KEY_ID>",
    password="<ACCESS_KEY_SECRET>",
)
conn.autocommit = True
conn.add_notice_handler(notice_handler)   # NOTICE コールバックを登録します。

cur = conn.cursor()
# クエリ ID の表示を有効にします。このパラメーターはデフォルトでオフになっており、このセッションにのみ適用されます。
cur.execute("set hg_experimental_display_query_id = on;")

def run_sql(sql, params=None, fetch=True):
    """ステートメントを実行し、(rows, query_id, notices) を返します。結果セットがない場合、rows は None です。"""
    NOTICES.clear()
    QUERY_IDS.clear()
    cur.execute(sql, params)
    rows = None
    if fetch and cur.description is not None:
        rows = cur.fetchall()
    query_id = QUERY_IDS[-1] if QUERY_IDS else None
    return rows, query_id, list(NOTICES)

# DML ステートメント
sql_insert = "insert into holo_query_id_demo select i, 'v' || i from generate_series(1, 1000) i;"
_, query_id_insert, notices_insert = run_sql(sql_insert, fetch=False)
print("SQL       :", sql_insert)
print("query_id  :", query_id_insert)
print("notices   :", notices_insert)
print()

# クエリステートメント
sql_select = "select count(*), sum(id) from holo_query_id_demo;"
rows_select, query_id_select, notices_select = run_sql(sql_select)
print("SQL       :", sql_select)
print("query_id  :", query_id_select)
print("rows      :", rows_select)
print("notices   :", notices_select)

SQL       : insert into holo_query_id_demo select i, 'v' || i from generate_series(1, 1000) i;
query_id  : 1002002606817130830
notices   : ['QueryID: 1002002606817130830']

SQL       : select count(*), sum(id) from holo_query_id_demo;
query_id  : 1002002606817139331
rows      : [(1000, 500500)]
notices   : ['One or more columns in the following table(s) do not have statistics: holo_query_id_demo', 'QueryID: 1002002606817139331']

# まず pip install psycopg_pool を実行します。
from psycopg_pool import ConnectionPool

def configure(conn):
    conn.autocommit = True
    conn.add_notice_handler(notice_handler)
    conn.execute("set hg_experimental_display_query_id = on;")

pool = ConnectionPool(kwargs=CONN_INFO, configure=configure, min_size=1, max_size=4)

with pool.connection() as conn:
    conn.execute("select 1;").fetchall()

// HikariCP: connectionInitSql は、各物理接続が確立されるときに実行されます。
HikariConfig config = new HikariConfig();
config.setJdbcUrl(url);
config.setUsername(user);
config.setPassword(password);
config.setConnectionInitSql("set hg_experimental_display_query_id = on;");
HikariDataSource dataSource = new HikariDataSource(config);

// Druid: connectionInitSqls は、複数の初期化ステートメントを受け入れます。
DruidDataSource druid = new DruidDataSource();
druid.setUrl(url);
druid.setUsername(user);
druid.setPassword(password);
druid.setConnectionInitSqls(Collections.singletonList("set hg_experimental_display_query_id = on;"));