Naeel
d286d92a05
Without ContentLength, Python BaseHTTPRequestHandler read 0 bytes from body. operator v0.1.37, python runtime v0.1.4, pg-table-writer HTML form
105 lines
4.6 KiB
Go
105 lines
4.6 KiB
Go
// Изменено: 2026-03-11
|
|
// invoke.go — прокси-обработчик для вызова HTTP-триггеров функций.
|
|
// Маршрут: ANY /fn/{namespace}/{name} и /fn/{namespace}/{name}/**
|
|
// Не защищён auth-токеном — это публичный эндпоинт для вызова функций.
|
|
// Проксирует запрос к ClusterIP Service функции внутри кластера:
|
|
// http://{name}.sless-fn-{namespace}.svc.cluster.local:8080
|
|
// Sub-path и query string пробрасываются как есть:
|
|
// /fn/ns/notes/add?title=x → http://notes.sless-fn-ns.svc.../add?title=x
|
|
|
|
package handler
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/gorilla/mux"
|
|
)
|
|
|
|
// httpClient используется для обращения к функциям внутри кластера.
|
|
// Таймаут 30s — достаточно для холодного старта функции.
|
|
var httpClient = &http.Client{Timeout: 30 * time.Second}
|
|
|
|
// hopByHopHeaders — заголовки которые нельзя пробрасывать через прокси (RFC 2616 §13.5.1).
|
|
// Они управляют соединением между двумя узлами, а не end-to-end.
|
|
// Особо опасен Transfer-Encoding: если пробросить его, клиент неверно интерпретирует тело.
|
|
var hopByHopHeaders = map[string]bool{
|
|
"Connection": true,
|
|
"Keep-Alive": true,
|
|
"Proxy-Authenticate": true,
|
|
"Proxy-Authorization": true,
|
|
"Te": true,
|
|
"Trailers": true,
|
|
"Transfer-Encoding": true,
|
|
"Upgrade": true,
|
|
}
|
|
|
|
// InvokeFunction проксирует входящий запрос к Service функции в кластере.
|
|
// Namespace выбирается из пути, имя функции — тоже из пути.
|
|
// Сохраняет метод, тело, Content-Type, sub-path и query string.
|
|
func (h *Handler) InvokeFunction(w http.ResponseWriter, r *http.Request) {
|
|
vars := mux.Vars(r)
|
|
ns := vars["namespace"]
|
|
name := vars["name"]
|
|
|
|
// Вычисляем sub-path после /fn/{namespace}/{name}
|
|
// Например: /fn/default/notes/add → subPath = /add
|
|
prefix := "/fn/" + ns + "/" + name
|
|
subPath := strings.TrimPrefix(r.URL.Path, prefix)
|
|
|
|
// Внутренний URL к Service функции (DNS внутри кластера)
|
|
target := fmt.Sprintf("http://%s.sless-fn-%s.svc.cluster.local:8080%s", name, ns, subPath)
|
|
|
|
// Пробрасываем query string если есть
|
|
if r.URL.RawQuery != "" {
|
|
target += "?" + r.URL.RawQuery
|
|
}
|
|
|
|
// Создаём проксируемый запрос с тем же методом и телом
|
|
proxyReq, err := http.NewRequestWithContext(r.Context(), r.Method, target, r.Body)
|
|
if err != nil {
|
|
h.Log.Error("invoke: failed to create proxy request", "err", err, "ns", ns, "fn", name)
|
|
writeJSON(w, http.StatusInternalServerError, errResp("failed to create proxy request"))
|
|
return
|
|
}
|
|
|
|
// Пробрасываем Content-Type и Content-Length если есть.
|
|
// Content-Length обязателен: Python BaseHTTPRequestHandler читает тело
|
|
// ровно столько байт, сколько указано в заголовке; без него body = пусто.
|
|
if ct := r.Header.Get("Content-Type"); ct != "" {
|
|
proxyReq.Header.Set("Content-Type", ct)
|
|
}
|
|
proxyReq.ContentLength = r.ContentLength
|
|
|
|
resp, err := httpClient.Do(proxyReq)
|
|
if err != nil {
|
|
// "no such host" — Service не существует (функция удалена), возвращаем 404.
|
|
// Это отличается от временной сетевой ошибки: NXDOMAIN строго означает отсутствие записи.
|
|
if strings.Contains(err.Error(), "no such host") {
|
|
writeJSON(w, http.StatusNotFound, errResp("function not found"))
|
|
return
|
|
}
|
|
h.Log.Error("invoke: function unreachable", "err", err, "ns", ns, "fn", name, "target", target)
|
|
writeJSON(w, http.StatusBadGateway, errResp("function unreachable: "+err.Error()))
|
|
return
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
// Копируем заголовки и статус из ответа функции.
|
|
// Hop-by-hop заголовки фильтруем: они управляют конкретным TCP-соединением
|
|
// и не должны пробрасываться через прокси (RFC 2616 §13.5.1).
|
|
for k, vals := range resp.Header {
|
|
if hopByHopHeaders[k] {
|
|
continue
|
|
}
|
|
for _, v := range vals {
|
|
w.Header().Add(k, v)
|
|
}
|
|
}
|
|
w.WriteHeader(resp.StatusCode)
|
|
_, _ = io.Copy(w, resp.Body)
|
|
}
|