mirror of
https://github.com/Kong/go-pluginserver.git
synced 2025-10-05 16:17:01 +08:00
f8c9246577
* it runs... something * use strings for dict keys * small cleanups / better locking * add StepXXX RPC methods to express non-primitive PDK return types * remove commented code * define recognizable error type * config schema as a structure, not a string that happens to be decodable * changed Kong capitalization because of Go * reread plugin if modified on disk * docs about event cycle and callback return types * The standard package can't reload Plugins * add status report, including some timekeeping * expire instances after 60sec idle * a bit of logging * send PID on status and on connection (as unrequested notification)
148 lines
3.3 KiB
Bash
Executable File
148 lines
3.3 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
if [ "$1" == "-v" ]; then
|
|
VERBOSE='true'
|
|
fi
|
|
if [ "$1" == "-vv" ]; then
|
|
VERBOSE='very'
|
|
fi
|
|
|
|
pkill go-pluginserver
|
|
rq --help >/dev/null
|
|
|
|
echo "pwd: $PWD"
|
|
|
|
SOCKET='sock'
|
|
|
|
if pgrep go-pluginserver -l; then
|
|
PREVIOUS_SERVER="yes"
|
|
else
|
|
echo "starting server..."
|
|
[ -S "$SOCKET" ] && rm "$SOCKET"
|
|
./go-pluginserver -socket "$SOCKET" &
|
|
pgrep go-pluginserver -l
|
|
sleep 0.1s
|
|
fi
|
|
|
|
msg() {
|
|
query="$1"
|
|
[ -v VERBOSE ] && rq <<< "$query"
|
|
[ "$VERBOSE" == "very" ] && rq <<< "$query" -M | hd
|
|
METHOD="$(rq <<< "$query" -- 'at([2])')"
|
|
response="$(rq <<< "$query" -M | nc -NU "$SOCKET" | rq -m | jq 'select(.[0]==1)' )"
|
|
[ -v VERBOSE ] && rq <<< "$response"
|
|
|
|
ERROR="$(jq <<< "$response" '.[2]')"
|
|
RESULT="$(jq <<< "$response" '.[3]')"
|
|
}
|
|
|
|
assert_noerr() {
|
|
if [ "$ERROR" != "null" ]; then
|
|
echo "query: $query"
|
|
echo "response: $response"
|
|
echo "$METHOD : $ERROR" > /dev/stderr
|
|
exit 1
|
|
fi
|
|
echo "$METHOD: ok"
|
|
}
|
|
|
|
assert_fld_match() {
|
|
fld="$1"
|
|
pattern="$2"
|
|
|
|
fld_v="$(query_result '."'$fld'"')"
|
|
if [[ "$fld_v" =~ "$pattern" ]]; then
|
|
echo "==> $fld_v : ok"
|
|
else
|
|
echo "==> $fld_v : no match '$pattern'"
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
query_result() {
|
|
# echo "$RESULT"
|
|
jq <<< "$RESULT" "$1"
|
|
}
|
|
|
|
msg '[0, 19, "plugin.SetPluginDir", ["'$PWD'"]]'
|
|
assert_noerr
|
|
|
|
msg '[0, 19, "plugin.GetStatus", []]'
|
|
assert_noerr
|
|
assert_fld_match 'Plugins' '{}'
|
|
|
|
msg '[0, 19, "plugin.GetPluginInfo", ["go-log"]]'
|
|
assert_noerr
|
|
|
|
msg '[0, 19, "plugin.StartInstance", [{"Name":"go-log", "Config":"{\"reopen\":false, \"path\":\"/some/where/else/\"}"}]]'
|
|
assert_noerr
|
|
instanceID=$(query_result '."Id"')
|
|
echo "instanceID: $instanceID"
|
|
|
|
msg '[0, 19, "plugin.HandleEvent", [{"InstanceId": '$instanceID', "EventName": "access", "Params": [45, 23]}]]'
|
|
assert_noerr
|
|
eventId=$(query_result '."EventId"')
|
|
|
|
assert_fld_match 'Data.Method' 'kong.request.get_header'
|
|
assert_fld_match 'Data.Args' '"host"'
|
|
|
|
msg '[0, 19, "plugin.GetStatus", []]'
|
|
assert_noerr
|
|
|
|
|
|
msg '[0, 19, "plugin.Step", [{"EventId": '$eventId', "Data": "example.com"}]]'
|
|
assert_noerr
|
|
assert_fld_match 'Data.Method' 'kong.response.set_header'
|
|
assert_fld_match 'Data.Args[0]' '"x-hello-go"'
|
|
assert_fld_match 'Data.Args[1]' '"Go says hello to example.com (/some/where/else/)"'
|
|
|
|
msg '[0, 19, "plugin.Step", [{"EventId": '$eventId', "Data": "ok"}]]'
|
|
assert_noerr
|
|
# assert_fld_match 'Data.Method' 'kong.router.get_route'
|
|
#
|
|
# # msg '[0, 19, "plugin.StepError", [{"EventId": '$eventId', "Data": "not in the mood for routes"}]]'
|
|
# msg "$(cat <<-EOF
|
|
# [
|
|
# 0, 19, "plugin.StepRoute",
|
|
# [{
|
|
# "Data": {
|
|
# "created_at": 1574445198,
|
|
# "https_redirect_status_code": 426,
|
|
# "id": "c0ba987b-99e0-4342-a255-61ff47d54fe6",
|
|
# "paths": [ "/" ],
|
|
# "preserve_host": false,
|
|
# "protocols": [ "http", "https" ],
|
|
# "regex_priority": 0,
|
|
# "service": {
|
|
# "id": "a1a72823-4c75-42b3-92e6-79c865175287"
|
|
# },
|
|
# "strip_path": true,
|
|
# "updated_at": 1574445198
|
|
# },
|
|
# "EventId": 0
|
|
# }]
|
|
# ]
|
|
# EOF
|
|
# )"
|
|
# assert_noerr
|
|
# # callBack=$(query_result '."Data"')
|
|
assert_fld_match 'Data' '"ret"'
|
|
|
|
|
|
msg '[0, 19, "plugin.InstanceStatus", ['$instanceID']]'
|
|
assert_noerr
|
|
|
|
msg "[0, 19, \"plugin.CloseInstance\", [$instanceID]]"
|
|
assert_noerr
|
|
|
|
msg '[0, 19, "plugin.GetStatus", []]'
|
|
assert_noerr
|
|
assert_fld_match 'Plugins.go-log' '"Name":"go-log"'
|
|
|
|
|
|
if [ ! -v PREVIOUS_SERVER ]; then
|
|
pkill go-pluginserver
|
|
rm "$SOCKET"
|
|
fi
|
|
|