SinTan1729/chhoto-url
SinTan1729/chhoto-url
1
0
Fork 0
mirror of https://github.com/SinTan1729/chhoto-url synced 2025-04-28 11:56:52 -05:00
Code Issues Releases Activity

Merge pull request #53 from magnusja/main

Browse source 
Create UNIQUE INDEX on short URL
This commit is contained in:
Sayantan Santra 2025-04-19 20:07:01 -05:00 committed by GitHub
commit a170954232
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 21 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
actix/src/database.rs
View file

@ -99,5 +99,12 @@ pub fn open_db(path: String) -> Connection {
)
.expect("Unable to initialize empty database.");
// Create index on short_url for faster lookups
db.execute(
"CREATE UNIQUE INDEX IF NOT EXISTS idx_short_url ON urls (short_url)",
[],
)
.expect("Unable to create index on short_url.");
db
}

28
actix/src/utils.rs
View file

@ -63,8 +63,8 @@ pub fn is_api_ok(http: HttpRequest) -> Response {
// If the API key isn't set, but an API Key header is provided
if auth::api_header(&http).is_some() {
Response {
success: false,
error: true,
success: false,
error: true,
reason: "An API key was provided, but the 'api_key' environment variable is not configured in the Chhoto URL instance".to_string(),
pass: false
}
@ -80,7 +80,11 @@ pub fn is_api_ok(http: HttpRequest) -> Response {
}
// Request the DB for searching an URL
pub fn get_longurl(shortlink: String, db: &Connection, needhits: bool) -> (Option<String>, Option<i64>) {
pub fn get_longurl(
shortlink: String,
db: &Connection,
needhits: bool,
) -> (Option<String>, Option<i64>) {
if validate_link(&shortlink) {
database::find_url(shortlink.as_str(), db, needhits)
} else {
@ -123,18 +127,14 @@ pub fn add_link(req: String, db: &Connection) -> (bool, String) {
chunks.shortlink = gen_link(style, len);
}
if validate_link(chunks.shortlink.as_str())
&& get_longurl(chunks.shortlink.clone(), db, false).0.is_none()
{
(
database::add_link(chunks.shortlink.clone(), chunks.longlink, db),
chunks.shortlink,
)
if validate_link(chunks.shortlink.as_str()) {
if database::add_link(chunks.shortlink.clone(), chunks.longlink, db) {
(true, chunks.shortlink)
} else {
(false, String::from("Short URL is already in use!"))
}
} else {
(
false,
String::from("Short URL not valid or already in use!"),
)
(false, String::from("Short URL is not valid!"))
}
}