create table if need before load and update database

sudopark · Jan 1, 2024 · d99a413 · d99a413
1 parent e2997a0
commit d99a413
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/Sources/SQLiteService/SQLiteDataBase.swift b/Sources/SQLiteService/SQLiteDataBase.swift
@@ -239,31 +239,39 @@ extension SQLiteDataBase {
 
     public func loadValue<T: Table, S: ScalarType>(_ query: SelectQuery<T>) throws -> S? {
 
+        try self.createTableOrNot(T.self)
         return try iterateDeserialize(query: query, deserialize: { $0.next() }).first
     }
 
     public func load<T: Table, R: RowValueType>(_ query: SelectQuery<T>) throws -> [R] {
 
+        try self.createTableOrNot(T.self)
         return try iterateDeserialize(query: query, deserialize: R.init)
     }
 
     public func load<T: Table>(_ table: T.Type, query: SelectQuery<T>) throws -> [T.EntityType] {
 
+        try self.createTableOrNot(T.self)
         return try iterateDeserialize(query: query, deserialize: T.EntityType.init)
     }
 
     public func load<T: Table, V>(_ query: SelectQuery<T>,
                                   mapping: (CursorIterator) throws -> V) throws -> [V] {
+
+        try self.createTableOrNot(T.self)
         return try iterateDeserialize(query: query, deserialize: mapping)
     }
 
     public func load<T: Table, R: RowValueType>(_ query: JoinQuery<T>) throws -> [R] {
 
+        try self.createTableOrNot(T.self)
         return try iterateDeserialize(query: query, deserialize: R.init)
     }
 
     public func load<T: Table, V>(_ query: JoinQuery<T>,
                                   mapping: (CursorIterator) throws -> V) throws -> [V] {
+
+        try self.createTableOrNot(T.self)
         return try iterateDeserialize(query: query, deserialize: mapping)
     }
 
@@ -302,6 +310,8 @@ extension SQLiteDataBase {
 
     public func update<T>(_ table: T.Type, query: UpdateQuery<T>) throws where T : Table {
 
+        try self.createTableOrNot(table)
+
         let stmt = try prepare(statement: query.asStatement())
 
         defer {

diff --git a/Tests/SQLiteServiceTests/SQLiteDatabaseTests.swift b/Tests/SQLiteServiceTests/SQLiteDatabaseTests.swift
@@ -135,7 +135,6 @@ extension SQLiteDatabaseTests {
     func testDatabase_insertModels() {
         // given
         self.openDataBase()
-        try? self.database.createTableOrNot(self.table)
         var inserted: Bool = false
 
         // when