itsscb/sqlx
1
0
Fork 0
mirror of https://github.com/launchbadge/sqlx.git synced 2026-03-27 13:46:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add a Transaction type to simplify dealing with Transactions

Browse Source
This commit is contained in:
Ryan Leckey
2020-01-03 22:42:10 -08:00
parent 28ed854b03
commit b1a27ddac2
10 changed files with 272 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
sqlx-macros/src/lib.rs
View File

@@ -29,14 +29,14 @@ use query_macros::*;
macro_rules! async_macro (
($db:ident => $expr:expr) => {{
let res: Result<proc_macro2::TokenStream> = task::block_on(async {
use sqlx::Connection;
use sqlx::Connect;
let db_url = Url::parse(&dotenv::var("DATABASE_URL").map_err(|_| "DATABASE_URL not set")?)?;
match db_url.scheme() {
#[cfg(feature = "postgres")]
"postgresql" | "postgres" => {
let $db = sqlx::postgres::PgConnection::open(db_url.as_str())
let $db = sqlx::postgres::PgConnection::connect(db_url.as_str())
.await
.map_err(|e| format!("failed to connect to database: {}", e))?;
@@ -50,7 +50,7 @@ macro_rules! async_macro (
).into()),
#[cfg(feature = "mysql")]
"mysql" | "mariadb" => {
let $db = sqlx::mysql::MySqlConnection::open(db_url.as_str())
let $db = sqlx::mysql::MySqlConnection::connect(db_url.as_str())
.await
.map_err(|e| format!("failed to connect to database: {}", e))?;