vultr/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-09 20:10:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(plugin-state): add registerIfAbsent keyed store (#77135)

Browse Source
This commit is contained in:
Alex Knight
2026-05-04 18:20:04 +10:00
committed by GitHub
parent 071db2ca69
commit fcb396bf65
8 changed files with 283 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

128
src/plugin-state/plugin-state-store.sqlite.ts
View File

@@ -40,6 +40,7 @@ type UserVersionRow = {
type PluginStateStatements = {
upsertEntry: StatementSync;
insertEntryIfAbsent: StatementSync;
selectEntry: StatementSync;
selectEntries: StatementSync;
deleteEntry: StatementSync;
@@ -190,6 +191,23 @@ function createStatements(db: DatabaseSync): PluginStateStatements {
created_at = excluded.created_at,
expires_at = excluded.expires_at
`),
insertEntryIfAbsent: db.prepare(`
INSERT OR IGNORE INTO plugin_state_entries (
plugin_id,
namespace,
entry_key,
value_json,
created_at,
expires_at
) VALUES (
@plugin_id,
@namespace,
@entry_key,
@value_json,
@created_at,
@expires_at
)
`),
selectEntry: db.prepare(`
SELECT plugin_id, namespace, entry_key, value_json, created_at, expires_at
FROM plugin_state_entries
@@ -363,6 +381,44 @@ function runWriteTransaction<T>(
}
}
function enforcePostRegisterLimits(params: {
store: PluginStateDatabase;
pluginId: string;
namespace: string;
maxEntries: number;
now: number;
}): void {
const namespaceCount = countRow(
params.store.statements.countLiveNamespace.get(
params.pluginId,
params.namespace,
params.now,
) as CountRow | undefined,
);
if (namespaceCount > params.maxEntries) {
params.store.statements.deleteOldestNamespace.run(
params.pluginId,
params.namespace,
params.now,
namespaceCount - params.maxEntries,
);
}
const pluginCount = countRow(
params.store.statements.countLivePlugin.get(params.pluginId, params.now) as
| CountRow
| undefined,
);
if (pluginCount > MAX_ENTRIES_PER_PLUGIN) {
throw createPluginStateError({
code: "PLUGIN_STATE_LIMIT_EXCEEDED",
operation: "register",
message: `Plugin state for ${params.pluginId} exceeds the ${MAX_ENTRIES_PER_PLUGIN} live row limit.`,
path: params.store.path,
});
}
}
export function pluginStateRegister(params: {
pluginId: string;
namespace: string;
@@ -384,32 +440,56 @@ export function pluginStateRegister(params: {
created_at: now,
expires_at: expiresAt,
});
enforcePostRegisterLimits({
store,
pluginId: params.pluginId,
namespace: params.namespace,
maxEntries: params.maxEntries,
now,
});
});
} catch (error) {
throw wrapPluginStateError(
error,
"register",
"PLUGIN_STATE_WRITE_FAILED",
"Failed to register plugin state entry.",
);
}
}
const namespaceCount = countRow(
store.statements.countLiveNamespace.get(params.pluginId, params.namespace, now) as
| CountRow
| undefined,
);
if (namespaceCount > params.maxEntries) {
store.statements.deleteOldestNamespace.run(
params.pluginId,
params.namespace,
now,
namespaceCount - params.maxEntries,
);
}
const pluginCount = countRow(
store.statements.countLivePlugin.get(params.pluginId, now) as CountRow | undefined,
);
if (pluginCount > MAX_ENTRIES_PER_PLUGIN) {
throw createPluginStateError({
code: "PLUGIN_STATE_LIMIT_EXCEEDED",
operation: "register",
message: `Plugin state for ${params.pluginId} exceeds the ${MAX_ENTRIES_PER_PLUGIN} live row limit.`,
path: store.path,
});
export function pluginStateRegisterIfAbsent(params: {
pluginId: string;
namespace: string;
key: string;
valueJson: string;
maxEntries: number;
ttlMs?: number;
}): boolean {
try {
return runWriteTransaction("register", (store) => {
const now = Date.now();
const expiresAt = params.ttlMs == null ? null : now + params.ttlMs;
store.statements.pruneExpiredNamespace.run(params.pluginId, params.namespace, now);
const result = store.statements.insertEntryIfAbsent.run({
plugin_id: params.pluginId,
namespace: params.namespace,
entry_key: params.key,
value_json: params.valueJson,
created_at: now,
expires_at: expiresAt,
});
if (result.changes === 0) {
return false;
}
enforcePostRegisterLimits({
store,
pluginId: params.pluginId,
namespace: params.namespace,
maxEntries: params.maxEntries,
now,
});
return true;
});
} catch (error) {
throw wrapPluginStateError(