Skip to content

Commit 995a991

Browse files
committed
chore: cleaned up the tryset_value macro
1 parent 45c486f commit 995a991

File tree

1 file changed

+22
-22
lines changed

1 file changed

+22
-22
lines changed

src/components/database.rs

Lines changed: 22 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -81,24 +81,24 @@ impl UserData for Database {
8181
let table = lua.create_table()?;
8282

8383
macro_rules! try_set_value {
84-
($table:expr, $lua:expr, $row:expr, $i:expr, $key:expr, $ty:ty) => {
85-
if let Ok(v) = $row.try_get::<$ty, _>($i) {
86-
$table.set($key, v)?;
84+
($i:expr, $key:expr, $ty:ty) => {
85+
if let Ok(v) = row.try_get::<$ty, _>($i) {
86+
table.set($key, v)?;
8787
continue;
88-
} else if let Ok(v) = $row.try_get::<Option<$ty>, _>($i) {
89-
$table.set($key, v)?;
88+
} else if let Ok(v) = row.try_get::<Option<$ty>, _>($i) {
89+
table.set($key, v)?;
9090
continue;
9191
}
9292
};
9393
}
9494

9595
macro_rules! try_set_lua_value {
96-
($table:expr, $lua:expr, $row:expr, $i:expr, $key:expr, $ty:ty) => {
97-
if let Ok(v) = $row.try_get::<$ty, _>($i) {
98-
$table.set($key, $lua.to_value(&v)?)?;
96+
($i:expr, $key:expr, $ty:ty) => {
97+
if let Ok(v) = row.try_get::<$ty, _>($i) {
98+
table.set($key, lua.to_value(&v)?)?;
9999
continue;
100-
} else if let Ok(v) = $row.try_get::<Option<$ty>, _>($i) {
101-
$table.set($key, $lua.to_value(&v)?)?;
100+
} else if let Ok(v) = row.try_get::<Option<$ty>, _>($i) {
101+
table.set($key, lua.to_value(&v)?)?;
102102
continue;
103103
}
104104
};
@@ -107,19 +107,19 @@ impl UserData for Database {
107107
for i in 0..row.len() {
108108
let key = row.column(i).name();
109109

110-
try_set_value!(table, lua, row, i, key, i64);
111-
try_set_value!(table, lua, row, i, key, i32);
112-
try_set_value!(table, lua, row, i, key, i16);
113-
try_set_value!(table, lua, row, i, key, i8);
114-
try_set_value!(table, lua, row, i, key, f32);
115-
try_set_value!(table, lua, row, i, key, f64);
116-
try_set_value!(table, lua, row, i, key, bool);
117-
try_set_value!(table, lua, row, i, key, String);
118-
try_set_value!(table, lua, row, i, key, Vec<u8>);
110+
try_set_value!(i, key, i64);
111+
try_set_value!(i, key, i32);
112+
try_set_value!(i, key, i16);
113+
try_set_value!(i, key, i8);
114+
try_set_value!(i, key, f32);
115+
try_set_value!(i, key, f64);
116+
try_set_value!(i, key, bool);
117+
try_set_value!(i, key, String);
118+
try_set_value!(i, key, Vec<u8>);
119119

120-
try_set_lua_value!(table, lua, row, i, key, serde_json::Value);
121-
try_set_lua_value!(table, lua, row, i, key, chrono::DateTime<chrono::Utc>);
122-
try_set_lua_value!(table, lua, row, i, key, uuid::Uuid);
120+
try_set_lua_value!(i, key, serde_json::Value);
121+
try_set_lua_value!(i, key, chrono::DateTime<chrono::Utc>);
122+
try_set_lua_value!(i, key, uuid::Uuid);
123123

124124
// fallback if all fail
125125
table.set(key, mlua::Value::Nil)?;

0 commit comments

Comments
 (0)