Cancel responses on startup

This commit is contained in:
Gregory Schier
2023-11-06 13:06:15 -08:00
parent 655f5a8eed
commit 6ccc42dc3f
7 changed files with 48 additions and 25 deletions

View File

@@ -764,6 +764,16 @@
}, },
"query": "\n SELECT\n id,\n model,\n workspace_id,\n folder_id,\n created_at,\n updated_at,\n name,\n url,\n method,\n body,\n body_type,\n authentication AS \"authentication!: Json<HashMap<String, JsonValue>>\",\n authentication_type,\n sort_priority,\n headers AS \"headers!: sqlx::types::Json<Vec<HttpRequestHeader>>\"\n FROM http_requests\n WHERE workspace_id = ?\n " "query": "\n SELECT\n id,\n model,\n workspace_id,\n folder_id,\n created_at,\n updated_at,\n name,\n url,\n method,\n body,\n body_type,\n authentication AS \"authentication!: Json<HashMap<String, JsonValue>>\",\n authentication_type,\n sort_priority,\n headers AS \"headers!: sqlx::types::Json<Vec<HttpRequestHeader>>\"\n FROM http_requests\n WHERE workspace_id = ?\n "
}, },
"ac1b4ffbd98b67f0a1a74e3525387d679dd6f44c561d55c7bbea747053e53671": {
"describe": {
"columns": [],
"nullable": [],
"parameters": {
"Right": 0
}
},
"query": "\n UPDATE http_responses\n SET (elapsed, status_reason) = (-1, 'Cancelled')\n WHERE elapsed = 0;\n "
},
"aeb0712785a9964d516dc8939bc54aa8206ad852e608b362d014b67a0f21b0ed": { "aeb0712785a9964d516dc8939bc54aa8206ad852e608b362d014b67a0f21b0ed": {
"describe": { "describe": {
"columns": [], "columns": [],

View File

@@ -77,17 +77,10 @@ async fn send_ephemeral_request(
let response = models::HttpResponse::default(); let response = models::HttpResponse::default();
let environment_id2 = environment_id.unwrap_or("n/a").to_string(); let environment_id2 = environment_id.unwrap_or("n/a").to_string();
request.id = "".to_string(); request.id = "".to_string();
return actually_send_ephemeral_request( return actually_send_request(request, &response, &environment_id2, &app_handle, pool).await;
request,
&response,
&environment_id2,
&app_handle,
pool,
)
.await;
} }
async fn actually_send_ephemeral_request( async fn actually_send_request(
request: models::HttpRequest, request: models::HttpRequest,
response: &models::HttpResponse, response: &models::HttpResponse,
environment_id: &str, environment_id: &str,
@@ -302,7 +295,7 @@ async fn send_request(
let pool2 = pool.clone(); let pool2 = pool.clone();
tokio::spawn(async move { tokio::spawn(async move {
actually_send_ephemeral_request(req, &response2, &environment_id2, &app_handle2, &pool2) actually_send_request(req, &response2, &environment_id2, &app_handle2, &pool2)
.await .await
.expect("Failed to send request"); .expect("Failed to send request");
}); });
@@ -317,6 +310,7 @@ async fn response_err(
pool: &Pool<Sqlite>, pool: &Pool<Sqlite>,
) -> Result<models::HttpResponse, String> { ) -> Result<models::HttpResponse, String> {
let mut response = response.clone(); let mut response = response.clone();
response.elapsed = -1;
response.error = Some(error.clone()); response.error = Some(error.clone());
response = models::update_response_if_id(&response, pool) response = models::update_response_if_id(&response, pool)
.await .await
@@ -744,6 +738,8 @@ fn main() {
.expect("Failed to migrate database"); .expect("Failed to migrate database");
app.manage(m); app.manage(m);
let _ = models::cancel_pending_responses(&pool).await;
// TODO: Move this somewhere better // TODO: Move this somewhere better
match app.get_cli_matches() { match app.get_cli_matches() {
Ok(matches) => { Ok(matches) => {

View File

@@ -600,6 +600,19 @@ pub async fn create_response(
get_response(&id, pool).await get_response(&id, pool).await
} }
pub async fn cancel_pending_responses(pool: &Pool<Sqlite>) -> Result<(), sqlx::Error> {
sqlx::query!(
r#"
UPDATE http_responses
SET (elapsed, status_reason) = (-1, 'Cancelled')
WHERE elapsed = 0;
"#,
)
.execute(pool)
.await?;
Ok(())
}
pub async fn update_response_if_id( pub async fn update_response_if_id(
response: &HttpResponse, response: &HttpResponse,
pool: &Pool<Sqlite>, pool: &Pool<Sqlite>,

View File

@@ -48,7 +48,6 @@ export function RequestActionsDropdown({ requestId, children }: Props) {
variant: 'danger', variant: 'danger',
leftSlot: <Icon icon="trash" />, leftSlot: <Icon icon="trash" />,
}, },
{ type: 'separator', label: 'Yaak Settings' },
]} ]}
> >
{children} {children}

View File

@@ -549,19 +549,10 @@ const SidebarItem = forwardRef(function SidebarItem(
{itemModel === 'folder' && ( {itemModel === 'folder' && (
<Dropdown <Dropdown
items={[ items={[
{
key: 'createRequest',
label: 'New Request',
onSelect: () => createRequest.mutate({ folderId: itemId, sortPriority: -1 }),
},
{
key: 'createFolder',
label: 'New Folder',
onSelect: () => createFolder.mutate({ folderId: itemId, sortPriority: -1 }),
},
{ {
key: 'sendAll', key: 'sendAll',
label: 'Send All', label: 'Send All',
leftSlot: <Icon icon="paperPlane" />,
onSelect: () => { onSelect: () => {
for (const { item } of child.children) { for (const { item } of child.children) {
if (item.model === 'http_request') { if (item.model === 'http_request') {
@@ -570,13 +561,27 @@ const SidebarItem = forwardRef(function SidebarItem(
} }
}, },
}, },
{ type: 'separator' }, { type: 'separator', label: itemName },
{ {
key: 'deleteFolder', key: 'deleteFolder',
label: 'Delete', label: 'Delete',
variant: 'danger', variant: 'danger',
leftSlot: <Icon icon="trash" />,
onSelect: () => deleteRequest.mutate(), onSelect: () => deleteRequest.mutate(),
}, },
{ type: 'separator' },
{
key: 'createRequest',
label: 'New Request',
leftSlot: <Icon icon="plus" />,
onSelect: () => createRequest.mutate({ folderId: itemId, sortPriority: -1 }),
},
{
key: 'createFolder',
label: 'New Folder',
leftSlot: <Icon icon="plus" />,
onSelect: () => createFolder.mutate({ folderId: itemId, sortPriority: -1 }),
},
]} ]}
> >
<IconButton <IconButton

View File

@@ -8,8 +8,8 @@ interface Props {
} }
export function StatusTag({ response, className, showReason }: Props) { export function StatusTag({ response, className, showReason }: Props) {
const { status, error } = response; const { status } = response;
const label = error ? 'ERR' : status; const label = status < 100 ? 'ERR' : status;
return ( return (
<span <span
className={classNames( className={classNames(

View File

@@ -87,7 +87,7 @@ export interface HttpResponse extends BaseModel {
} }
export function isResponseLoading(response: HttpResponse): boolean { export function isResponseLoading(response: HttpResponse): boolean {
return !(response.body || response.status || response.error); return response.elapsed === 0;
} }
export function modelsEq(a: Model, b: Model) { export function modelsEq(a: Model, b: Model) {