节操代码修养妹子和其他(Go语言版)

//代码来自 https://github.com/xgdapg/xconn/blob/master/xconn.go，已验证
//Conn 对应一个tcp连接
type Conn struct {
        //原生TCP连接
    conn         net.Conn
        //发送数据的channel（类似队列）
    send         chan *MsgData
        //消息处理方法
    msgHandler   MsgHandler
        //tcp缓冲区
    recvBuffer   []byte
        //消息一次封装，后续会进行二次封装
    msgPacker    MsgPacker
        //健康检查周期
    pingInterval uint
        //健康检查方法
    pingHandler  PingHandler
        //停止健康检查channel
    pingStop     chan bool
        //自定义关闭连接时所调用方法
    closeHandler CloseHandler
}
//MsgHandler 为自定义处理消息
type MsgHandler interface {
    HandleMsg(*MsgData)
}
//MsgPacker 为自定义拆包解包方式，为了性能可以直接将此段与recvLoop端合并
type MsgPacker interface {
    PackMsg(*MsgData) []byte
    UnpackMsg([]byte) *MsgData
}
//MsgData 自定义消息类型
type MsgData struct {
    Data []byte
    Ext  interface{}
}
//PingHandler 为自定义心跳命令
type PingHandler interface {
    HandlePing()
}
//CloseHandler 为连接断开时的自定义操作
type CloseHandler interface {
    HandleClose()
}
//NewConn 为处理新连接方式：启动两个协程，一个只负责读，一个只负责写，
//也可以认为开启了三个协程，第三个协程负责进行定时ping操作
func NewConn(conn net.Conn) *Conn {
    c := &Conn{
        conn:         conn,
        send:         make(chan *MsgData, 64),
        msgHandler:   nil,
        recvBuffer:   []byte{},
        msgPacker:    nil,
        pingInterval: 0,
        pingHandler:  nil,
        pingStop:     nil,
        closeHandler: nil,
    }
    go c.recvLoop()
    go c.sendLoop()
    return c
}
//recoverPanic 程序panic情况下的处理方法（例如向已经关闭的tcp连接写数据会造成panic）
func recoverPanic() {
    if err := recover(); err != nil {
        //fmt.Println(err)
    }
}
//SetMsgHandler 为 Getter Setter 方法
func (this *Conn) SetMsgHandler(hdlr MsgHandler) {
    this.msgHandler = hdlr
}
//SetMsgPacker 为 Getter Setter 方法
func (this *Conn) SetMsgPacker(packer MsgPacker) {
    this.msgPacker = packer
}
//SetPing 为 Getter Setter 方法
func (this *Conn) SetPing(sec uint, hdlr PingHandler) {
    this.pingInterval = sec
    this.pingHandler = hdlr
    if this.pingStop == nil {
        this.pingStop = make(chan bool)
    }
    if sec > 0 {
        go this.pingLoop()
    }
}
//SetCloseHandler 为 Getter Setter 方法
func (this *Conn) SetCloseHandler(hdlr CloseHandler) {
    this.closeHandler = hdlr
}
//pingLoop 为定时健康检查操作
func (this *Conn) pingLoop() {
    defer recoverPanic()
    for {
        select {
        case <-this.pingStop:
            return
        case <-time.After(time.Duration(this.pingInterval) * time.Second):
            this.Ping()
        }
    }
}
//RawConn 返回原始的tcp连接
func (this *Conn) RawConn() net.Conn {
    return this.conn
}
//recvLoop 用于处理接收到的tcp包，并进行拆包等操作，然后调用recvMsg方法进行处理
func (this *Conn) recvLoop() {
    defer recoverPanic()
    defer this.Close()
    buffer := make([]byte, 2048)
        //一次封包协议：四个字节（int32）表示包长度，根据包长度截取消息长度作为包。
    for {
        bytesRead, err := this.conn.Read(buffer)
        if err != nil {
            return
        }
        this.recvBuffer = append(this.recvBuffer, buffer[0:bytesRead]...)
        for len(this.recvBuffer) > 4 {
            length := binary.BigEndian.Uint32(this.recvBuffer[0:4])
            readToPtr := length + 4
            if uint32(len(this.recvBuffer)) < readToPtr {
                break
            }
            if length == 0 {
                if this.pingHandler != nil {
                    this.pingHandler.HandlePing()
                }
            } else {
                buf := this.recvBuffer[4:readToPtr]
                go this.recvMsg(buf)
            }
            this.recvBuffer = this.recvBuffer[readToPtr:]
        }
    }
}
//recvMsg 为代理，实际执行的是后台的HandleMsg方法。
func (this *Conn) recvMsg(data []byte) {
    defer recoverPanic()
    msg := &MsgData{
        Data: data,
        Ext:  nil,
    }
        //调用UnackMsg对信息进行二次解包
    if this.msgPacker != nil {
        msg = this.msgPacker.UnpackMsg(data)
    }
    if this.msgHandler != nil {
        this.msgHandler.HandleMsg(msg)
    }
}
//sendLoop 用于发送数据包
func (this *Conn) sendLoop() {
    defer recoverPanic()
    for {
        msg, ok := <-this.send
        if !ok {
            break
        }
        go this.sendMsg(msg)
    }
}
//sendMsg 用于发送数据包，实际先调用PackMsg进行信息持久化，然后二次封包，转换为本框架能接受的形式
func (this *Conn) sendMsg(msg *MsgData) {
    defer recoverPanic()
    sendBytes := make([]byte, 4)
    if msg != nil {
        data := msg.Data
        if this.msgPacker != nil {
            data = this.msgPacker.PackMsg(msg)
        }
        length := len(data)
        binary.BigEndian.PutUint32(sendBytes, uint32(length))
        sendBytes = append(sendBytes, data...)
    }
    this.conn.Write(sendBytes)
}
//Close 关闭连接
func (this *Conn) Close() {
    defer recoverPanic()
    this.conn.Close()
    close(this.send)
    if this.pingStop != nil {
        close(this.pingStop)
    }
    if this.closeHandler != nil {
        this.closeHandler.HandleClose()
    }
}
//SendMsg 用于发送数据
func (this *Conn) SendMsg(msg *MsgData) {
    this.send <- msg
}
//SendData 用于发送数据
func (this *Conn) SendData(data []byte) {
    this.SendMsg(&MsgData{Data: data, Ext: nil})
}
//Ping 用于健康监测
func (this *Conn) Ping() {
    go this.sendMsg(nil)
}

//代码来自《Go语言游戏项目应用情况汇报》
func (db *Database) Transaction(work func()) {
    db.lock.Lock()
    defer db.lock.UnLock()
    //事务控制
    defer func() {
        if err := recover; err == nil {
            db.commit(info)
        } else {
            db.rollback()
            //选择性抛出panic
            panic(TransError{err})
        }
    }()
    //执行传入的函数
    work()
}

var err error
write := func(buf []byte) {
    if err != nil {
        return
    }
    _, err = w.Write(buf)
}
write(p0[a:b])
write(p1[c:d])
write(p2[e:f])
// and so on
if err != nil {
    return err
}

func (ew *errWriter) write(buf []byte) {
    if ew.err != nil {
        return
    }
    _, ew.err = ew.w.Write(buf)
}