mirror of
https://github.com/mountain-loop/yaak.git
synced 2026-04-20 07:41:22 +02:00
Cancel responses on startup
This commit is contained in:
@@ -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": [],
|
||||||
|
|||||||
@@ -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) => {
|
||||||
|
|||||||
@@ -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>,
|
||||||
|
|||||||
@@ -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}
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
@@ -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(
|
||||||
|
|||||||
@@ -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) {
|
||||||
|
|||||||
Reference in New Issue
Block a user