mirror of
https://github.com/Phundrak/georm.git
synced 2025-08-30 22:25:35 +00:00
Lucien Cartier-Tilet
3307aa679d
This commit introduces support for PostgreSQL generated columns by adding two new field attributes to the `Georm` derive macro: `#[georm(generated)]` and `#[georm(generated_always)]`. The `#[georm(generated_always)]` attribute is for fields that are always generated by the database, such as `GENERATED ALWAYS AS IDENTITY` columns or columns with a `GENERATED ALWAYS AS (expression) STORED` clause. These fields are now excluded from `INSERT` and `UPDATE` statements, preventing accidental writes and ensuring data integrity at compile time. The `#[georm(generated)]` attribute is for fields that have a default value generated by the database but can also be manually overridden, such as `GENERATED BY DEFAULT AS IDENTITY` columns. These fields behave similarly to `#[georm(defaultable)]` fields, allowing them to be omitted from `INSERT` statements to use the database-generated value. For now, the behaviour is the same between `#[georm(generated)]` and `#[georm(defaultable)]`, but the addition of the former now will be useful for future features. Key changes: - Added `generated` and `generated_always` attributes to `GeormFieldAttributes`. - Introduced `GeneratedType` enum in the IR to represent the different generation strategies. - Modified the `create` and `update` query generation to exclude fields marked with `#[georm(generated_always)]`. - Integrated `#[georm(generated)]` fields with the existing defaultable struct logic. - Added validation to prevent conflicting attribute usage, namely `#[georm(generated)]` and `#[georm(generated_always)]` on the same field. Implements #3
65 lines
2.1 KiB
Rust
65 lines
2.1 KiB
Rust
use georm::{Defaultable, Georm};
|
|
use sqlx::types::BigDecimal;
|
|
|
|
mod models;
|
|
use models::{Product, ProductDefault};
|
|
|
|
#[sqlx::test()]
|
|
async fn create_without_generated_values(pool: sqlx::PgPool) -> sqlx::Result<()> {
|
|
let base = ProductDefault {
|
|
name: "Desktop".to_owned(),
|
|
price: BigDecimal::from(2000),
|
|
discount_percent: 5,
|
|
sku_number: None,
|
|
};
|
|
let result = base.create(&pool).await?;
|
|
assert_eq!(BigDecimal::from(1900), result.final_price.unwrap());
|
|
Ok(())
|
|
}
|
|
|
|
#[sqlx::test()]
|
|
async fn create_with_manual_generated_value(pool: sqlx::PgPool) -> sqlx::Result<()> {
|
|
let base = ProductDefault {
|
|
name: "Monitor".to_owned(),
|
|
price: BigDecimal::from(750),
|
|
discount_percent: 10,
|
|
sku_number: Some(12345),
|
|
};
|
|
let result = base.create(&pool).await?;
|
|
assert_eq!(12345, result.sku_number);
|
|
assert_eq!("Monitor", result.name);
|
|
Ok(())
|
|
}
|
|
|
|
#[sqlx::test(fixtures("generated"))]
|
|
async fn update_does_not_change_generated_always_field(pool: sqlx::PgPool) -> sqlx::Result<()> {
|
|
let products = Product::find_all(&pool).await?;
|
|
dbg!(&products);
|
|
let mut product = products.first().unwrap().clone();
|
|
let original_final_price = product.clone().final_price;
|
|
product.name = "Gaming Laptop".to_owned();
|
|
product.final_price = Some(BigDecimal::from(1000000));
|
|
dbg!(&product);
|
|
let updated = product.update(&pool).await?;
|
|
assert_eq!(original_final_price, updated.final_price);
|
|
assert_eq!("Gaming Laptop", updated.name);
|
|
Ok(())
|
|
}
|
|
|
|
#[sqlx::test(fixtures("generated"))]
|
|
async fn upsert_handles_generated_fields(pool: sqlx::PgPool) -> sqlx::Result<()> {
|
|
let product = Product::find_by_name("Laptop".to_owned(), &pool).await?;
|
|
let mut modified_product = product.clone();
|
|
modified_product.price = BigDecimal::from(1200);
|
|
|
|
let upserted = modified_product.create_or_update(&pool).await?;
|
|
|
|
// price is updated
|
|
assert_eq!(upserted.price, BigDecimal::from(1200));
|
|
// final_price is re-calculated by the DB
|
|
// 1200 * (1 - 10 / 100.0) = 1080
|
|
assert_eq!(BigDecimal::from(1080), upserted.final_price.unwrap());
|
|
|
|
Ok(())
|
|
}
|