Mirrors/sqlx
2
0
Fork 0
mirror of https://github.com/launchbadge/sqlx synced 2024-11-10 06:24:16 +00:00
Code Issues Projects Releases Packages Wiki Activity

feat(macros): implement type_override and column_override from sqlx.toml

Browse source
This commit is contained in:
Austin Bonander 2024-09-18 18:17:43 -07:00
parent 3e812b6c08
commit 572918a150
1 changed files with 19 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
sqlx-macros-core/src/query/output.rs
View file

@ -2,7 +2,7 @@ use proc_macro2::{Ident, Span, TokenStream};
use quote::{quote, ToTokens, TokenStreamExt}; use quote::{quote, ToTokens, TokenStreamExt};
use syn::Type; use syn::Type;
use sqlx_core::column::Column; use sqlx_core::column::{Column, ColumnOrigin};
use sqlx_core::describe::Describe; use sqlx_core::describe::Describe;
use crate::database::DatabaseExt; use crate::database::DatabaseExt;
@ -12,6 +12,8 @@ use sqlx_core::type_checking::TypeChecking;
use std::fmt::{self, Display, Formatter}; use std::fmt::{self, Display, Formatter};
use syn::parse::{Parse, ParseStream}; use syn::parse::{Parse, ParseStream};
use syn::Token; use syn::Token;
use sqlx_core::config::Config;
use sqlx_core::type_info::TypeInfo;
pub struct RustColumn { pub struct RustColumn {
pub(super) ident: Ident, pub(super) ident: Ident,
@ -229,8 +231,24 @@ pub fn quote_query_scalar<DB: DatabaseExt>(
} }
fn get_column_type<DB: DatabaseExt>(i: usize, column: &DB::Column) -> TokenStream { fn get_column_type<DB: DatabaseExt>(i: usize, column: &DB::Column) -> TokenStream {
if let ColumnOrigin::Table(origin) = column.origin() {
if let Some(column_override) = Config::from_crate()
.macros
.column_override(&origin.table, &origin.name)
{
return column_override.parse().unwrap();
}
}
let type_info = column.type_info(); let type_info = column.type_info();
if let Some(type_override) = Config::from_crate()
.macros
.type_override(type_info.name())
{
return type_override.parse().unwrap();
}
<DB as TypeChecking>::return_type_for_id(type_info).map_or_else( <DB as TypeChecking>::return_type_for_id(type_info).map_or_else(
|| { || {
let message = let message =