Fix grdb create

This commit is contained in:
世界
2024-09-20 22:49:51 +08:00
parent 10c64390c3
commit de84957b21
3 changed files with 47 additions and 48 deletions
+7 -8
View File
@@ -1,13 +1,11 @@
import Foundation import Foundation
import GRDB import GRDB
actor Database { enum Database {
private static var writer: (any DatabaseWriter)? static let sharedWriter = makeShared()
static func sharedWriter() throws -> any DatabaseWriter { private static func makeShared() -> any DatabaseWriter {
if let writer { do {
return writer
}
try FileManager.default.createDirectory(at: FilePath.sharedDirectory, withIntermediateDirectories: true) try FileManager.default.createDirectory(at: FilePath.sharedDirectory, withIntermediateDirectories: true)
let database = try DatabasePool(path: FilePath.sharedDirectory.appendingPathComponent("settings.db").relativePath) let database = try DatabasePool(path: FilePath.sharedDirectory.appendingPathComponent("settings.db").relativePath)
var migrator = DatabaseMigrator().disablingDeferredForeignKeyChecks() var migrator = DatabaseMigrator().disablingDeferredForeignKeyChecks()
@@ -32,9 +30,10 @@ actor Database {
t.add(column: "autoUpdateInterval", .integer).notNull().defaults(to: 0) t.add(column: "autoUpdateInterval", .integer).notNull().defaults(to: 0)
} }
} }
try migrator.migrate(database) try migrator.migrate(database)
writer = database
return database return database
} catch {
fatalError(error.localizedDescription)
}
} }
} }
+14 -14
View File
@@ -4,37 +4,37 @@ import GRDB
public enum ProfileManager { public enum ProfileManager {
public nonisolated static func create(_ profile: Profile) async throws { public nonisolated static func create(_ profile: Profile) async throws {
profile.order = try await nextOrder() profile.order = try await nextOrder()
try await Database.sharedWriter().write { db in try await Database.sharedWriter.write { db in
try profile.insert(db, onConflict: .fail) try profile.insert(db, onConflict: .fail)
} }
} }
public nonisolated static func get(_ profileID: Int64) async throws -> Profile? { public nonisolated static func get(_ profileID: Int64) async throws -> Profile? {
try await Database.sharedWriter().read { db in try await Database.sharedWriter.read { db in
try Profile.fetchOne(db, id: profileID) try Profile.fetchOne(db, id: profileID)
} }
} }
public nonisolated static func get(by profileName: String) async throws -> Profile? { public nonisolated static func get(by profileName: String) async throws -> Profile? {
try await Database.sharedWriter().read { db in try await Database.sharedWriter.read { db in
try Profile.filter(Column("name") == profileName).fetchOne(db) try Profile.filter(Column("name") == profileName).fetchOne(db)
} }
} }
public nonisolated static func delete(_ profile: Profile) async throws { public nonisolated static func delete(_ profile: Profile) async throws {
_ = try await Database.sharedWriter().write { db in _ = try await Database.sharedWriter.write { db in
try profile.delete(db) try profile.delete(db)
} }
} }
public nonisolated static func delete(by id: Int64) async throws { public nonisolated static func delete(by id: Int64) async throws {
_ = try await Database.sharedWriter().write { db in _ = try await Database.sharedWriter.write { db in
try Profile.deleteOne(db, id: id) try Profile.deleteOne(db, id: id)
} }
} }
public nonisolated static func delete(_ profileList: [Profile]) async throws -> Int { public nonisolated static func delete(_ profileList: [Profile]) async throws -> Int {
try await Database.sharedWriter().write { db in try await Database.sharedWriter.write { db in
try Profile.deleteAll(db, keys: profileList.map { try Profile.deleteAll(db, keys: profileList.map {
["id": $0.id!] ["id": $0.id!]
}) })
@@ -42,20 +42,20 @@ public enum ProfileManager {
} }
public nonisolated static func delete(by id: [Int64]) async throws -> Int { public nonisolated static func delete(by id: [Int64]) async throws -> Int {
try await Database.sharedWriter().write { db in try await Database.sharedWriter.write { db in
try Profile.deleteAll(db, ids: id) try Profile.deleteAll(db, ids: id)
} }
} }
public nonisolated static func update(_ profile: Profile) async throws { public nonisolated static func update(_ profile: Profile) async throws {
_ = try await Database.sharedWriter().write { db in _ = try await Database.sharedWriter.write { db in
try profile.updateChanges(db) try profile.updateChanges(db)
} }
} }
public nonisolated static func update(_ profileList: [Profile]) async throws { public nonisolated static func update(_ profileList: [Profile]) async throws {
// TODO: batch update // TODO: batch update
try await Database.sharedWriter().write { db in try await Database.sharedWriter.write { db in
for profile in profileList { for profile in profileList {
try profile.updateChanges(db) try profile.updateChanges(db)
} }
@@ -63,25 +63,25 @@ public enum ProfileManager {
} }
public nonisolated static func list() async throws -> [Profile] { public nonisolated static func list() async throws -> [Profile] {
try await Database.sharedWriter().read { db in try await Database.sharedWriter.read { db in
try Profile.all().order(Column("order").asc).fetchAll(db) try Profile.all().order(Column("order").asc).fetchAll(db)
} }
} }
public nonisolated static func listRemote() async throws -> [Profile] { public nonisolated static func listRemote() async throws -> [Profile] {
try await Database.sharedWriter().read { db in try await Database.sharedWriter.read { db in
try Profile.filter(Column("type") == ProfileType.remote.rawValue).order(Column("order").asc).fetchAll(db) try Profile.filter(Column("type") == ProfileType.remote.rawValue).order(Column("order").asc).fetchAll(db)
} }
} }
public nonisolated static func listAutoUpdateEnabled() async throws -> [Profile] { public nonisolated static func listAutoUpdateEnabled() async throws -> [Profile] {
try await Database.sharedWriter().read { db in try await Database.sharedWriter.read { db in
try Profile.filter(Column("autoUpdate") == true).order(Column("order").asc).fetchAll(db) try Profile.filter(Column("autoUpdate") == true).order(Column("order").asc).fetchAll(db)
} }
} }
public nonisolated static func nextID() async throws -> Int64 { public nonisolated static func nextID() async throws -> Int64 {
try await Database.sharedWriter().read { db in try await Database.sharedWriter.read { db in
if let lastProfile = try Profile.select(Column("id")).order(Column("id").desc).fetchOne(db) { if let lastProfile = try Profile.select(Column("id")).order(Column("id").desc).fetchOne(db) {
return lastProfile.id! + 1 return lastProfile.id! + 1
} else { } else {
@@ -91,7 +91,7 @@ public enum ProfileManager {
} }
private nonisolated static func nextOrder() async throws -> UInt32 { private nonisolated static func nextOrder() async throws -> UInt32 {
try await Database.sharedWriter().read { db in try await Database.sharedWriter.read { db in
try UInt32(Profile.fetchCount(db)) try UInt32(Profile.fetchCount(db))
} }
} }
@@ -37,7 +37,7 @@ extension SharedPreferences {
} }
private nonisolated static func read<T: Codable>(_ name: String) async throws -> T? { private nonisolated static func read<T: Codable>(_ name: String) async throws -> T? {
guard let item = try await (Database.sharedWriter().read { db in guard let item = try await (Database.sharedWriter.read { db in
try Item.fetchOne(db, id: name) try Item.fetchOne(db, id: name)
}) })
else { else {
@@ -48,12 +48,12 @@ extension SharedPreferences {
private nonisolated static func write(_ name: String, _ value: (some Codable)?) async throws { private nonisolated static func write(_ name: String, _ value: (some Codable)?) async throws {
if value == nil { if value == nil {
_ = try await Database.sharedWriter().write { db in _ = try await Database.sharedWriter.write { db in
try Item.deleteOne(db, id: name) try Item.deleteOne(db, id: name)
} }
} else { } else {
let data = try BinaryEncoder().encode(value) let data = try BinaryEncoder().encode(value)
try await Database.sharedWriter().write { db in try await Database.sharedWriter.write { db in
try Item(name: name, data: data).insert(db) try Item(name: name, data: data).insert(db)
} }
} }