登录/ 注册
 找回密码
 立即注册
关闭

CSDN热搜

程序园 精品问答 技术交流 资源下载
返回列表 发新帖
首页 › 资源区 › 代码 ›【Swift】GRDB数据库本地存储聊天记录

【Swift】GRDB数据库本地存储聊天记录

梅克 2025-6-4 20:09:00
　　
  1. //
  2. //  DataManager.swift
  3. //  GRDBDemo
  4. //
  5. //  Created by Apple on 2021/4/21.
  6. //
  9. import GRDB
  11. struct DataBaseName {
  12.     /// 数据库名字
  13.     static let test = "conversation.db"
  14. }
  16. /// 数据库表名
  17. struct TableName {
  18.    
  19.     static let message = "ChatMessage"
  20. }
  22. /// 数据库连接
  23. class DBManager: NSObject {
  24.     /// 数据库路径
  25.     private static var dbPath: String = {
  26.         // 获取工程内容数据库名字
  27.         let filePath: String = NSSearchPathForDirectoriesInDomains(FileManager.SearchPathDirectory.documentDirectory, FileManager.SearchPathDomainMask.userDomainMask, true).first!.appending("/\(DataBaseName.test)")
  28.         
  29.         //print("数据库地址：", filePath as Any)
  30.         return filePath
  31.     }()
  32.    
  33.     /// 数据库配置
  34.     private static var configuration: Configuration = {
  35.         // 配置
  36.         var configuration = Configuration()
  37.         // 设置超时
  38.         configuration.busyMode = Database.BusyMode.timeout(5.0)
  39.         // 试图访问锁着的数据
  40.         //configuration.busyMode = Database.BusyMode.immediateError
  41.         
  42.         return configuration
  43.     }()
  44.    
  45.     // MARK: 创建数据 多线程
  46.     /// 数据库 用于多线程事务处理
  47.     static var dbQueue: DatabaseQueue = {
  48.         // 创建数据库
  49.         let db = try! DatabaseQueue(path: DBManager.dbPath, configuration: DBManager.configuration)
  50.         db.releaseMemory()
  51.         // 设备版本
  52.         return db
  53.     }()
  54. }
复制代码
  1. //
  2. //  ChatMessage.swift
  3. //  GRDBDemo
  4. //
  5. //  Created by Apple on 2021/4/21.
  6. //
  8. import Foundation
  9. import GRDB
  11. /// 聊天消息类
  12. struct ChatMessage: Codable {
  13.    
  14.     var messageId : String?
  15.     var messageType : String?
  16.     var messageContent : String?
  17.     var senderId : String?
  18.     var targetId : String?
  19.    
  20.     private enum Columns: String, CodingKey, ColumnExpression {
  21.         
  22.         case messageId
  23.         case messageType
  24.         case messageContent
  25.         case senderId
  26.         case targetId
  27.     }
  28. }
  30. extension ChatMessage: MutablePersistableRecord, FetchableRecord {
  31.     /// 获取数据库对象
  32.     private static let dbQueue: DatabaseQueue = DBManager.dbQueue
  33.    
  34.     //MARK: 创建
  35.     /// 创建数据库
  36.     private static func createTable() -> Void {
  37.         try! self.dbQueue.inDatabase { (db) -> Void in
  38.             // 判断是否存在数据库
  39.             if try db.tableExists(TableName.message) {
  40.                 debugPrint("表已经存在")
  41.                 return
  42.             }
  43.             // 创建数据库表
  44.             try db.create(table: TableName.message, temporary: false, ifNotExists: true, body: { (t) in
  45.                 t.column(Columns.messageId.rawValue, Database.ColumnType.text)
  46.                 t.column(Columns.messageType.rawValue, Database.ColumnType.text)
  47.                 t.column(Columns.messageContent.rawValue, Database.ColumnType.text)
  48.                 t.column(Columns.senderId.rawValue, Database.ColumnType.text)
  49.                 t.column(Columns.targetId.rawValue, Database.ColumnType.text)
  50.             })
  51.         }
  52.     }
  53.    
  54.     //MARK: 插入
  55.     /// 插入单个数据
  56.     static func insert(message: ChatMessage) -> Void {
  57.         // 判断是否存在
  58.         guard ChatMessage.query(messageId: message.messageId!) == nil else {
  59.             debugPrint("插入消息 内容重复")
  60.             // 更新
  61.             self.update(message: message)
  62.             return
  63.         }
  64.         
  65.         // 创建表
  66.         self.createTable()
  67.         // 事务
  68.         try! self.dbQueue.inTransaction { (db) -> Database.TransactionCompletion in
  69.             do {
  70.                 var messageTemp = message
  71.                 // 插入到数据库
  72.                 try messageTemp.insert(db)
  73.                 return Database.TransactionCompletion.commit
  74.             } catch {
  75.                 return Database.TransactionCompletion.rollback
  76.             }
  77.         }
  78.     }
  79.    
  80.     //MARK: 查询一条记录
  81.     static func query(messageId: String) -> ChatMessage? {
  82.         // 创建数据库
  83.         self.createTable()
  84.         // 返回查询结果
  85.         return try! self.dbQueue.unsafeRead({ (db) -> ChatMessage? in
  86.             return try ChatMessage.filter(Column(Columns.messageId.rawValue) == messageId).fetchOne(db)
  87.         })
  88.     }
  89.    
  90.     //MARK:查询与某人聊天的多条记录 - 从第几页开始
  91.     static func query(userId:String,page:Int) -> [ChatMessage] {
  92.         // 创建数据库
  93.         self.createTable()
  94.         
  95.         return try! self.dbQueue.unsafeRead({ (db) -> [ChatMessage] in
  96.             return try ChatMessage.fetchAll(db, sql: "Select * from ChatMessage where senderId = '\(userId)' or targetId = '\(userId)' limit \(20 * page),20")
  97.         })
  98.     }
  99.    
  100.     /// 查询所有
  101.     static func queryAll() -> [ChatMessage] {
  102.         // 创建数据库
  103.         self.createTable()
  104.         // 返回查询结果
  105.         return try! self.dbQueue.unsafeRead({ (db) -> [ChatMessage] in
  106.             return try ChatMessage.fetchAll(db)
  107.         })
  108.     }
  109.    
  110.     //MARK: 更新
  111.     /// 更新
  112.     static func update(message: ChatMessage) -> Void {
  113.         /// 创建数据库表
  114.         self.createTable()
  115.         // 事务 更新场景
  116.         try! self.dbQueue.inTransaction { (db) -> Database.TransactionCompletion in
  117.             do {
  118.                 // 赋值
  119.                 try message.update(db)
  120.                 return Database.TransactionCompletion.commit
  121.             } catch {
  122.                 return Database.TransactionCompletion.rollback
  123.             }
  124.         }
  125.     }
  126.    
  127.     //MARK: 删除
  128.     /// 根据messageId删除聊天记录
  129.     static func delete(messageId: String) -> Void {
  130.         // 查询
  131.         guard let message = self.query(messageId: messageId) else {
  132.             return
  133.         }
  134.         // 删除
  135.         self.delete(message: message)
  136.     }
  137.    
  138.     /// 删除单个聊天信息
  139.     static func delete(message: ChatMessage) -> Void {
  140.         // 是否有数据库表
  141.         self.createTable()
  142.         // 事务
  143.         try! self.dbQueue.inTransaction { (db) -> Database.TransactionCompletion in
  144.             do {
  145.                 // 删除数据
  146.                 try message.delete(db)
  147.                 return Database.TransactionCompletion.commit
  148.             } catch {
  149.                 return Database.TransactionCompletion.rollback
  150.             }
  151.         }
  152.     }
  153. }
复制代码
 　　使用示例：
  1. override func viewWillAppear(_ animated: Bool) {
  2.         super.viewWillAppear(animated)
  3.    
  4.         let user1 = ChatMessage(messageId: "1010", messageType: "text", messageContent: "12345", senderId: "1", targetId: "2")
  5.         ChatMessage.insert(message: user1)
  6.         
  7.         let user2 = ChatMessage(messageId: "1012", messageType: "text", messageContent: "12345", senderId: "123", targetId: "2")
  8.         ChatMessage.insert(message: user2)
  9.         
  10.         let user3 = ChatMessage(messageId: "1013", messageType: "text", messageContent: "12345", senderId: "1", targetId: "123")
  11.         ChatMessage.insert(message: user3)
  12.         
  13.         let user4 = ChatMessage(messageId: "1014", messageType: "text", messageContent: "12345", senderId: "1", targetId: "2")
  14.         ChatMessage.insert(message: user4)
  15.         
  16.         let user5 = ChatMessage(messageId: "1015", messageType: "text", messageContent: "12345", senderId: "1", targetId: "123")
  17.         ChatMessage.insert(message: user5)
  18.         
  19.         let user6 = ChatMessage(messageId: "1016", messageType: "text", messageContent: "12345", senderId: "123", targetId: "2")
  20.         ChatMessage.insert(message: user6)
  21.         
  22.         let user7 = ChatMessage(messageId: "1017", messageType: "text", messageContent: "12345", senderId: "1", targetId: "2")
  23.         ChatMessage.insert(message: user7)
  24.         
  25.         let user8 = ChatMessage(messageId: "1018", messageType: "text", messageContent: "12345", senderId: "1", targetId: "123")
  26.         ChatMessage.insert(message: user8)
  27.         
  28.         let message1 = ChatMessage.query(userId: "123", page: 0)
  29.         print(message1)
  31.     }
复制代码
 
　　

来源：程序园用户自行投稿发布，如果侵权，请联系站长删除
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！

使用道具 举报
B Color Image Link Quote Code Smilies
您需要登录后才可以回帖 登录 | 立即注册
发帖
梅克
2025-6-4 20:09:00

0

粉丝关注

10

主题发布

板块介绍填写区域，请于后台编辑
微信扫一扫