readonly/mirror-chhoto-url
1
0
Fork 0
mirror of https://github.com/minoplhy/chhoto-url.git synced 2024-11-22 17:26:45 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use 404 response code when appropriate

Browse Source
This commit is contained in:
SinTan1729 2023-04-28 00:08:16 -05:00
parent ff4801a476
commit ca01676916
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
actix/src/main.rs
View File

@ -2,7 +2,9 @@ use actix_files::{Files, NamedFile};
use actix_session::{storage::CookieSessionStore, Session, SessionMiddleware}; use actix_session::{storage::CookieSessionStore, Session, SessionMiddleware};
use actix_web::{ use actix_web::{
cookie::Key, cookie::Key,
delete, get, middleware, post, delete, get,
http::StatusCode,
middleware, post,
web::{self, Redirect}, web::{self, Redirect},
App, HttpResponse, HttpServer, Responder, App, HttpResponse, HttpServer, Responder,
}; };
@ -67,7 +69,7 @@ async fn link_handler(shortlink: web::Path<String>, data: web::Data<AppState>) -
let shortlink_str = shortlink.to_string(); let shortlink_str = shortlink.to_string();
let longlink = utils::get_longurl(shortlink_str, &data.db); let longlink = utils::get_longurl(shortlink_str, &data.db);
if longlink == *"" { if longlink == *"" {
Redirect::to("/err/404") Redirect::to("/err/404").using_status_code(StatusCode::NOT_FOUND)
} else { } else {
database::add_hit(shortlink.as_str(), &data.db); database::add_hit(shortlink.as_str(), &data.db);
Redirect::to(longlink).permanent() Redirect::to(longlink).permanent()