swift——从零开始自己的IM

之前旧项目里的聊天是集成的融云，然而有诸多不方便，很多需求都无法正常且简单的实现。而且使用聊天的人也并不多。。。
所以决定公司自己开发即时通讯（反正用户少=。=）

开发中是基于Socket.IO封装使用，写这篇文章时版本是13.0.0。后台是Node.js。这是Github地址 Socket.IO-Client-Swift

切记！好的产品就别指望了！多和后台的大兄弟交流，一定得有讨论连接，用户认证和消息文本格式的思维。

先上一个简单的思维导图

思维导图.png

自己补的，大概就是这么个流程。

下面开始上代码！（去除掉需求逻辑）

//连接状态
enum linkState: String {
    case no_connection = "未连接"
    case in_connection = "连接中"
    case connected =  "已连接"
    case connection_error = "连接错误"
}

class BSIM {
    
    /// 静态变量(常量)  static 修饰的静态方法不能被重写
    static let shared = BSIM()
    
    var manager:SocketManager?
    var socket:SocketIOClient?
    
    var onlineTimer:Timer?
    var onlineTimerNum = 30;
    
    /*后台状态保活*/
    let app = UIApplication.shared
    var taskID = UIBackgroundTaskIdentifier()
    var backTimer:Timer?
    
    var backTime = 0;
    //APP进入后台保活时间
    var backTimeOut = 60 * 3;
    
    // MARK:- 初始连接并进行认证
    func initAndConnect(server:String,userid:String,random:String,result:@escaping BSIMConnectResult){
        /// 一些判断 例如
        guard server != "" else {
            self.connectResult!(-9999, "服务器地址不能为空")
            return
        }
        
        /// 连接  第一步的HTTP请求                                                    /// 打印调试信息     使用websockets
        self.manager = SocketManager(socketURL: URL(string: self.server)!, config: [.log(false), .forceWebsockets(true)])
        self.socket = self.manager?.defaultSocket
        /// 监听事件
        self.socket?.on("你和后台约定好的字段", callback: { (data, ack) in
            /// 记得移除  避免重复监听
            self.socket?.off("某些监听")
            
            /// 初始化定时器 心跳包
            if(self.onlineTimer == nil){
                self.onlineTimer = Timer.scheduledTimer(timeInterval: TimeInterval(self.onlineTimerNum),target:self,selector:#selector(self.onlineTimerRun),userInfo:nil,repeats:true)
            }else{
                self.onlineTimer?.invalidate()
                self.onlineTimer = nil
                self.onlineTimer = Timer.scheduledTimer(timeInterval: TimeInterval(self.onlineTimerNum),target:self,selector:#selector(self.onlineTimerRun),userInfo:nil,repeats:true)
            }
            
            /// 更改连接状态
            self.linkStatePush(state: linkState.connected)
           
           /*
           有关认证的逻辑代码
           */
           
           /// 基本设置
           self.completion()
        })
        }
        
        
        /// 初始化设置
    private func completion(){
        /// 移除自身通知
        NotificationCenter.default.removeObserver(self)
        
        //注册进入后台通知
        NotificationCenter.default.addObserver(self, selector: #selector(self.EnterBackgroundNotification), name:NSNotification.Name.UIApplicationDidEnterBackground, object: nil)
        
        //注册进入前台通知
        NotificationCenter.default.addObserver(self, selector: #selector(self.EnterForegroundNotification), name: NSNotification.Name.UIApplicationWillEnterForeground, object: nil)
        
        self.onMessage(MsgAction: "message") { (dataString) in
            
            let data = dataString.data(using: String.Encoding.utf8)
            let js = JSON(data:data!)
            
            /// 自定义的Model
            let msgModel = MsgBaseModel.init()
            
            /// 收到数据
            if js["msgId"].stringValue != ""{
                
                /*
                解析数据
                数据解析用的SwiftyJSON
                本地数据存储用的FMDB
                 */
                
                //声音提醒
                self.applicationState(pushData: ((js["sendName"].stringValue) == "" ? (js["pushData"].stringValue) : (js["sendName"].stringValue) + ": " + (js["pushData"].stringValue)))
                
                if(self.checkedMessage(MsgId: msgModel.msgId)){
                    print("\(msgModel.msgId) 消息数据库中已存在")
                    return
                }
                
                let ins = self.insertMsg(model: msgModel)
                
                if(ins == false){
                    print("\(msgModel.msgId) 消息插入失败")
                    return
                }
                
                /// 通知 更新UI界面
                self.SessionListNotice?.newMessage(MsgModel: msgModel)
                self.SessionNotice?.newMessage(MsgModel: msgModel)
                
                /// 未读消息条数
                let arr = self.getSessionListData()
                var number = 0
                for model in arr{
                    number = number + model.notReadNumber
                    let na = NSNotification.Name(rawValue:"news")
                    NotificationCenter.default.post(name: na, object:(number))
                }
            }
        }
        
    }
        
        /// 监听消息
    private func onMessage(MsgAction:String,cb:@escaping (_ data:String)->Void){
        
        self.socket?.on(MsgAction, callback: { (data, ack) in
            /// 在TCP/IP协议中，如果接收方成功的接收到数据，那么会回复一个ACK数据
            ack.with(UUID().uuidString)
            if data[0] is String{
                
                cb(data[0] as? String ?? "")
                
            }else {
                //此处后期可能语音图片等格式文件需要做相应判断
                cb(bs_String.objectToJson(object: (data[0] as? Dictionary) ?? [:]) ?? "")
            }
        })
    }
    
    @objc func EnterForegroundNotification(){
        print("进入前台")
        
        if(self.backTime < self.backTimeOut){
            self.app.endBackgroundTask(self.taskID)
        }
        
        self.backTime = 0
        self.backTimer?.invalidate()
    }
    
    /// 执行进入后台任务
    @objc func EnterBackgroundNotification(){
        print("进入后台")
        
        //创建定时任务
        self.backTimer = Timer.scheduledTimer(timeInterval: 1,target:self,selector:#selector(self.backstageTiming),userInfo:nil,repeats:true)
        /// 应用在被用户切换到后台的时候就会被系统暂停掉， 这个方法可以延迟系统暂停你的应用，并申请额外的时间来完成未完成的任务
        self.taskID = self.app.beginBackgroundTask(expirationHandler: {
            /// 后台任务到期执行，好像是10分钟  我暂时没设置这么久
        })
    }
    
    /// 后台任务计时
    @objc func backstageTiming(){
        
        self.backTime += 1
        if(self.backTime == self.backTimeOut){
            self.backTimer?.invalidate
            /// 结束掉保活
            self.app.endBackgroundTask(self.taskID)
        }
    }
    
    /// 定时任务心跳包
    @objc private func onlineTimerRun(){
        
        self.socket?.emitWithAck("time", "").timingOut(after: 3, callback: { (data) in
            
            if data[0] is Dictionary {
                /// 逻辑代码
            }
        })
    }
 }

以上代码差不多是连接，设置（心跳包，通知等），收到消息（处理数据，更新UI）等一系列方法的去逻辑版。本地数据存储也就是检查下有没有重复，不存在就插入。

这样就基本实现了用户登录，可以接收到产品想要的自定义消息，比如系统消息，账单消息等等。但是如果需要单聊，你就需要根据用户id取聊天数据，然后条数多了肯定需要分页。所以要多一个 getUserMsgListPageData(userid:String, page:Int) 这种方法。
然后消息主页，子界面的那些处理就需要自己去慢慢拓展了。

有关即时通讯理解，本人也是初探，有问题欢迎讨论。互勉！

转载请注明出处，谢谢。