新钛云服已累计为您分享665篇技术干货
最近做了一个项目,其中用到了网络编程,下面和大家分享下在Go中网络编程的实现
在Go中, 网络编程主要通过 net 包实现。支持 包括TCP/IP、UDP、域名解析和Unix域socket等连接,此外,还通过 net/http ,net/rpc 等提供了 HTTP,RPC等主流应用层的连接协议。

1、TCP 服务

TCP 是最常用的网络连接方式,以 TCP 连接为例,一个简单的 TCP 连接代码示例。
Client:
package main


import (

"bufio"
"fmt"
"os"
"strings"
"net"
)


func 
main
() {

 conn, err := net.Dial(
"tcp"
"tyun.cn:8088"
)

if
 err != nil {

  //handle

 }

 defer conn.Close()

 inputReader := bufio.NewReader(os.Stdin)

for
{

  //读取输入

  readString,_ :=inputReader.ReadString(
'\n'
)

  trim := strings.Trim(readString,
"\r\n"
)

  _,err :=conn.Write([]byte(trim))

if
 err!=nil{

return
  }

  buf := [512]byte{}

  n,err := conn.Read(buf[:])

if
 err!=nil{

return
  }

  fmt.Println(string(buf[:n]))

 }


}

Server:
package main


import (

"bufio"
"fmt"
"net"
)


func 
main
() {

 listen,err:= net.Listen(
"tcp"
,
":8088"
)

if
 err != nil{

  //handle

 }

for
 {

  conn,err := listen.Accept()

if
 err != nil{

   //handle

continue
  }

  defer conn.Close()

for
 {

   //处理

   reader := bufio.NewReader(conn)

   var buf [1024]byte

read
,err:= reader.Read(buf[:])

if
 err != nil{

    //handle

   }

   recv := string(buf[:
read
])

   fmt.Println(
"GET MESSAGE: "
,recv)

   conn.Write([]byte(recv))

  }

 }

}

2、TCP 连接在系统调用层面的实现

包括 TCP/IP 在内的各种网络连接,在类 unix 的操作系统里,都是通过网络系统调用实现的,使用系统调用创建 TCP 服务器的核心流程是:
· 首先得能创建socket ,也就是得有个 create 的接口;
· 其次还得能绑定 ip 和端口号,也就是 bind,不然别人找不到你;
· 有了这些还得有点东西,那就是别人怎么连接你。也就是要有个 connect 和 accept 的功能;
· 再者还得实现读和写,也就是 read 和 write,因为我们要通过socket 通信,不能读不能写肯定不行;
· 通信完毕后socket 要关闭;
· 创建 socket 连接 s = socket(AF_INET, SOCK_STREAM, 0);
· 绑定 地址 bind(s, (struct sockaddr *) &serv_addr, sizeof(serv_addr));
· 启动监听 listen(s,5);
· 接收客户端的连接 ns= accept(s, (struct sockaddr *)&cli_addr, &clilen);
· 接收数据 n = read( ns,buffer,255 );
· 发送数据,n = write(ns,"hello world",11)。
创建 TCP 客户端的核心流程:
· 创建 socket 连接 s= socket(AF_INET, SOCK_STREAM, 0);
· 连接服务器地址:connect(s, (struct sockaddr*)&serv_addr, sizeof(serv_addr));
· 接收数据 n = read( ns,buffer,255 );
· 发送数据,n = write(ns,"hello world",11)。
socket() 等系统调用的实现,是 kernel 层面完成的,而net包就是对这些底层的流程做了封装。

3、Go中TCP连接的实现

在客户端中,建立连接conn, err := net.Dial("tcp", "tyun.cn:8088")的底层源码实现方法是:
func (d *Dialer) DialContext(ctx context.Context, network, address string) (Conn, error)
这个方法解析完网络和地址后 ,主要有2种连接方式:
if
 len(fallbacks) > 0 {
  c, err = sd.dialParallel(ctx, primaries, fallbacks)
else
 {
  c, err = sd.dialSerial(ctx, primaries)
}
而这两种连接方式的都是调用:
func (sd *sysDialer) dialSingle(ctx context.Context, ra Addr) (c Conn, err error)
sd.dialSingle又支持4种类型:
sd.dialTCP(ctx, la, ra)
sd.dialUDP(ctx, la, ra)
sd.dialIP(ctx, la, ra)
sd.dialUnix(ctx, la, ra)
接续往下追溯,就能看到其实调用的是socket函数,实现系统调用socket()。
如果多个 goroutine 对 conn 进行读写,就会有多重读,多重写两种情况,socket 是全双工,所以读写之间互不影响。多 goroutine 读的时候,其实没什么影响。因为读的话,不会重复读,多重读不会引发安全问题。
多 goroutine 写的时候,就有问题了。多个 goroutine 写不能每个写一半,必须保证每次写是原子操作,好在 Go内部实现写的时候加了锁。
TCP 连接建立后,每当客户端发来请求时,服务端会建立新的连接。
conn,err := listen.Accept()
这个方法同样往下追溯,能看到其实最底层的实现是系统调用syscall.Accept。
连接通信的过程中,如果某一方突然关闭,那另一方会有啥反应?在实践的过程中,总结如下:
· 当对端异常关闭,如果己方 socket 中有数据,那么己方是会继续读取;当 socket 中无数据时,己方 Read 函数返回 EOF;
· 当对端异常关闭,己方如果还在 socket 的缓冲中写入,那么本次写入成功,下次写入就报错;
· 当己端关闭,读写均异常错误。

4、结语

以上就是关于Go网络编程的一些分享,虽然net包提供了对网络原语的访问,但大多数使用者只需要Dial、Listen和Accpet函数的基本接口;以及Conn和Listener接口。
net包主要是增加 context 控制,封装了一些不同的连接类型以及DNS 查找等等,同时在有需要的地方引入 goroutine 提高处理效率。
了解新钛云服
往期技术干货
点👇分享
戳👇在看
继续阅读
阅读原文