go Socket 编程

go Socket 编程

Socket抽象层

我们知道两个进程如果需要进行通讯最基本的一个前提是能够唯一标示一个进程，在本地进程通讯中我们可以使用PID来唯一标示一个进程，但PID只在本地唯一，网络中的两个进程PID冲突几率很大，这时候我们需要另辟它径了，我们知道IP层的IP地址可以唯一标示主机，而TCP层协议和端口号可以唯一标示主机的一个进程，这样我们可以利用IP地址＋协议＋端口号唯一标示网络中的一个进程。

能够唯一标示网络中的进程后，它们就可以利用Socket进行通信了，什么是Socket呢？我们经常把Socket翻译为套接字，Socket是在应用层和传输层之间的一个抽象层，它把TCP/IP层复杂的操作抽象为几个简单的接口供应用层调用已实现进程在网络中通信。

Socket起源于UNIX，在Unix一切皆文件哲学的思想下，Socket是一种"打开—读/写—关闭"模式的实现，服务器和客户端各自维护一个"文件"，在建立连接打开后，可以向自己文件写入内容供对方读取或者读取对方内容，通讯结束时关闭文件。

TCP编程

TCP协议

TCP/IP(Transmission Control Protocol/Internet Protocol) 即传输控制协议/网间协议，是一种面向连接（连接导向）的、可靠的、基于字节流的传输层（Transport layer）通信协议，因为是面向连接的协议，数据像水流一样传输，会存在黏包问题。

TCP服务端

一个TCP服务端可以同时连接很多个客户端。因为Go语言中创建多个goroutine实现并发非常方便和高效，所以我们可以每建立一次链接就创建一个goroutine去处理。

TCP服务端程序的处理流程：

1.监听端口
2.接收客户端请求建立链接
3.创建goroutine处理链接。

我们使用Go语言的net包实现的TCP服务端代码如下：

package main

import (
	"bytes"
	"fmt"
	"net"
	"time"
)

func main() {
	listen, err := net.Listen("tcp", "127.0.0.1:8080")
	if err != nil {
		fmt.Println("监听出错：", err)
		return
	}
	for {
		conn, err := listen.Accept()
		if err != nil {
			fmt.Println("接收出错", err)
			continue
		}

		fmt.Println("客户端连接了：", conn.RemoteAddr())
		fmt.Println("服务端为：", conn.LocalAddr())
		go handleConn(conn)

	}
}

func handleConn(conn net.Conn) {
	defer conn.Close()
	for {
		var buf [128]byte
		n, err := conn.Read(buf[:])
		if err != nil {
			fmt.Println("读取出错")
			break
		}
		fmt.Println(string(buf[:n]))
		time.Sleep(1 * time.Second)
		conn.Write(bytes.ToUpper(buf[:]))
	}
}

将上面的代码保存之后编译成server或server.exe可执行文件。

TCP客户端

一个TCP客户端进行TCP通信的流程如下：

1.建立与服务端的链接
2.进行数据收发
3.关闭链接

使用Go语言的net包实现的TCP客户端代码如下：

package main

import (
	"fmt"
	"net"
)

func main() {
	conn,err:=net.Dial("tcp","127.0.0.01:8080")
	if err != nil {
		fmt.Println("拨号错误:",err)
		return
	}
	defer conn.Close()
	for  {
		conn.Write([]byte("lqz"))
		var buf [128]byte
		i,err:=conn.Read(buf[:])
		if err != nil {
			fmt.Println("接收数据错误:",err)
		}
		fmt.Println(string(buf[:i]))
	}

}

UDP编程

UDP协议

UDP协议（User Datagram Protocol）中文名称是用户数据报协议，是OSI（Open System Interconnection，开放式系统互联）参考模型中一种无连接的传输层协议，不需要建立连接就能直接进行数据发送和接收，属于不可靠的、没有时序的通信，但是UDP协议的实时性比较好，通常用于视频直播相关领域。

UDP服务端

使用Go语言的net包实现的UDP服务端代码如下：

package main

import (
	"bytes"
	"fmt"
	"net"
	"time"
)

func main() {
	listen, err := net.ListenUDP("udp", &net.UDPAddr{
		IP:   net.IPv4(0, 0, 0, 0),
		Port: 8888,
	})
	if err != nil {
		fmt.Println("监听失败:", err)
		return
	}
	defer listen.Close()
	for {
		var data [1024]byte
		n, addr, err := listen.ReadFromUDP(data[:]) // 接收数据
		if err != nil {
			fmt.Println("读数据出错:", err)
			continue
		}
		fmt.Printf("收到的数据是:%v\n 客户端地址是:%v\n 读到的字节数:%v\n", string(data[:n]), addr, n)
		_, err = listen.WriteToUDP(data[:n], addr) // 发送数据
		if err != nil {
			fmt.Println("写数据出错:", err)
			continue
		}
	}
}

func handleConn(conn net.Conn) {
	defer conn.Close()
	for {
		var buf [128]byte
		n, err := conn.Read(buf[:])
		if err != nil {
			fmt.Println("读取出错")
			break
		}
		fmt.Println(string(buf[:n]))
		time.Sleep(1 * time.Second)
		conn.Write(bytes.ToUpper(buf[:]))
	}
}

UDP客户端

使用Go语言的net包实现的UDP客户端代码如下：

package main

import (
	"fmt"
	"net"
)

func main() {
	socket, err := net.DialUDP("udp", nil, &net.UDPAddr{
		IP:   net.IPv4(0, 0, 0, 0),
		Port: 8888,
	})
	if err != nil {
		fmt.Println("连接服务端失败:", err)
		return
	}
	defer socket.Close()
	sendData := []byte("Hello server")
	_, err = socket.Write(sendData) // 发送数据
	if err != nil {
		fmt.Println("发送数据失败，err:", err)
		return
	}
	data := make([]byte, 4096)
	n, remoteAddr, err := socket.ReadFromUDP(data) // 接收数据
	if err != nil {
		fmt.Println("接收数据失败，err:", err)
		return
	}
	fmt.Printf("收到的数据是:%v \n 服务端地址:%v\n 读到的字节数:%v\n", string(data[:n]), remoteAddr, n)
}