swift——从零开始自己的IM
2018-01-26 本文已影响71人
Bart_Simpson
之前旧项目里的聊天是集成的融云，然而有诸多不方便，很多需求都无法正常且简单的实现。而且使用聊天的人也并不多。。。
所以决定公司自己开发即时通讯（反正用户少=。=）
开发中是基于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<String,Any>) ?? [:]) ?? "")
}
})
}
@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<String,AnyObject> {
/// 逻辑代码
}
})
}
}
以上代码差不多是连接，设置（心跳包，通知等），收到消息（处理数据，更新UI）等一系列方法的去逻辑版。本地数据存储也就是检查下有没有重复，不存在就插入。
这样就基本实现了用户登录，可以接收到产品想要的自定义消息，比如系统消息，账单消息等等。但是如果需要单聊，你就需要根据用户id取聊天数据，然后条数多了肯定需要分页。所以要多一个 getUserMsgListPageData(userid:String, page:Int) 这种方法。
然后消息主页，子界面的那些处理就需要自己去慢慢拓展了。
有关即时通讯理解，本人也是初探，有问题欢迎讨论。互勉！
