Function Compute:インスタンスライフサイクルフック

コードの呼び出し

コードの呼び出しフックを設定すると、インスタンスの起動時または停止直前に、システムは `POST /initialize` や `GET /pre-stop` などの HTTP リクエストを関数に送信します。アプリケーションコードでこのリクエストを処理する必要があります。

カスタムイメージで Initializer フックを使用するには、HTTP サーバーに `/initialize` パスと POST メソッドのロジックを実装する必要があります。同様に、PreStop フックでは `/pre-stop` パスと GET メソッドのロジックが必要です。次のサンプルコードは、カスタム Python 3.10 ランタイムでの実装例です。

import os
from flask import Flask
from flask import request

app = Flask(__name__)

@app.route('/initialize', methods=['POST'])
def init_invoke():
    rid = request.headers.get('x-fc-request-id')
    print("FC Initialize Start RequestId: " + rid)
    # 独自の処理を記述
    print("FC Initialize End RequestId: " + rid)
    return "OK"

@app.route('/', defaults={'path': ''})
@app.route('/<path:path>', methods=['GET', 'POST', 'PUT', 'DELETE'])
def hello_world(path):
    rid = request.headers.get('x-fc-request-id')
    print("FC invoke Start RequestId: " + rid)
    # 独自の処理を記述
    print("FC invoke End RequestId: " + rid)
    return "Hello, World!", 200, [('Function-Name', os.getenv('FC_FUNCTION_NAME'))]


@app.route('/pre-stop', methods=['GET'])
def prestop_invoke():
    rid = request.headers.get('x-fc-request-id')
    print("FC PreStop Start RequestId: " + rid)
    # 独自の処理を記述
    print("FC PreStop End RequestId: " + rid)
    return "OK"


if __name__ == '__main__':
    app.run(host='0.0.0.0', port=9000)

上記の例は、初期化が成功した場合のものです。次の例は、Python で /initialize フックが失敗するケースを示しています。

@app.route('/initialize', methods=['POST'])
def init():
    raise Exception("hahaha")
    return "OK", 200, []

@app.route('/initialize', methods=['POST'])
def init():
    return "OK", 404, []

命令の実行

命令の実行フックは、インスタンスの起動後にコマンドを実行してインスタンスを初期化します。関数の実行環境でサポートされている任意のシェル (例：/bin/bash、/bin/sh、/bin/csh、/bin/zsh) を使用できます。

Function Compute コンソールには、/bin/bash と /bin/sh の 2 つのプリセットオプションが用意されています。このトピックでは、/bin/sh を例として使用します。

#!/bin/sh

URL="http://localhost:7860"
REQUEST_PATH="sdapi/v1/txt2img"
JSON_DATA='{
	"prompt": "",
	"steps": 1,
	"height": 8,
	"width": 8
}'
temp_file=$(mktemp)
trap 'rm -f "$temp_file"' EXIT

if ! http_code=$(curl -s -w "%{http_code}" \
	-H "Content-Type: application/json" \
	-d "$JSON_DATA" \
	-o "$temp_file" \
	-X POST "$URL"/"$REQUEST_PATH"); then
	echo "{\"status\": \"curl_error\", \"code\": $curl_exit_code}" \
	| jq . >&2
	exit 1
fi

response=$(<"$temp_file")
echo "http code $http_code"

if [ "$http_code" -eq 200 ]; then
	echo "$response" | jq -r '.' || printf "%s\n" "$response"
	exit 0
else
	echo "$response_body" \
	| jq -c 'if type == "object" then . else {raw: .} end' 2>/dev/null \
	| jq -s '{status: "http_error", code: $code, response: .[0]}' \
		--arg code "$http_code" \
		>&2
	exit 1
fi

終了コード 0 は成功を示します。それ以外の終了コードは失敗を示します。トラブルシューティングを容易にするために、スクリプト内でエラー詳細を標準エラー (stderr) に書き込むことができます。例えば、次のスニペットは、curl リクエストが失敗した場合にエラー詳細を出力し、終了コードを 1 に設定する方法を示しています。

if ! http_code=$(curl -s -w "%{http_code}" \
	-H "Content-Type: application/json" \
	-d "$JSON_DATA" \
	-o "$temp_file" \
	-X POST "$URL"/"$REQUEST_PATH"); then
	echo "{\"status\": \"curl_error\", \"code\": $curl_exit_code}" \
	| jq . >&2
	exit 1
fi

コードの呼び出し

命令の実行

終了コードが 0 でない場合、命令は失敗し、再実行されません。その後、システムはエラーを返します。