feat: add pg-query example + upload.go supports requirements.txt
- examples/pg-query/handler.py: Python function querying PostgreSQL invocations table - examples/pg-query/requirements.txt: psycopg2-binary==2.9.9 - examples/pg-query/main.tf: terraform config for sless_function + sless_trigger - internal/api/handler/upload.go: generateDockerfile() now accepts hasRequirements bool - scans zip for requirements.txt at upload time - adds RUN pip install --no-cache-dir to Dockerfile when requirements.txt present - doc/progress.md: updated status for pg-query e2e task
This commit is contained in:
commit
d32d0b9766
39
pg-query/handler.py
Normal file
39
pg-query/handler.py
Normal file
@ -0,0 +1,39 @@
|
||||
# 2026-03-07
|
||||
# handler.py — пример serverless функции, работающей с PostgreSQL.
|
||||
# Подключается к postgres.sless.svc.cluster.local:5432 (внутри кластера).
|
||||
# DSN берётся из env переменной PG_DSN (задаётся через env_vars ресурса sless_function).
|
||||
# handle(event) — принимает JSON event, возвращает список записей из таблицы invocations.
|
||||
import os
|
||||
import json
|
||||
|
||||
def handle(event):
|
||||
# psycopg2 входит в python:3.11-slim через pip install ниже,
|
||||
# либо нужно добавить в Dockerfile — для этого примера устанавливаем через requirements.txt.
|
||||
try:
|
||||
import psycopg2
|
||||
import psycopg2.extras
|
||||
except ImportError:
|
||||
return {"error": "psycopg2 not installed — add requirements.txt with psycopg2-binary"}
|
||||
|
||||
dsn = os.environ.get("PG_DSN", "")
|
||||
if not dsn:
|
||||
return {"error": "PG_DSN env variable is not set"}
|
||||
|
||||
limit = event.get("limit", 5)
|
||||
|
||||
conn = psycopg2.connect(dsn)
|
||||
try:
|
||||
cur = conn.cursor(cursor_factory=psycopg2.extras.RealDictCursor)
|
||||
cur.execute("SELECT id, namespace, function_name, status, started_at FROM invocations ORDER BY started_at DESC LIMIT %s", (limit,))
|
||||
rows = cur.fetchall()
|
||||
# RealDictCursor возвращает объекты, сериализуем вручную
|
||||
result = []
|
||||
for row in rows:
|
||||
r = dict(row)
|
||||
# datetime → str
|
||||
if r.get("started_at"):
|
||||
r["started_at"] = str(r["started_at"])
|
||||
result.append(r)
|
||||
return {"invocations": result, "count": len(result)}
|
||||
finally:
|
||||
conn.close()
|
||||
64
pg-query/main.tf
Normal file
64
pg-query/main.tf
Normal file
@ -0,0 +1,64 @@
|
||||
# 2026-03-07
|
||||
# main.tf — e2e тест: создать serverless функцию, которая читает из PostgreSQL.
|
||||
#
|
||||
# Использование:
|
||||
# 1. Убедиться что оператор запущен локально: source hack/local.env && go run main.go
|
||||
# 2. zip handler.py + requirements.txt:
|
||||
# zip handler.zip handler.py requirements.txt
|
||||
# 3. terraform init && terraform apply
|
||||
# 4. После apply (ждёт ~2 мин пока kaniko соберёт образ):
|
||||
# curl -s -X POST <url_из_trigger> -d '{}'
|
||||
#
|
||||
# Функция подключается к postgres.sless.svc.cluster.local:5432 изнутри кластера.
|
||||
# PG_DSN задаётся как env_var ресурса sless_function.
|
||||
|
||||
terraform {
|
||||
required_providers {
|
||||
sless = {
|
||||
source = "terra.k8c.ru/naeel/sless"
|
||||
version = "~> 0.1"
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
provider "sless" {
|
||||
endpoint = "http://localhost:9090"
|
||||
token = "dev-token-change-me"
|
||||
}
|
||||
|
||||
resource "sless_function" "pg_query" {
|
||||
namespace = "default"
|
||||
name = "pg-query"
|
||||
runtime = "python3.11"
|
||||
entrypoint = "handler.handle"
|
||||
memory_mb = 128
|
||||
timeout_sec = 30
|
||||
|
||||
# DSN для подключения к postgres внутри кластера
|
||||
# Хост: postgres.sless.svc.cluster.local (Service в namespace sless)
|
||||
env_vars = {
|
||||
PG_DSN = "postgres://sless:sless-pg-password@postgres.sless.svc.cluster.local:5432/sless?sslmode=disable"
|
||||
}
|
||||
|
||||
code_path = "${path.module}/handler.zip"
|
||||
code_hash = filemd5("${path.module}/handler.zip")
|
||||
}
|
||||
|
||||
resource "sless_trigger" "pg_query_http" {
|
||||
namespace = "default"
|
||||
name = "pg-query-http"
|
||||
type = "http"
|
||||
function = sless_function.pg_query.name
|
||||
}
|
||||
|
||||
output "function_phase" {
|
||||
value = sless_function.pg_query.phase
|
||||
}
|
||||
|
||||
output "function_image" {
|
||||
value = sless_function.pg_query.image_ref
|
||||
}
|
||||
|
||||
output "trigger_url" {
|
||||
value = sless_trigger.pg_query_http.url
|
||||
}
|
||||
1
pg-query/requirements.txt
Normal file
1
pg-query/requirements.txt
Normal file
@ -0,0 +1 @@
|
||||
psycopg2-binary==2.9.9
|
||||
Loading…
Reference in New Issue
Block a user