From a992bc0f75e4b72455f78c3d58621f11dc601348 Mon Sep 17 00:00:00 2001
From: Caleb Cox <canac@users.noreply.github.com>
Date: Mon, 30 Sep 2024 20:35:05 -0500
Subject: [PATCH] Use check

---
 database/src/sqlite_backend.rs | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/database/src/sqlite_backend.rs b/database/src/sqlite_backend.rs
index 0ac6180..001adac 100644
--- a/database/src/sqlite_backend.rs
+++ b/database/src/sqlite_backend.rs
@@ -101,22 +101,27 @@ impl SqliteBackend {
                 ColumnDef::new(MessageIden::Mailbox)
                     .string()
                     .not_null()
-                    .extra(String::from("CHECK (LENGTH(mailbox) > 0)")),
+                    .check(Expr::expr(Func::char_length(Expr::col(MessageIden::Mailbox))).gt(0)),
             )
             .col(
                 ColumnDef::new(MessageIden::Content)
                     .string()
                     .not_null()
-                    .extra(String::from("CHECK (LENGTH(content) > 0)")),
+                    .check(Expr::expr(Func::char_length(Expr::col(MessageIden::Content))).gt(0)),
             )
             .col(
                 ColumnDef::new(MessageIden::State)
                     .integer()
                     .not_null()
                     .default(Value::Int(Some(0)))
-                    .extra(String::from("CHECK (state >= 0 AND state <= 2)")),
+                    .check(
+                        Expr::col(MessageIden::State)
+                            .gte(0)
+                            .and(Expr::col(MessageIden::State).lte(2)),
+                    ),
             )
             .build(SqliteQueryBuilder);
+        println!("{sql}");
         query(&sql)
             .execute(&self.pool)
             .await