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

コードの呼び出し

`invoke code` フックを設定すると、システムは関数に HTTP リクエストを送信します。インスタンスの起動時に `POST /initialize` リクエストが送信され、インスタンスの停止時に `GET /pre-stop` リクエストが送信されます。ビジネスコードでこれらのリクエストを処理する必要があります。

カスタムイメージで Initializer フックを使用するには、HTTP サーバーに `POST /initialize` パスのロジックを実装する必要があります。同様に、PreStop フックの場合は、`GET /pre-stop` パスのロジックを実装する必要があります。次の例では、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 でない場合、命令は再実行されず、システムはエラーを返します。