53. Socket服务三次握手的示例

什么是三次握手呢?这是服务器和客户端之间沟通的过程。
首先,客户端对服务器发送了条信息。
然后,服务端对客户端说,我收到了。
最后,客户端对服务端说,好的,我知道你收到了。
怎么样?这就是三次握手。哈哈!

服务端,我们先声明本地要监听的地址和端口。

netListent, err := net.Listen("tcp", "localhost:7373")

有开始就有结束,当主函数运行结束时,需要释放资源。

defer netListent.Close()

在服务端做标记,表明现在开始等待客户端访问了。

Log(time.Now().Format("2006-01-02 15:04:05.0000000"),"Waiting for client ...")

监听是一个不停息的循环运行。所以使用 for{}

    for{
        conn, err := netListent.Accept()
        if err != nil{
            continue
        }
        //标记发生了一次连接
        Log(conn.RemoteAddr().String(), "tcp connect success")
        go handleConnection(conn)
    }

每个从客户端请求的连接,都会在服务端生成一个 goroutine 协程去处理。
go handleConnection(conn)
服务端处理的过程是这样的。建立一个缓存,接收客户端信息。如果接收正确,就反馈给客户端说“我收到了”。如果客户端没有反应,就证明客户端“没有收到回执”。如果客户端有反馈,就说明“客户端收到回执”。
最后,沟通结束,释放连接资源。defer conn.Close()

//客户端连接处理
func handleConnection(conn net.Conn) {
    buffer := make([]byte, 1024)
    for{
        //接收客户端信息
        msg, err := conn.Read(buffer)
        if err != nil{
            //接收错误,日志打印
            Log(conn.RemoteAddr().String(), "connection error: ", err)
            return
        }
        //接收正确,日志打印
        Log(conn.RemoteAddr().String(), "receive data: ", string(buffer[:msg]))
        //反馈给客户端
        bufferReturn := "我收到了"
        msgR, err2 := conn.Write([]byte(bufferReturn))
        //确认客户端未收到回执
        if err2 != nil{
            Log(conn.RemoteAddr().String(), "没有收到回执")
            return
        }
        //确认客户端收到回执
        msg, err = conn.Read(buffer)
        Log(conn.RemoteAddr().String(), "客户端收到回执", string(buffer[:msg]), "客户收到了", msgR, ";实际发送了", len(bufferReturn))
    }
    defer conn.Close()
}

其中应用到的日志记录和错误处理。

//日志记录
func Log(i ...interface{}) {
    fmt.Println(i...)
    return
}
//错误处理
func CheckErr(err error) {
    if err != nil{
        fmt.Fprintln(os.Stderr, err)
        os.Exit(1)
    }
}

完整的服务端代码示例

/**
* MySocketProtocolServer
* @Author:  Jian Junbo
* @Email:   junbojian@qq.com
* @Create:  2017/9/15 9:07
* Copyright (c) 2017 Jian Junbo All rights reserved.
*
* Description:  服务端 Socket 信息接收
*/
package main

import (
    "fmt"
    "net"
    "os"
    "time"
)

func main() {
    //监听服务声明实例化
    netListent, err := net.Listen("tcp", "localhost:7373")
    CheckErr(err)
    defer netListent.Close()
    //标记开始服务
    Log(time.Now().Format("2006-01-02 15:04:05.0000000"),"Waiting for client ...")
    //服务监听
    for{
        conn, err := netListent.Accept()
        if err != nil{
            continue
        }
        //标记发生了一次连接
        Log(conn.RemoteAddr().String(), "tcp connect success")
        go handleConnection(conn)
    }
}
//客户端连接处理
func handleConnection(conn net.Conn) {
    buffer := make([]byte, 1024)
    for{
        //接收客户端信息
        msg, err := conn.Read(buffer)
        if err != nil{
            //接收错误,日志打印
            Log(conn.RemoteAddr().String(), "connection error: ", err)
            return
        }
        //接收正确,日志打印
        Log(conn.RemoteAddr().String(), "receive data: ", string(buffer[:msg]))
        //反馈给客户端
        bufferReturn := "我收到了"
        msgR, err2 := conn.Write([]byte(bufferReturn))
        //确认客户端未收到回执
        if err2 != nil{
            Log(conn.RemoteAddr().String(), "没有收到回执")
            return
        }
        //确认客户端收到回执
        msg, err = conn.Read(buffer)
        Log(conn.RemoteAddr().String(), "客户端收到回执", string(buffer[:msg]), "客户收到了", msgR, ";实际发送了", len(bufferReturn))
    }
    defer conn.Close()
}
//日志记录
func Log(i ...interface{}) {
    fmt.Println(i...)
    return
}
//错误处理
func CheckErr(err error) {
    if err != nil{
        fmt.Fprintln(os.Stderr, err)
        os.Exit(1)
    }
}

客户端的任务是发送信息给服务端,等待服务端的反馈。收到服务端的反馈后,再通知服务端说“ok”,表示已经知道服务端收到信息了。
先要确定获取服务端的地址和通讯端口。

server := "localhost:7373"
tcpAddr, err := net.ResolveTCPAddr("tcp4", server)

建立服务器连接

conn, err := net.DialTCP("tcp", nil, tcpAddr)

连接成功后,日志打印表示一下

Log("connection success")

发送信息到服务器,是一个函数 sender(conn) 最终完成的。

func sender(conn *net.TCPConn) {
    myage := int(time.Now().Year())
    myage -= 1973
    words := "{\"ID\":\"i\",\"Name\":\"Joel\",\"Age\":\""+strconv.Itoa(myage)+"\",\"Programming Language\":\"go\"}"
    //msgBack, err := conn.Write(protocol.Enpack([]byte(words)))
    msgBack, err := conn.Write([]byte(words))
    if err != nil{
        Log(conn.RemoteAddr().String(), "Fatal error: ", err)
        os.Exit(1)
    }
    buffer := make([]byte, 1024)
    msg, err := conn.Read(buffer)
    Log(conn.RemoteAddr().String(), "服务器反馈: ", string(buffer[:msg]), msgBack,";实际发送了", len(words))
    conn.Write([]byte("ok"))
}

这个函数中,我们准备好了要发送的信息 words。
写入信息到连接中

msgBack, err := conn.Write([]byte(words))

接收服务器对信息的反馈

msg, err := conn.Read(buffer)

在告诉服务器,它的反馈收到了。

conn.Write([]byte("ok"))

完整的客户端代码示例

/**
* MySocketProtocolClient
* @Author:  Jian Junbo
* @Email:   junbojian@qq.com
* @Create:  2017/9/15 10:44
* Copyright (c) 2017 Jian Junbo All rights reserved.
*
* Description:  客户端 Socket
*/
package main

import (
    "net"
    "fmt"
    "os"
    "time"
)

func main() {
    //获取服务器地址和端口
    server := "localhost:7373"
    tcpAddr, err := net.ResolveTCPAddr("tcp4", server)
    if err != nil{
        Log(os.Stderr, "Fatal error: ", err)
        os.Exit(1)
    }
    //建立服务器连接
    conn, err := net.DialTCP("tcp", nil, tcpAddr)
    if err != nil{
        Log(conn.RemoteAddr().String(), os.Stderr, "Fatal error: ", err)
        os.Exit(1)
    }
    Log("connection success")
    sender(conn)
    fmt.Println("send over")
}

func sender(conn *net.TCPConn) {
    myage := int(time.Now().Year())
    myage -= 1973
    words := "{\"ID\":\"i\",\"Name\":\"Joel\",\"Age\":\""+strconv.Itoa(myage)+"\",\"Programming Language\":\"go\"}"
    msgBack, err := conn.Write([]byte(words))
    if err != nil{
        Log(conn.RemoteAddr().String(), "Fatal error: ", err)
        os.Exit(1)
    }
    buffer := make([]byte, 1024)
    msg, err := conn.Read(buffer)
    Log(conn.RemoteAddr().String(), "服务器反馈: ", string(buffer[:msg]), msgBack,";实际发送了", len(words))
    conn.Write([]byte("ok"))
}
//日志打印
func Log(v ...interface{}) {
    fmt.Println(v...)
}

服务端运行结果

127.0.0.1:50756 tcp connect success
127.0.0.1:50756 receive data:  {"ID":"i","Name":"Joel","Age":"44","Programming Language":"go"}
127.0.0.1:50756 客户端收到回执 ok 客户收到了 12 ;实际发送了 12
127.0.0.1:50756 connection error:  read tcp 127.0.0.1:7373->127.0.0.1:50756: wsarecv: An existing connection was forcibly closed by the remote host.

客户端运行结果

connection success
127.0.0.1:7373 服务器反馈:  我收到了 63 ;实际发送了 63
send over

现在通过网络传递的信息,没有特别的加工,没有自定义的通讯协议。如果要增加这个,建立自己的通讯协议后(封装 Enpack、解析 Depack),在客户端进行封装,在服务端进行解析。
在当前的代码中做修改。
服务端

Log(conn.RemoteAddr().String(), "receive data: ", string(buffer[:msg]))

改成 ↓

tmpBuffer := make([]byte, 1024)
tmpBuffer = protocol.Depack(append(tmpBuffer,buffer[:msg]...))
Log(conn.RemoteAddr().String(), "receive data: ", string(tmpBuffer))

客户端

msgBack, err := conn.Write([]byte(words))
...
Log(conn.RemoteAddr().String(), "服务器反馈: ", string(buffer[:msg]), msgBack,";实际发送了", len(words))

改成 ↓

msgBack, err := conn.Write(protocol.Enpack([]byte(words)))
...
Log(conn.RemoteAddr().String(), "服务器反馈: ", string(buffer[:msg]), msgBack,";实际发送了", len(protocol.Enpack([]byte(words))))

由于添加了协议后,发送内容增加了头部的一些信息,所以实际发送信息的长度也变化了。
协议protocol的代码示例

/**
* protocol
* @Author:  Jian Junbo
* @Email:   junbojian@qq.com
* @Create:  2017/9/14 11:49
* Copyright (c) 2017 Jian Junbo All rights reserved.
*
* Description:  通讯协议处理
*/
package protocol

import (
    "bytes"
    "encoding/binary"
)

const (
    ConstHeader = "Headers"
    ConstHeaderLength = 7
    ConstMLength = 4
)

//封包
func Enpack(message []byte) []byte {
    return append(append([]byte(ConstHeader), IntToBytes(len(message))...), message...)
}

//解包
func Depack(buffer []byte) []byte {
    length := len(buffer)

    var i int
    data := make([]byte, 32)
    for i = 0; i < length; i++ {

        if length < i + ConstHeaderLength + ConstMLength{
            break
        }
        if string(buffer[i:i+ConstHeaderLength]) == ConstHeader {
            messageLength := ByteToInt(buffer[i+ConstHeaderLength : i+ConstHeaderLength+ConstMLength])
            if length < i+ConstHeaderLength+ConstMLength+messageLength {
                break
            }
            data = buffer[i+ConstHeaderLength+ConstMLength : i+ConstHeaderLength+ConstMLength+messageLength]
        }
    }

    if i == length {
        return make([]byte, 0)
    }

    return data
}

//字节转换成整形
func ByteToInt(n []byte) int {
    bytesbuffer := bytes.NewBuffer(n)
    var x int32
    binary.Read(bytesbuffer, binary.BigEndian, &x)

    return int(x)
}

//整数转换成字节
func IntToBytes(n int) []byte {
    x := int32(n)
    bytesBuffer := bytes.NewBuffer([]byte{})
    binary.Write(bytesBuffer, binary.BigEndian, x)
    return bytesBuffer.Bytes()
}

增加了解析功能的服务端代码示例

/**
* MySocketProtocolServer
* @Author:  Jian Junbo
* @Email:   junbojian@qq.com
* @Create:  2017/9/15 9:07
* Copyright (c) 2017 Jian Junbo All rights reserved.
*
* Description:  服务端 Socket 信息接收
*/
package main

import (
    "fmt"
    "net"
    "os"
    "time"
    "protocol"
)

func main() {
    //监听服务声明实例化
    netListent, err := net.Listen("tcp", "localhost:7373")
    CheckErr(err)
    defer netListent.Close()
    //标记开始服务
    Log(time.Now().Format("2006-01-02 15:04:05.0000000"),"Waiting for client ...")
    //服务监听
    for{
        conn, err := netListent.Accept()
        if err != nil{
            continue
        }
        //标记发生了一次连接
        Log(conn.RemoteAddr().String(), "tcp connect success")
        go handleConnection(conn)
    }
}
//客户端连接处理
func handleConnection(conn net.Conn) {
    buffer := make([]byte, 2048)
    for{
        //接收客户端信息
        msg, err := conn.Read(buffer)
        if err != nil{
            //接收错误,日志打印
            Log(conn.RemoteAddr().String(), "connection error: ", err)
            return
        }
        //接收正确,日志打印
        tmpBuffer := make([]byte, 1024)
        tmpBuffer = protocol.Depack(append(tmpBuffer,buffer[:msg]...))
        Log(conn.RemoteAddr().String(), "receive data: ", string(tmpBuffer))
        //反馈给客户端
        bufferReturn := "我收到了"
        msgR, err2 := conn.Write([]byte(bufferReturn))
        //确认客户端未收到回执
        if err2 != nil{
            Log(conn.RemoteAddr().String(), "没有收到回执")
            return
        }
        //确认客户端收到回执
        msg, err = conn.Read(buffer)
        Log(conn.RemoteAddr().String(), "客户端收到回执", string(buffer[:msg]), "客户收到了", msgR, ";实际发送了", len(bufferReturn))
    }
    defer conn.Close()
}
//日志记录
func Log(i ...interface{}) {
    fmt.Println(i...)
    return
}
//错误处理
func CheckErr(err error) {
    if err != nil{
        fmt.Fprintln(os.Stderr, err)
        os.Exit(1)
    }
}

增加了封装的客户端代码示例

/**
* MySocketProtocolClient
* @Author:  Jian Junbo
* @Email:   junbojian@qq.com
* @Create:  2017/9/15 10:44
* Copyright (c) 2017 Jian Junbo All rights reserved.
*
* Description:  客户端 Socket
*/
package main

import (
    "net"
    "fmt"
    "os"
    "time"
    "strconv"
    "protocol"
)

func main() {
    //获取服务器地址和端口
    server := "localhost:7373"
    tcpAddr, err := net.ResolveTCPAddr("tcp4", server)
    if err != nil{
        Log(os.Stderr, "Fatal error: ", err)
        os.Exit(1)
    }
    //建立服务器连接
    conn, err := net.DialTCP("tcp", nil, tcpAddr)
    if err != nil{
        Log(conn.RemoteAddr().String(), os.Stderr, "Fatal error: ", err)
        os.Exit(1)
    }
    Log("connection success")
    sender(conn)
    fmt.Println("send over")
}

func sender(conn *net.TCPConn) {
    myage := int(time.Now().Year())
    myage -= 1973
    words := "{\"ID\":\"i\",\"Name\":\"Joel\",\"Age\":\""+strconv.Itoa(myage)+"\",\"Programming Language\":\"go\"}"
    msgBack, err := conn.Write(protocol.Enpack([]byte(words)))
    //msgBack, err := conn.Write([]byte(words))
    if err != nil{
        Log(conn.RemoteAddr().String(), "Fatal error: ", err)
        os.Exit(1)
    }
    buffer := make([]byte, 1024)
    msg, err := conn.Read(buffer)
    Log(conn.RemoteAddr().String(), "服务器反馈: ", string(buffer[:msg]), msgBack,";实际发送了", len(protocol.Enpack([]byte(words))))
    conn.Write([]byte("ok"))
}
//日志打印
func Log(v ...interface{}) {
    fmt.Println(v...)
}

运行效果上只是客户端这里显示增加了几个字节

connection success
127.0.0.1:7373 服务器反馈:  我收到了 74 ;实际发送了 74
send over
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,590评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 86,808评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,151评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,779评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,773评论 5 367
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,656评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,022评论 3 398
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,678评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 41,038评论 1 299
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,659评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,756评论 1 330
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,411评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,005评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,973评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,203评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,053评论 2 350
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,495评论 2 343

推荐阅读更多精彩内容

  • 说明 本文 翻译自 realpython 网站上的文章教程 Socket Programming in Pytho...
    keelii阅读 2,100评论 0 16
  • 网络编程 一.楔子 你现在已经学会了写python代码,假如你写了两个python文件a.py和b.py,分别去运...
    go以恒阅读 1,988评论 0 6
  • 点击查看原文 Web SDK 开发手册 SDK 概述 网易云信 SDK 为 Web 应用提供一个完善的 IM 系统...
    layjoy阅读 13,659评论 0 15
  • 实时消息协议---流的分块 版权声明: 版权(c)2009 Adobe系统有限公司。全权所有。 摘要: 本备忘录描...
    一个人zy阅读 1,885评论 0 9
  • 今天在浏览简友网页的时候,发现自己一个问题,那就是不管出于什么原因,总会错过高分简友,自己会不自觉地做到:不关注动...
    梅花开在春天阅读 670评论 6 15