rust/neon
1
0
Fork 0
mirror of https://github.com/neondatabase/neon.git synced 2026-01-15 09:22:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

pageserver/http: add /v1/deletion_queue/flush_execute

Browse Source 
This is principally for tesing, but might be useful in
the field if we want to e.g. flush a deletion queue
before running an external scrub tool
This commit is contained in:
John Spray
2023-08-14 16:22:07 +01:00
parent 5e2b8b376c
commit bae62916dc
2 changed files with 40 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
pageserver/src/http/openapi_spec.yml
View File

@@ -52,6 +52,21 @@ paths:
schema:
type: object
/v1/deletion_queue/flush_execute:
put:
description: Execute any deletions currently enqueued
security: []
responses:
"200":
description: |
Flush completed: deletion has been attempted for enqueued objects. This does not guarantee
that deletion executed successfully.
content:
application/json:
schema:
type: object
/v1/tenant/{tenant_id}:
parameters:
- name: tenant_id

25
pageserver/src/http/routes.rs
View File

@@ -1132,6 +1132,28 @@ async fn always_panic_handler(
json_response(StatusCode::NO_CONTENT, ())
}
async fn deletion_queue_flush_execute(
r: Request<Body>,
cancel: CancellationToken,
) -> Result<Response<Body>, ApiError> {
let state = get_state(&r);
if let None = state.remote_storage {
// Nothing to do if remote storage is disabled.
return json_response(StatusCode::OK, ());
}
let queue_client = state.deletion_queue.new_client();
tokio::select! {
_ = queue_client.flush_execute()=> {
json_response(StatusCode::OK, ())
},
_ = cancel.cancelled() => {
Err(ApiError::ShuttingDown)
}
}
}
async fn disk_usage_eviction_run(
mut r: Request<Body>,
_cancel: CancellationToken,
@@ -1456,6 +1478,9 @@ pub fn make_router(
.put("/v1/disk_usage_eviction/run", |r| {
api_handler(r, disk_usage_eviction_run)
})
.put("/v1/deletion_queue/flush_execute", |r| {
api_handler(r, deletion_queue_flush_execute)
})
.put("/v1/tenant/:tenant_id/break", |r| {
testing_api_handler("set tenant state to broken", r, handle_tenant_break)
})