mirror of
https://github.com/Youzini-afk/ST-Bionic-Memory-Ecology.git
synced 2026-05-15 22:30:38 +08:00
feat(authority): add server-primary capability probe
This commit is contained in:
Youzini-afk
367
runtime/authority-capabilities.js
Normal file
367
runtime/authority-capabilities.js
Normal file
@@ -0,0 +1,367 @@
|
||||
const DEFAULT_AUTHORITY_BASE_URL = "/api/plugins/authority";
|
||||
const DEFAULT_AUTHORITY_PROBE_INTERVAL_MS = 60000;
|
||||
|
||||
const SQL_FEATURES = ["sql", "sql.query", "sql.page", "sql.pageall", "querysql"];
|
||||
const SQL_MUTATION_FEATURES = ["sql", "sql.mutation", "sql.execute", "sql.write", "sql.transaction"];
|
||||
const TRIVIUM_FEATURES = ["trivium", "trivium.search", "trivium.query", "trivium.filterwhere", "trivium.bulkupsert"];
|
||||
const JOB_FEATURES = ["jobs", "jobs.list", "jobs.wait", "events", "sse"];
|
||||
const BLOB_FEATURES = ["blob", "blob.write", "privatefiles", "private.files", "files.private"];
|
||||
|
||||
function toBoolean(value, fallback = false) {
|
||||
if (typeof value === "boolean") return value;
|
||||
if (typeof value === "number") return value !== 0;
|
||||
if (typeof value === "string") {
|
||||
const normalized = value.trim().toLowerCase();
|
||||
if (["true", "1", "yes", "on"].includes(normalized)) return true;
|
||||
if (["false", "0", "no", "off"].includes(normalized)) return false;
|
||||
}
|
||||
return fallback;
|
||||
}
|
||||
|
||||
function clampInteger(value, fallback, min, max) {
|
||||
const numeric = Number(value);
|
||||
if (!Number.isFinite(numeric)) return fallback;
|
||||
return Math.min(max, Math.max(min, Math.trunc(numeric)));
|
||||
}
|
||||
|
||||
function normalizeMode(value, fallback, allowed) {
|
||||
const normalized = String(value ?? fallback).trim().toLowerCase();
|
||||
return allowed.includes(normalized) ? normalized : fallback;
|
||||
}
|
||||
|
||||
function normalizeFeatureName(value) {
|
||||
return String(value ?? "").trim().toLowerCase();
|
||||
}
|
||||
|
||||
function addFeature(features, value) {
|
||||
const normalized = normalizeFeatureName(value);
|
||||
if (normalized) features.add(normalized);
|
||||
}
|
||||
|
||||
function addFeatureObject(features, value, prefix = "") {
|
||||
if (!value || typeof value !== "object" || Array.isArray(value)) return;
|
||||
for (const [key, enabled] of Object.entries(value)) {
|
||||
if (!enabled) continue;
|
||||
addFeature(features, key);
|
||||
if (prefix) addFeature(features, `${prefix}.${key}`);
|
||||
if (enabled && typeof enabled === "object" && !Array.isArray(enabled)) {
|
||||
addFeatureObject(features, enabled, prefix ? `${prefix}.${key}` : key);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
function hasAnyFeature(features, aliases) {
|
||||
return aliases.some((alias) => features.has(normalizeFeatureName(alias)));
|
||||
}
|
||||
|
||||
function createFeatureReadiness(features) {
|
||||
return {
|
||||
sql: hasAnyFeature(features, SQL_FEATURES),
|
||||
sqlMutation: hasAnyFeature(features, SQL_MUTATION_FEATURES),
|
||||
trivium: hasAnyFeature(features, TRIVIUM_FEATURES),
|
||||
jobs: hasAnyFeature(features, JOB_FEATURES),
|
||||
blob: hasAnyFeature(features, BLOB_FEATURES),
|
||||
};
|
||||
}
|
||||
|
||||
function collectMissingFeatures(readiness) {
|
||||
const missing = [];
|
||||
if (!readiness.sql) missing.push("sql.query");
|
||||
if (!readiness.sqlMutation) missing.push("sql.mutation");
|
||||
if (!readiness.trivium) missing.push("trivium.search");
|
||||
if (!readiness.jobs) missing.push("jobs");
|
||||
if (!readiness.blob) missing.push("blob-or-private-files");
|
||||
return missing;
|
||||
}
|
||||
|
||||
function isRelativeAuthorityUrl(baseUrl) {
|
||||
return /^\//.test(String(baseUrl || ""));
|
||||
}
|
||||
|
||||
function normalizeLatencyMs(startedAt, finishedAt) {
|
||||
return Math.max(0, Math.round((Number(finishedAt) - Number(startedAt)) * 10) / 10);
|
||||
}
|
||||
|
||||
function readNowMs() {
|
||||
if (typeof performance === "object" && typeof performance.now === "function") {
|
||||
return performance.now();
|
||||
}
|
||||
return Date.now();
|
||||
}
|
||||
|
||||
export function normalizeAuthorityBaseUrl(baseUrl = DEFAULT_AUTHORITY_BASE_URL) {
|
||||
const normalized = String(baseUrl || DEFAULT_AUTHORITY_BASE_URL).trim() || DEFAULT_AUTHORITY_BASE_URL;
|
||||
return normalized.replace(/\/+$/, "");
|
||||
}
|
||||
|
||||
export function normalizeAuthoritySettings(settings = {}) {
|
||||
const source = settings && typeof settings === "object" && !Array.isArray(settings) ? settings : {};
|
||||
const enabledMode = normalizeMode(source.authorityEnabled ?? source.enabledMode, "auto", ["auto", "on", "off", "true", "false"]);
|
||||
return {
|
||||
enabledMode: enabledMode === "true" ? "on" : enabledMode === "false" ? "off" : enabledMode,
|
||||
enabled: enabledMode !== "off" && enabledMode !== "false",
|
||||
baseUrl: normalizeAuthorityBaseUrl(source.authorityBaseUrl ?? source.baseUrl),
|
||||
primaryWhenAvailable: toBoolean(source.authorityPrimaryWhenAvailable ?? source.primaryWhenAvailable, true),
|
||||
storageMode: normalizeMode(source.authorityStorageMode ?? source.storageMode, "auto-server-primary", ["auto-server-primary", "server-primary", "local-primary", "off"]),
|
||||
vectorMode: normalizeMode(source.authorityVectorMode ?? source.vectorMode, "auto-primary", ["auto-primary", "primary", "local-fallback", "off"]),
|
||||
sqlPrimary: toBoolean(source.authoritySqlPrimary ?? source.sqlPrimary, true),
|
||||
triviumPrimary: toBoolean(source.authorityTriviumPrimary ?? source.triviumPrimary, true),
|
||||
jobsEnabled: toBoolean(source.authorityJobsEnabled ?? source.jobsEnabled, true),
|
||||
blobCheckpointEnabled: toBoolean(source.authorityBlobCheckpointEnabled ?? source.blobCheckpointEnabled, true),
|
||||
diagnosticsEnabled: toBoolean(source.authorityDiagnosticsEnabled ?? source.diagnosticsEnabled, true),
|
||||
failOpen: toBoolean(source.authorityFailOpen ?? source.failOpen, true),
|
||||
probeIntervalMs: clampInteger(source.authorityProbeIntervalMs ?? source.probeIntervalMs, DEFAULT_AUTHORITY_PROBE_INTERVAL_MS, 1000, 3600000),
|
||||
};
|
||||
}
|
||||
|
||||
export function buildAuthorityProbeUrls(baseUrl = DEFAULT_AUTHORITY_BASE_URL) {
|
||||
const normalizedBaseUrl = normalizeAuthorityBaseUrl(baseUrl);
|
||||
return [
|
||||
`${normalizedBaseUrl}/v1/diagnostics/probe`,
|
||||
`${normalizedBaseUrl}/v1/probe`,
|
||||
`${normalizedBaseUrl}/probe`,
|
||||
normalizedBaseUrl,
|
||||
];
|
||||
}
|
||||
|
||||
export function collectAuthorityFeatures(payload = {}) {
|
||||
const features = new Set();
|
||||
const source = payload && typeof payload === "object" && !Array.isArray(payload) ? payload : {};
|
||||
for (const value of Array.isArray(source.features) ? source.features : []) {
|
||||
addFeature(features, value);
|
||||
}
|
||||
for (const value of Array.isArray(source.capabilities) ? source.capabilities : []) {
|
||||
addFeature(features, value);
|
||||
}
|
||||
addFeatureObject(features, source.features);
|
||||
addFeatureObject(features, source.capabilities);
|
||||
addFeatureObject(features, source.services);
|
||||
addFeatureObject(features, source.featureFlags);
|
||||
addFeatureObject(features, source.flags);
|
||||
return features;
|
||||
}
|
||||
|
||||
export function createDefaultAuthorityCapabilityState(overrides = {}) {
|
||||
return {
|
||||
enabledMode: "auto",
|
||||
baseUrl: DEFAULT_AUTHORITY_BASE_URL,
|
||||
installed: false,
|
||||
healthy: false,
|
||||
sessionReady: false,
|
||||
permissionReady: false,
|
||||
minimumFeatureSetReady: false,
|
||||
serverPrimaryReady: false,
|
||||
storagePrimaryReady: false,
|
||||
triviumPrimaryReady: false,
|
||||
jobsReady: false,
|
||||
blobReady: false,
|
||||
features: [],
|
||||
missingFeatures: ["sql.query", "sql.mutation", "trivium.search", "jobs", "blob-or-private-files"],
|
||||
reason: "not-probed",
|
||||
lastError: "",
|
||||
endpoint: "",
|
||||
status: 0,
|
||||
latencyMs: 0,
|
||||
lastProbeAt: 0,
|
||||
updatedAt: "",
|
||||
...overrides,
|
||||
};
|
||||
}
|
||||
|
||||
export function normalizeAuthorityCapabilityState(input = {}, settings = {}) {
|
||||
const normalizedSettings = normalizeAuthoritySettings(settings);
|
||||
const source = input && typeof input === "object" && !Array.isArray(input) ? input : {};
|
||||
const features = new Set((Array.isArray(source.features) ? source.features : []).map(normalizeFeatureName).filter(Boolean));
|
||||
const readiness = createFeatureReadiness(features);
|
||||
const missingFeatures = Array.isArray(source.missingFeatures) && source.missingFeatures.length
|
||||
? source.missingFeatures.map(String)
|
||||
: collectMissingFeatures(readiness);
|
||||
const healthy = Boolean(source.healthy);
|
||||
const sessionReady = source.sessionReady == null ? healthy : Boolean(source.sessionReady);
|
||||
const permissionReady = source.permissionReady == null ? sessionReady : Boolean(source.permissionReady);
|
||||
const storagePrimaryReady = healthy && sessionReady && permissionReady && readiness.sql && readiness.sqlMutation;
|
||||
const triviumPrimaryReady = healthy && sessionReady && permissionReady && readiness.trivium;
|
||||
const jobsReady = healthy && readiness.jobs;
|
||||
const blobReady = healthy && readiness.blob;
|
||||
const minimumFeatureSetReady = storagePrimaryReady && triviumPrimaryReady && jobsReady && blobReady;
|
||||
const serverPrimaryRequested =
|
||||
normalizedSettings.enabled &&
|
||||
normalizedSettings.primaryWhenAvailable &&
|
||||
normalizedSettings.storageMode !== "local-primary" &&
|
||||
normalizedSettings.storageMode !== "off";
|
||||
return createDefaultAuthorityCapabilityState({
|
||||
...source,
|
||||
enabledMode: normalizedSettings.enabledMode,
|
||||
baseUrl: normalizedSettings.baseUrl,
|
||||
installed: Boolean(source.installed),
|
||||
healthy,
|
||||
sessionReady,
|
||||
permissionReady,
|
||||
minimumFeatureSetReady,
|
||||
serverPrimaryReady: serverPrimaryRequested && minimumFeatureSetReady,
|
||||
storagePrimaryReady,
|
||||
triviumPrimaryReady,
|
||||
jobsReady,
|
||||
blobReady,
|
||||
features: Array.from(features).sort(),
|
||||
missingFeatures,
|
||||
reason: String(source.reason || (healthy ? "ok" : "not-ready")),
|
||||
lastError: String(source.lastError || ""),
|
||||
endpoint: String(source.endpoint || ""),
|
||||
status: clampInteger(source.status, 0, 0, 999),
|
||||
latencyMs: Math.max(0, Number(source.latencyMs) || 0),
|
||||
lastProbeAt: Math.max(0, Number(source.lastProbeAt) || 0),
|
||||
updatedAt: String(source.updatedAt || ""),
|
||||
});
|
||||
}
|
||||
|
||||
export function normalizeAuthorityProbeResponse(payload = {}, context = {}) {
|
||||
const settings = normalizeAuthoritySettings(context.settings || {});
|
||||
const features = collectAuthorityFeatures(payload);
|
||||
const readiness = createFeatureReadiness(features);
|
||||
const missingFeatures = collectMissingFeatures(readiness);
|
||||
const sessionReady = payload?.sessionReady ?? payload?.session?.ready ?? payload?.session?.active ?? true;
|
||||
const permissionReady = payload?.permissionReady ?? payload?.permissions?.ready ?? payload?.authorized ?? sessionReady;
|
||||
const healthy = payload?.healthy ?? payload?.ok ?? true;
|
||||
return normalizeAuthorityCapabilityState(
|
||||
{
|
||||
installed: true,
|
||||
healthy: Boolean(healthy),
|
||||
sessionReady: Boolean(sessionReady),
|
||||
permissionReady: Boolean(permissionReady),
|
||||
features: Array.from(features),
|
||||
missingFeatures,
|
||||
reason: missingFeatures.length ? "missing-required-features" : "ok",
|
||||
endpoint: context.endpoint || "",
|
||||
status: context.status || 200,
|
||||
latencyMs: context.latencyMs || 0,
|
||||
lastProbeAt: context.nowMs || Date.now(),
|
||||
updatedAt: new Date(context.nowMs || Date.now()).toISOString(),
|
||||
},
|
||||
settings,
|
||||
);
|
||||
}
|
||||
|
||||
export async function probeAuthorityCapabilities(options = {}) {
|
||||
const settings = normalizeAuthoritySettings(options.settings || {});
|
||||
const nowMs = Number(options.nowMs) || Date.now();
|
||||
if (!settings.enabled || settings.storageMode === "off") {
|
||||
return normalizeAuthorityCapabilityState(
|
||||
{
|
||||
reason: "disabled",
|
||||
lastProbeAt: nowMs,
|
||||
updatedAt: new Date(nowMs).toISOString(),
|
||||
},
|
||||
settings,
|
||||
);
|
||||
}
|
||||
|
||||
const fetchImpl = options.fetchImpl || (typeof fetch === "function" ? fetch : null);
|
||||
if (typeof fetchImpl !== "function") {
|
||||
return normalizeAuthorityCapabilityState(
|
||||
{
|
||||
reason: "fetch-unavailable",
|
||||
lastError: "fetch unavailable",
|
||||
lastProbeAt: nowMs,
|
||||
updatedAt: new Date(nowMs).toISOString(),
|
||||
},
|
||||
settings,
|
||||
);
|
||||
}
|
||||
|
||||
if (options.allowRelativeUrl === false && isRelativeAuthorityUrl(settings.baseUrl)) {
|
||||
return normalizeAuthorityCapabilityState(
|
||||
{
|
||||
reason: "relative-url-unavailable",
|
||||
lastError: "relative Authority URL cannot be probed in this runtime",
|
||||
lastProbeAt: nowMs,
|
||||
updatedAt: new Date(nowMs).toISOString(),
|
||||
},
|
||||
settings,
|
||||
);
|
||||
}
|
||||
|
||||
let headers = { Accept: "application/json" };
|
||||
if (typeof options.headerProvider === "function") {
|
||||
try {
|
||||
headers = { ...headers, ...(options.headerProvider() || {}) };
|
||||
} catch {
|
||||
headers = { ...headers };
|
||||
}
|
||||
}
|
||||
|
||||
let lastError = "";
|
||||
let lastStatus = 0;
|
||||
for (const endpoint of buildAuthorityProbeUrls(settings.baseUrl)) {
|
||||
const startedAt = readNowMs();
|
||||
try {
|
||||
const response = await fetchImpl(endpoint, { method: "GET", headers });
|
||||
const finishedAt = readNowMs();
|
||||
const status = Number(response?.status || 0);
|
||||
lastStatus = status;
|
||||
if (status === 404) continue;
|
||||
if (status === 401 || status === 403) {
|
||||
return normalizeAuthorityCapabilityState(
|
||||
{
|
||||
installed: true,
|
||||
healthy: false,
|
||||
sessionReady: false,
|
||||
permissionReady: false,
|
||||
reason: "permission-denied",
|
||||
lastError: `HTTP ${status}`,
|
||||
endpoint,
|
||||
status,
|
||||
latencyMs: normalizeLatencyMs(startedAt, finishedAt),
|
||||
lastProbeAt: nowMs,
|
||||
updatedAt: new Date(nowMs).toISOString(),
|
||||
},
|
||||
settings,
|
||||
);
|
||||
}
|
||||
if (!response?.ok) {
|
||||
return normalizeAuthorityCapabilityState(
|
||||
{
|
||||
installed: status > 0,
|
||||
healthy: false,
|
||||
reason: "http-error",
|
||||
lastError: `HTTP ${status || "unknown"}`,
|
||||
endpoint,
|
||||
status,
|
||||
latencyMs: normalizeLatencyMs(startedAt, finishedAt),
|
||||
lastProbeAt: nowMs,
|
||||
updatedAt: new Date(nowMs).toISOString(),
|
||||
},
|
||||
settings,
|
||||
);
|
||||
}
|
||||
let payload = {};
|
||||
try {
|
||||
payload = typeof response.json === "function" ? await response.json() : {};
|
||||
} catch {
|
||||
payload = {};
|
||||
}
|
||||
return normalizeAuthorityProbeResponse(payload, {
|
||||
settings,
|
||||
endpoint,
|
||||
status,
|
||||
latencyMs: normalizeLatencyMs(startedAt, finishedAt),
|
||||
nowMs,
|
||||
});
|
||||
} catch (error) {
|
||||
lastError = error?.message || String(error);
|
||||
}
|
||||
}
|
||||
|
||||
return normalizeAuthorityCapabilityState(
|
||||
{
|
||||
installed: false,
|
||||
healthy: false,
|
||||
reason: lastStatus === 404 ? "not-installed" : "probe-failed",
|
||||
lastError,
|
||||
status: lastStatus,
|
||||
lastProbeAt: nowMs,
|
||||
updatedAt: new Date(nowMs).toISOString(),
|
||||
},
|
||||
settings,
|
||||
);
|
||||
}
|
||||
Reference in New Issue
Block a user