salsa/salsa-2022-tests/tests/override_new_get_set.rs

//! Test that the `constructor` macro overrides
//! the `new` method's name and `get` and `set`
//! change the name of the getter and setter of the fields.
#![allow(warnings)]

use std::fmt::Display;

#[salsa::jar(db = Db)]
struct Jar(MyInput, MyInterned, MyTracked);

trait Db: salsa::DbWithJar<Jar> {}

#[salsa::input(jar = Jar, constructor = from_string)]
struct MyInput {
    #[get(text)]
    #[set(set_text)]
    field: String,
}

impl MyInput {
    pub fn new(db: &mut dyn Db, s: impl Display) -> MyInput {
        MyInput::from_string(db, s.to_string())
    }

    pub fn field(self, db: &dyn Db) -> String {
        self.text(db)
    }

    pub fn set_field(self, db: &mut dyn Db, id: String) {
        self.set_text(db).to(id);
    }
}

#[salsa::interned(constructor = from_string)]
struct MyInterned {
    #[get(text)]
    #[return_ref]
    field: String,
}

impl MyInterned {
    pub fn new(db: &dyn Db, s: impl Display) -> MyInterned {
        MyInterned::from_string(db, s.to_string())
    }

    pub fn field(self, db: &dyn Db) -> &str {
        &self.text(db)
    }
}

#[salsa::tracked(constructor = from_string)]
struct MyTracked {
    #[get(text)]
    field: String,
}

impl MyTracked {
    pub fn new(db: &dyn Db, s: impl Display) -> MyTracked {
        MyTracked::from_string(db, s.to_string())
    }

    pub fn field(self, db: &dyn Db) -> String {
        self.text(db)
    }
}

#[test]
fn execute() {
    #[salsa::db(Jar)]
    #[derive(Default)]
    struct Database {
        storage: salsa::Storage<Self>,
    }

    impl salsa::Database for Database {
        fn salsa_runtime(&self) -> &salsa::Runtime {
            self.storage.runtime()
        }

        fn salsa_runtime_mut(&mut self) -> &mut salsa::Runtime {
            self.storage.runtime_mut()
        }
    }

    impl Db for Database {}

    let mut db = Database::default();
}