test: полный прогон POSTGRES example — changes, delete/recreate, new function

2026-03-19 18:45:18 +03:00
parent 8a8b815492
commit d87981713d
22 changed files with 959 additions and 53 deletions
@@ -0,0 +1,94 @@
+# 2026-03-18 (обновлено: plain text вывод; фильтрация SLESS_EXCLUDE)
+# funcs_list.py — HTTP-функция: список пользовательских функций, человекочитаемый plain text.
+# Вызывает внутренний REST API оператора (ClusterIP, без TLS).
+# Возвращает str → python runtime отдаёт text/plain напрямую без json.dumps.
+#
+# Env vars:
+#   SLESS_API_URL      — URL оператора (http://sless-operator.sless.svc.cluster.local:9090)
+#   SLESS_NAMESPACE    — namespace пользователя (sless-{hex16})
+#   SLESS_TOKEN        — JWT токен для /v1/ API
+#   SLESS_EXTERNAL_URL — публичный базовый URL (https://sless.kube5s.ru)
+#   SLESS_EXCLUDE      — comma-separated имена функций, которые не показывать
+
+import os
+import requests
+
+SEP = "─" * 52
+
+
+def _comment(fn, http_trigs, cron_trigs):
+    phase   = fn.get("phase", "?")
+    runtime = fn.get("runtime", "?")
+    if http_trigs:
+        active = "активна" if http_trigs[0].get("active") else "неактивна"
+        return f"HTTP endpoint ({runtime}) — {phase}, {active}"
+    elif cron_trigs:
+        schedule = cron_trigs[0].get("schedule", "?")
+        active   = "активна" if cron_trigs[0].get("active") else "неактивна"
+        return f"Cron '{schedule}' ({runtime}) — {phase}, {active}"
+    else:
+        return f"Job/runner без триггера ({runtime}) — {phase}"
+
+
+def list_all(event):
+    api_url   = os.environ["SLESS_API_URL"].rstrip("/")
+    namespace = os.environ["SLESS_NAMESPACE"]
+    token     = os.environ["SLESS_TOKEN"]
+    ext_url   = os.environ.get("SLESS_EXTERNAL_URL", "").rstrip("/")
+    exclude   = {n.strip() for n in os.environ.get("SLESS_EXCLUDE", "").split(",") if n.strip()}
+
+    headers = {"Authorization": f"Bearer {token}"}
+    fns  = requests.get(f"{api_url}/v1/namespaces/{namespace}/functions",  headers=headers, timeout=10)
+    trs  = requests.get(f"{api_url}/v1/namespaces/{namespace}/triggers",   headers=headers, timeout=10)
+    fns.raise_for_status()
+    trs.raise_for_status()
+
+    trig_idx = {}
+    for tr in trs.json():
+        fn_name = tr.get("function") or tr.get("functionRef")
+        if fn_name:
+            trig_idx.setdefault(fn_name, []).append(tr)
+
+    items = []
+    for fn in fns.json():
+        name = fn["name"]
+        if name in exclude:
+            continue
+        http_t = [t for t in trig_idx.get(name, []) if t.get("type") == "http"]
+        cron_t = [t for t in trig_idx.get(name, []) if t.get("type") == "cron"]
+        is_active = any(t.get("enabled", True) and t.get("active", False) for t in trig_idx.get(name, []))
+        items.append((fn, http_t, cron_t, is_active))
+
+    # Сортировка: активные вверх, затем по имени
+    items.sort(key=lambda x: (not x[3], x[0]["name"]))
+
+    lines = []
+    for fn, http_t, cron_t, is_active in items:
+        name = fn["name"]
+        lines.append(SEP)
+        lines.append(f"  {_comment(fn, http_t, cron_t)}")
+        lines.append(f"  name:      {name}")
+        lines.append(f"  runtime:   {fn.get('runtime', '?')}")
+        lines.append(f"  phase:     {fn.get('phase', '?')}")
+        lines.append(f"  active:    {'да' if is_active else 'нет'}")
+
+        if http_t:
+            url = f"{ext_url}/fn/{namespace}/{name}" if ext_url else http_t[0].get("url", "")
+            lines.append(f"  url:       {url}")
+        if cron_t:
+            lines.append(f"  cron:      {cron_t[0].get('schedule', '?')}")
+        if fn.get("created_at"):
+            lines.append(f"  created:   {fn['created_at']}")
+        if fn.get("last_built_at"):
+            lines.append(f"  built:     {fn['last_built_at']}")
+        if fn.get("message"):
+            lines.append(f"  message:   {fn['message']}")
+
+    lines.append(SEP)
+    lines.append(f"  namespace: {namespace}  |  total: {len(items)}")
+    lines.append(SEP)
+
+    # Возвращаем str — python runtime отдаст text/plain напрямую
+    return "\n".join(lines) + "\n"
+
+
@@ -0,0 +1 @@
+requests==2.31.0
@@ -0,0 +1,8 @@
+{
+    "name": "pg-info",
+    "version": "1.0.0",
+    "description": "sless nodejs20 function: pg version + table info",
+    "dependencies": {
+        "pg": "8.11.0"
+    }
+}
@@ -0,0 +1,43 @@
+// 2026-03-18
+// pg_info.js — NodeJS-функция: проверка работы JS runtime + чтение мета-данных БД.
+// Подключается к PostgreSQL через пакет pg, возвращает версию сервера и счётчик строк.
+// Демонстрирует: nodejs20 runtime, npm-зависимость (package.json), PG из JS.
+//
+// ENV (те же что у python-функций):
+//   PGHOST, PGPORT, PGDATABASE, PGUSER, PGPASSWORD, PGSSLMODE
+//
+// Entrypoint: pg_info.info
+
+'use strict';
+
+const { Client } = require('pg');
+
+exports.info = async (event) => {
+    const client = new Client({
+        host: process.env.PGHOST,
+        port: parseInt(process.env.PGPORT || '5432'),
+        database: process.env.PGDATABASE,
+        user: process.env.PGUSER,
+        password: process.env.PGPASSWORD,
+        // pg-пакет требует явного ssl-объекта; rejectUnauthorized: false — т.к.
+        // self-signed cert на nubes managed PG, но канал всё равно шифруется.
+        ssl: process.env.PGSSLMODE === 'require' ? { rejectUnauthorized: false } : false,
+    });
+
+    await client.connect();
+    try {
+        const [versionRes, countRes] = await Promise.all([
+            client.query('SELECT version() AS v'),
+            client.query('SELECT COUNT(*) AS cnt FROM terraform_demo_table'),
+        ]);
+
+        return {
+            runtime: 'nodejs20',
+            node_version: process.version,
+            pg_version: versionRes.rows[0].v,
+            table_rows: parseInt(countRes.rows[0].cnt, 10),
+        };
+    } finally {
+        await client.end();
+    }
+};
@@ -0,0 +1,3 @@
+# 2026-03-17 00:00
+# requirements.txt — зависимости для функции запуска SQL.
+psycopg2-binary==2.9.9
@@ -0,0 +1,39 @@
+# 2026-03-17 00:00
+# sql_runner.py — функция для выполнения SQL-операторов из входного события.
+import os
+import psycopg2
+
+
+def run_sql(event):
+    # Выполняет список SQL-операторов в одной транзакции для атомарной инициализации схемы.
+    # Параметры подключения передаются раздельно, чтобы избежать ошибок парсинга DSN при спецсимволах.
+    pg_host = os.environ["PGHOST"]
+    pg_port = os.environ.get("PGPORT", "5432")
+    pg_database = os.environ["PGDATABASE"]
+    pg_user = os.environ["PGUSER"]
+    pg_password = os.environ["PGPASSWORD"]
+    pg_sslmode = os.environ.get("PGSSLMODE", "require")
+    statements = event.get("statements", [])
+
+    if not statements:
+        return {"error": "no statements provided"}
+
+    connection = psycopg2.connect(
+        host=pg_host,
+        port=pg_port,
+        dbname=pg_database,
+        user=pg_user,
+        password=pg_password,
+        sslmode=pg_sslmode,
+    )
+    try:
+        cursor = connection.cursor()
+        for statement in statements:
+            cursor.execute(statement)
+        connection.commit()
+        return {"ok": True, "executed": len(statements)}
+    except Exception as error:
+        connection.rollback()
+        return {"error": str(error)}
+    finally:
+        connection.close()
@@ -0,0 +1 @@
+psycopg2-binary==2.9.9
@@ -0,0 +1,130 @@
+# 2026-03-19
+# table_rw.py — чтение и запись строк в terraform_demo_table.
+# Два entrypoint в одном файле: list_rows (JSON API) и add_row (HTML-страница + POST-обработчик).
+# ENV: PGHOST, PGPORT, PGDATABASE, PGUSER, PGPASSWORD, PGSSLMODE
+
+import os
+import json
+import psycopg2
+import psycopg2.extras
+
+
+def _connect():
+    return psycopg2.connect(
+        host=os.environ["PGHOST"],
+        port=os.environ.get("PGPORT", "5432"),
+        dbname=os.environ["PGDATABASE"],
+        user=os.environ["PGUSER"],
+        password=os.environ["PGPASSWORD"],
+        sslmode=os.environ.get("PGSSLMODE", "require"),
+    )
+
+
+def list_rows(event):
+    # Возвращает все строки terraform_demo_table, отсортированные по убыванию created_at.
+    conn = _connect()
+    try:
+        cur = conn.cursor(cursor_factory=psycopg2.extras.RealDictCursor)
+        cur.execute(
+            "SELECT id, title, created_at::text FROM terraform_demo_table ORDER BY created_at DESC"
+        )
+        rows = [dict(r) for r in cur.fetchall()]
+        return {"rows": rows, "count": len(rows)}
+    finally:
+        conn.close()
+
+
+def _render_page(rows, message=""):
+    # HTML-страница с формой ввода и таблицей строк.
+    # message — статус последней операции (успех / ошибка).
+    rows_html = "".join(
+        f"<tr><td>{r['id']}</td><td>{r['title']}</td><td>{r['created_at']}</td></tr>"
+        for r in rows
+    )
+    msg_html = f'<p class="msg">{message}</p>' if message else ""
+    return f"""<!DOCTYPE html>
+<html lang="ru">
+<head>
+  <meta charset="utf-8">
+  <title>pg-table-writer</title>
+  <style>
+    body {{ font-family: sans-serif; max-width: 700px; margin: 40px auto; background: #111; color: #eee; }}
+    h1 {{ color: #7dd3fc; }}
+    form {{ display: flex; gap: 8px; margin-bottom: 24px; }}
+    input[type=text] {{ flex: 1; padding: 8px 12px; border-radius: 6px; border: 1px solid #444; background: #1e1e1e; color: #eee; font-size: 15px; }}
+    button {{ padding: 8px 18px; background: #2563eb; color: #fff; border: none; border-radius: 6px; cursor: pointer; font-size: 15px; }}
+    button:hover {{ background: #1d4ed8; }}
+    table {{ width: 100%; border-collapse: collapse; }}
+    th, td {{ padding: 8px 10px; border-bottom: 1px solid #333; text-align: left; }}
+    th {{ color: #7dd3fc; }}
+    .msg {{ color: #4ade80; margin-bottom: 12px; }}
+  </style>
+</head>
+<body>
+  <h1>pg-table-writer</h1>
+  <form method="POST">
+    <input type="text" name="title" placeholder="Введите строку..." autofocus required>
+    <button type="submit">Добавить</button>
+  </form>
+  {msg_html}
+  <table>
+    <thead><tr><th>#</th><th>title</th><th>created_at</th></tr></thead>
+    <tbody>{rows_html}</tbody>
+  </table>
+</body>
+</html>"""
+
+
+def add_row(event):
+    # GET  → HTML-страница с формой и списком строк.
+    # POST → вставляет строку из form-поля title или JSON-поля title,
+    #        затем возвращает обновлённую HTML-страницу.
+    # POST с Content-Type: application/json (curl/API) → возвращает JSON.
+    method = event.get("_method", "GET")
+
+    if method == "GET":
+        conn = _connect()
+        try:
+            cur = conn.cursor(cursor_factory=psycopg2.extras.RealDictCursor)
+            cur.execute("SELECT id, title, created_at::text FROM terraform_demo_table ORDER BY created_at DESC")
+            rows = [dict(r) for r in cur.fetchall()]
+        finally:
+            conn.close()
+        return _render_page(rows)
+
+    # POST — вставка строки
+    # Поле title приходит либо из JSON-тела, либо из application/x-www-form-urlencoded.
+    # Сервер уже распарсил JSON в event; form-данные приходят как event["body"] = "title=...".
+    title = event.get("title", "").strip()
+    if not title:
+        # Попытка распарсить form-encoded body (браузерная форма)
+        body = event.get("body", "")
+        if body.startswith("title="):
+            from urllib.parse import unquote_plus
+            title = unquote_plus(body[len("title="):].split("&")[0]).strip()
+
+    if not title:
+        return {"ok": False, "error": "title is required"}
+
+    conn = _connect()
+    try:
+        cur = conn.cursor(cursor_factory=psycopg2.extras.RealDictCursor)
+        cur.execute(
+            "INSERT INTO terraform_demo_table (title) VALUES (%s) RETURNING id, title, created_at::text",
+            (title,),
+        )
+        row = dict(cur.fetchone())
+        conn.commit()
+
+        # Если запрос из браузера (form POST) — возвращаем обновлённую страницу.
+        # Если из curl/API — возвращаем JSON.
+        accept = event.get("_accept", "")
+        if "application/json" in accept:
+            return {"ok": True, "row": row}
+
+        # Перечитываем все строки для обновлённой страницы
+        cur.execute("SELECT id, title, created_at::text FROM terraform_demo_table ORDER BY created_at DESC")
+        rows = [dict(r) for r in cur.fetchall()]
+        return _render_page(rows, message=f"Добавлено: «{row['title']}»")
+    finally:
+        conn.close()