mirror of
https://github.com/SinTan1729/chhoto-url
synced 2024-12-28 08:28:36 -06:00
24 lines
561 B
Rust
24 lines
561 B
Rust
|
use sqlite::{open, Row};
|
||
|
|
||
|
pub fn find_url(shortlink: &str) -> String {
|
||
|
let db = open("./urls.sqlite").expect("Unable to open database!");
|
||
|
|
||
|
let query = "SELECT long_url FROM urls WHERE short_url = ?";
|
||
|
|
||
|
let statement: Vec<Row> = db
|
||
|
.prepare(query)
|
||
|
.unwrap()
|
||
|
.into_iter()
|
||
|
.bind((1, shortlink))
|
||
|
.unwrap()
|
||
|
.map(|row| row.unwrap())
|
||
|
.collect();
|
||
|
|
||
|
let mut longlink = "";
|
||
|
if statement.len() == 1 {
|
||
|
longlink = statement[0].read::<&str, _>("long_url");
|
||
|
}
|
||
|
|
||
|
String::from(longlink)
|
||
|
}
|