打造一款属于自己的go语言socket框架4之session管理

前言:

上篇文章说道我们怎么通过不同的数据去调用框架中不同的函数,在socket编程中,服务器做的不仅仅是接受信息,还需要推送消息,最典型的场景就是聊天室了,每当一个客户端发送一个一条消息,服务端就需要把这条消息,循环发送给其它所有客户端。已经把所有代码整合了,希望给个星星支持一下 microSocket

实现思路:

我们知道在go语言socket编程时,服务端向客户端发送信息只需要 调用 conn.write()即可,一个连接就对应一个conn,每次服务端从一个客户端消息的时候,想要推送给别的客户端信息就需要别的客户端的coon,所以我们的解决思路就是

  1. 每次客户端和服务端握手成功后,就把当前的conn,保存到全局 协程安全的map里
  2. 当客户端与服务端断开连接的时候就把这个coon从map里面删除

每次服务器要广播全体的时候,就把当前的 map遍历一遍,一次推送就好了

代码实现:

每次连接成功的时候就把coon 封装成一个session

for {
        //不断地结束新的客户端连接
        conn, err := tcpListen.Accept()
        if err != nil {
            log.Println(err)
            continue
        }
        //把conn封装成一个session
        sess := NewSession(fd, conn)
        //把session 保存到sessionMaster
        this.SessionMaster.SetSession(fd, sess)
        fd++
        go this.connHandle(conn, sess)
    }

session 和sessionMaster的实现

//-------------------------------------------一个session代表一个连接------------------------------------------
type session struct {
    id    uint32
    con   net.Conn
    times int64
    lock  sync.Mutex
}

func NewSession(id uint32, con net.Conn) *session {
    return &session{
        id:    id,
        con:   con,
        times: time.Now().Unix(),
    }
}
//向conn写数据
func (this *session) write(msg string) error {
    this.lock.Lock()
    defer this.lock.Unlock()
    _ ,errs := this.con.Write([]byte(msg))
    return errs
}
//关闭conn
func (this *session)close(){
    this.con.Close()
}

//.......................................SESSION管理类.......................................

type SessionM struct {
    sessions map[uint32]*session
    num      uint32
    lock     sync.RWMutex
}

func NewSessonM() *SessionM {
    return &SessionM{
        sessions: make(map[uint32]*session),
        num:      0,
    }
}

func (this *SessionM) GetSessionById(id uint32) *session {
    if v, exit := this.sessions[id]; exit {
        return v
    }
    return nil
}

func (this *SessionM) SetSession(id uint32, sess *session) {
    this.lock.Lock()
    defer this.lock.Unlock()
    this.sessions[id] = sess
}

//关闭连接并删除
func (this *SessionM) DelSessionById(id uint32) {
    this.lock.Lock()
    defer this.lock.Unlock()
    if v,exit := this.sessions[id];exit{
        v.con.Close()
    }
    delete(this.sessions, id)
}

func (this *SessionM) WriteByid(id uint32, msg string) bool {
    if v, exit := this.sessions[id]; exit {
        if err := v.write(msg); err != nil {
            this.DelSessionById(id)
            return false
        } else {
            return true
        }
    }
    return false
}
代码解释:
  1. 首先我们要封装一个session类,一个session就对应一个conn,大家可能发现到了,为什么conn.write
    我们要加锁呢。原因也很简单,就是为了保证,同一时间,最多只会有一个协程向一个coon写数据,如果有多个协程同时写的话,就会发生数据混乱,所以这里的锁是很有必要的!
  2. 接下来就是sessionMaster类,服务端就是通过它来对所有客户端的连接进行操作,它的核心,其实就是一个sessions 的map,该map在删除和插入新数据时都有加锁处理,确保协程安全。当服务端想要像客户端推送消息的时候直接调用write方法就好了如下:
ser.SessionMaster.WriteByid(uint32(fd), "Hello")

总结:

  1. 通过全局sessionMaster对所有客户端的连接进行统一管理,方便推送客户端数据
  2. 每次对客户端推送数据时必须要加锁,不然数据可能会错乱

至此我们的socket框架分享,已经差不多讲完了,有什么不懂得,或者意见欢迎留言评论。

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 第一部分 Python基础篇(80题) 1、为什么学习Python? Python相对于其他编程语言有很多优点: ...
    清清子衿木子水心阅读 1,740评论 0 1
  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 134,982评论 19 139
  • 养只猫,再养只狗 一只叫Siri 一只叫Google Hey Siri! Ok Google!
    魏鞍阳阅读 204评论 1 3
  • 接着讲参加萌姐线上复盘课的感悟: 在萌姐的年度复盘课上我对可控时间又有了新一步的理解,以前总是觉得晚上跟着...
    lovent阅读 114评论 0 0
  • 参加过很多群体性活动,比如:21天不抱怨活、30天腹肌训练、天天平板撑晒朋友圈、365天改变自己训练营、每周读一本...
    己未阅读 576评论 4 6