added replace to safe storage

Author: fnc12

Sources/SQLiteORM/BaseStorage/BaseStorage+CRUD.swift

@@ -252,4 +252,56 @@ extension BaseStorage {
  return .failure(error)
  }
  }
+ 
+ func replaceInternal<T>(_ object: T) -> Result<Void, Error> {
+ guard let anyTable = self.tables.first(where: { $0.type == T.self }) else {
+ return .failure(Error.typeIsNotMapped)
+ }
+ var sql = "REPLACE INTO \(anyTable.name) ("
+ let columnsCount = anyTable.columns.count
+ for (columnIndex, column) in anyTable.columns.enumerated() {
+ sql += "\"\(column.name)\""
+ if columnIndex < columnsCount - 1 {
+ sql += ", "
+ }
+ }
+ sql += ") VALUES ("
+ for columnIndex in 0..<columnsCount {
+ sql += "?"
+ if columnIndex < columnsCount - 1 {
+ sql += ", "
+ }
+ }
+ sql += ")"
+ let connectionRefResult = self.connection.createConnectionRef()
+ switch connectionRefResult {
+ case .success(let connectionRef):
+ let prepareResult = connectionRef.prepare(sql: sql)
+ switch prepareResult {
+ case .success(let statement):
+ let table = anyTable as! Table<T>
+ let bindResult = table.bind(columnBinder: statement, object: object, apiProvider: self.apiProvider)
+ switch bindResult {
+ case .success(var resultCode):
+ guard resultCode == apiProvider.SQLITE_OK else {
+ let errorString = connectionRef.errorMessage
+ return .failure(Error.sqliteError(code: resultCode, text: errorString))
+ }
+ resultCode = statement.step()
+ guard apiProvider.SQLITE_DONE == resultCode else {
+ let errorString = connectionRef.errorMessage
+ return .failure(Error.sqliteError(code: resultCode, text: errorString))
+ }
+ return .success(())
+ case .failure(let error):
+ return .failure(error)
+ }
+ case .failure(let error):
+ return .failure(error)
+ }
+ case .failure(let error):
+ return .failure(error)
+ }
+ 
+ }
 }

Sources/SQLiteORM/SafeStorage/SafeStorage+CRUD.swift

@@ -16,4 +16,8 @@ extension SafeStorage {
  public func insert<T>(_ object: T) -> Result<Int64, Error> {
  return self.insertInternal(object)
  }
+ 
+ public func replace<T>(_ object: T) -> Result<Void, Error> {
+ return self.replaceInternal(object)
+ }
 }

Sources/SQLiteORM/Storage/Storage+CRUD.swift

@@ -42,37 +42,12 @@ extension Storage {
  }
 
  public func replace<T>(_ object: T) throws {
- guard let anyTable = self.tables.first(where: { $0.type == T.self }) else {
- throw Error.typeIsNotMapped
- }
- var sql = "REPLACE INTO \(anyTable.name) ("
- let columnsCount = anyTable.columns.count
- for (columnIndex, column) in anyTable.columns.enumerated() {
- sql += "\"\(column.name)\""
- if columnIndex < columnsCount - 1 {
- sql += ", "
- }
- }
- sql += ") VALUES ("
- for columnIndex in 0..<columnsCount {
- sql += "?"
- if columnIndex < columnsCount - 1 {
- sql += ", "
- }
- }
- sql += ")"
- let connectionRef = try ConnectionRef(connection: self.connection)
- let statement = try connectionRef.prepare(sql: sql)
- let table = anyTable as! Table<T>
- var resultCode = try table.bind(columnBinder: statement, object: object, apiProvider: self.apiProvider)
- guard resultCode == apiProvider.SQLITE_OK else {
- let errorString = connectionRef.errorMessage
- throw Error.sqliteError(code: resultCode, text: errorString)
- }
- resultCode = statement.step()
- guard apiProvider.SQLITE_DONE == resultCode else {
- let errorString = connectionRef.errorMessage
- throw Error.sqliteError(code: resultCode, text: errorString)
+ let replaceResult = self.replaceInternal(object)
+ switch replaceResult {
+ case .success():
+ return
+ case .failure(let error):
+ throw error
  }
  }
 }

Sources/SQLiteORM/Table.swift

@@ -28,7 +28,7 @@ public class Table<T>: AnyTable {
  return .success(result)
  }
 
- func bind(columnBinder: ColumnBinder, object: T, apiProvider: SQLiteApiProvider) throws -> Int32 {
+ func bind(columnBinder: ColumnBinder, object: T, apiProvider: SQLiteApiProvider) -> Result<Int32, Error> {
  var result = Int32(0)
  for (columnIndex, anyColumn) in self.columns.enumerated() {
  let binder = BinderImpl(columnIndex: columnIndex + 1, columnBinder: columnBinder)
@@ -40,9 +40,9 @@ public class Table<T>: AnyTable {
  break
  }
  case .failure(let error):
- throw error
+ return .failure(error)
  }
  }
- return result
+ return .success(result)
  }
 }