[新機能] Cortex Search Auto-Suspendがパブリックプレビューになったので試してみた

-- テスト用データベース・スキーマの準備
USE ROLE SYSADMIN;
CREATE DATABASE IF NOT EXISTS CORTEX_SEARCH_TEST_DB;
CREATE SCHEMA IF NOT EXISTS CORTEX_SEARCH_TEST_DB.AUTO_SUSPEND_TEST;
USE SCHEMA CORTEX_SEARCH_TEST_DB.AUTO_SUSPEND_TEST;

-- テスト用テーブルの作成
CREATE OR REPLACE TABLE SEARCH_SOURCE_DATA (
    id INT,
    title STRING,
    content STRING,
    category STRING
);

-- サンプルデータの挿入
INSERT INTO SEARCH_SOURCE_DATA VALUES
    (1, 'Snowflakeの概要', 'Snowflakeはクラウドネイティブなデータプラットフォームです。', 'platform'),
    (2, 'Cortex Searchとは', 'Cortex SearchはSnowflakeのハイブリッド検索サービスです。', 'cortex'),
    (3, 'Auto-Suspend機能', 'Auto-Suspendは非アクティブ時にサービングコンピュートを自動的にサスペンドします。', 'cortex'),
    (4, 'コスト最適化', 'Snowflakeではウェアハウスの自動停止でコストを最適化できます。', 'cost'),
    (5, 'データ共有', 'Snowflakeのデータ共有機能でセキュアにデータを共有できます。', 'platform');

-- Cortex Search Serviceの作成（AUTO_SUSPEND付き）
CREATE OR REPLACE CORTEX SEARCH SERVICE AUTO_SUSPEND_TEST_SERVICE
    ON content
    ATTRIBUTES category
    WAREHOUSE = COMPUTE_WH
    TARGET_LAG = '1 hour'
    AUTO_SUSPEND = 1800
AS (
    SELECT
        id,
        title,
        content,
        category
    FROM SEARCH_SOURCE_DATA
);

-- サービス一覧の確認
SHOW CORTEX SEARCH SERVICES;

-- サービスの詳細確認
DESCRIBE CORTEX SEARCH SERVICE AUTO_SUSPEND_TEST_SERVICE;

-- Cortex Search Serviceへのクエリ実行
SELECT PARSE_JSON(
    SNOWFLAKE.CORTEX.SEARCH_PREVIEW(
        'CORTEX_SEARCH_TEST_DB.AUTO_SUSPEND_TEST.AUTO_SUSPEND_TEST_SERVICE',
        '{
            "query": "コスト最適化",
            "columns": ["id", "title", "content", "category"],
            "limit": 3
        }'
    )
)['results'] AS results;

-- 30分以上経過後に状態を確認
SHOW CORTEX SEARCH SERVICES LIKE 'AUTO_SUSPEND_TEST_SERVICE';

-- サスペンド状態からクエリを実行（自動復帰）
SELECT PARSE_JSON(
    SNOWFLAKE.CORTEX.SEARCH_PREVIEW(
        'CORTEX_SEARCH_TEST_DB.AUTO_SUSPEND_TEST.AUTO_SUSPEND_TEST_SERVICE',
        '{
            "query": "Snowflake",
            "columns": ["id", "title", "content", "category"],
            "limit": 3
        }'
    )
)['results'] AS results;

-- 復帰後の状態確認
SHOW CORTEX SEARCH SERVICES LIKE 'AUTO_SUSPEND_TEST_SERVICE';

-- AUTO_SUSPEND時間を3600秒（1時間）に変更
ALTER CORTEX SEARCH SERVICE AUTO_SUSPEND_TEST_SERVICE SET AUTO_SUSPEND = 3600;

-- 設定変更の確認
SHOW CORTEX SEARCH SERVICES LIKE 'AUTO_SUSPEND_TEST_SERVICE';

-- AUTO_SUSPENDを無効化（NULLに設定）
ALTER CORTEX SEARCH SERVICE AUTO_SUSPEND_TEST_SERVICE SET AUTO_SUSPEND = NULL;

-- 無効化の確認
SHOW CORTEX SEARCH SERVICES LIKE 'AUTO_SUSPEND_TEST_SERVICE';