草庐IT

golang中的socket编程

liuyuede123 2023-03-28 原文

0.1、索引

https://waterflow.link/articles/1664591292871

1、tcp的3次握手(建立连接)

  1. 客户端的协议栈向服务器端发送了 SYN 包,并告诉服务器端当前发送序列号 j,客户端进入 SYNC_SENT 状态;
  2. 服务器端的协议栈收到这个包之后,和客户端进行 ACK 应答,应答的值为 j+1,表示对 SYN 包 j 的确认,同时服务器也发送一个 SYN 包,告诉客户端当前我的发送序列号为 k,服务器端进入 SYNC_RCVD 状态;
  3. 客户端协议栈收到 ACK 之后,使得应用程序从 connect 调用返回,表示客户端到服务器端的单向连接建立成功,客户端的状态为 ESTABLISHED,同时客户端协议栈也会对服务器端的 SYN 包进行应答,应答数据为 k+1;
  4. 应答包到达服务器端后,服务器端协议栈使得 accept 阻塞调用返回,这个时候服务器端到客户端的单向连接也建立成功,服务器端也进入 ESTABLISHED 状态。

2、tcp的4次挥手(关闭连接)

  1. 一方应用程序调用 close,我们称该方为主动关闭方,该端的 TCP 发送一个 FIN 包,表示需要关闭连接。之后主动关闭方进入 FIN_WAIT_1 状态。
  2. 接收到这个 FIN 包的对端执行被动关闭。这个 FIN 由 TCP 协议栈处理,我们知道,TCP 协议栈为 FIN 包插入一个文件结束符 EOF 到接收缓冲区中,应用程序可以通过 read 调用来感知这个 FIN 包。一定要注意,这个 EOF 会被放在已排队等候的其他已接收的数据之后,这就意味着接收端应用程序需要处理这种异常情况,因为 EOF 表示在该连接上再无额外数据到达。此时,被动关闭方进入 CLOSE_WAIT 状态。
  3. 被动关闭方将读到这个 EOF,于是,应用程序也调用 close 关闭它的套接字,这导致它的 TCP 也发送一个 FIN 包。这样,被动关闭方将进入 LAST_ACK 状态。
  4. 主动关闭方接收到对方的 FIN 包,并确认这个 FIN 包。主动关闭方进入 TIME_WAIT 状态,而接收到 ACK 的被动关闭方则进入 CLOSED 状态。进过 2MSL 时间之后,主动关闭方也进入 CLOSED 状态。

3、socket中的连接建立和关闭

我看先看下流程:

  1. 服务端调用socket、bind绑定ip端口、listen开启服务端监听。
  2. accept阻塞等待下次调用,并返回一个tcp连接。
  3. 客户端调用connect连接服务端。
  4. 此时服务端accept结束阻塞,代表客户端和服务端成功建立连接。
  5. 然后就是数据交互读写读写。
  6. 当客户端连接关闭时,服务端的read方法会读取一个io.EOF的错误,代表客户端关闭连接。服务端收到关闭连接的错误后也调用close关闭连接。

4、golang中的连接建立

我们先看下服务端:

package main

import (
	"fmt"
	"net"
)

func main() {
	server := ":8330"
	tcpAddr, err := net.ResolveTCPAddr("tcp", server)
	if err != nil {
		fmt.Println("resolve err:", err)
		return
	}

  // 监听某个端口的tcp网络
	listen, err := net.ListenTCP("tcp", tcpAddr)
	if err != nil {
		fmt.Println("listen err:", err)
		return
	}
	defer listen.Close()

	for {
    // 等待下次请求过来并建立连接
		conn, err := listen.Accept()
		if err != nil {
			fmt.Println("accept err:", err)
			continue
		}

    // 在这个连接上做一些事情
		go handler(conn)

	}
}

func handler(conn net.Conn) {
}
  1. 首先我们定义好ip和端口,开启监听
  2. 然后调用accept等待下次请求过来,并建立tcp连接

我们运行下上面的代码:

go run server.go

然后在另一个shell中执行下面的命令:

watch -d 'netstat -nat |grep "8330"'

Every 2.0s: netstat -nat |grep "8330"                                                 userdeMacBook-Pro.local: Thu Sep 29 16:38:42 2022

tcp46      0      0  *.8330                 *.*                    LISTEN

可以看到此时8330端口已经开启监听

客户端:

package main

import (
	"fmt"
	"net"
)

func main() {
	serverAddr := ":8330"

	tcpAddr, err := net.ResolveTCPAddr("tcp", serverAddr)
	if err != nil {
		fmt.Println("resolve err:", err)
		return
	}

  // 发起一个tcp的网络拨号
	_, err = net.DialTCP("tcp", nil, tcpAddr)
	if err != nil {
		fmt.Println("dial err:", err)
		return
	}


	closed := make(chan bool)


  // 客户端阻塞不直接关闭
	for  {
		select {
		case <-closed:
			fmt.Println("服务端关闭")
			return
		}
	}

}

其中核心的方法就是net.DialTCP,第一个参数会返回一个建立成功的连接,第二个参数会返回没建立成功的错误信息。

然后我们命令行执行下:

go run client.go

接着看下watch -d 'netstat -nat |grep "8330"'的返回,这个命令是实时的,所以不需要重复执行

Every 2.0s: netstat -nat |grep "8330"                                                 userdeMacBook-Pro.local: Thu Sep 29 16:45:57 2022

tcp4       0      0  127.0.0.1.8330         127.0.0.1.59146        ESTABLISHED
tcp4       0      0  127.0.0.1.59146        127.0.0.1.8330         ESTABLISHED
tcp46      0      0  *.8330                 *.*                    LISTEN

可以看到客户端服务端,服务端和客户端都成功建立了连接(连接是否建立成功不是看是否有条线真连上了,连接状态是维护在各个端的)

同时我们也可以在wireshark中看到三次握手建立连接的流程:

5、golang中的读和写

我们现在稍微修改下服务端的代码:

package main

import (
	"fmt"
	"io"
	"net"
	"time"
)

func main() {
	server := ":8330"
	tcpAddr, err := net.ResolveTCPAddr("tcp", server)
	if err != nil {
		fmt.Println("resolve err:", err)
		return
	}

	listen, err := net.ListenTCP("tcp", tcpAddr)
	if err != nil {
		fmt.Println("listen err:", err)
		return
	}
	defer listen.Close()

	for {
		conn, err := listen.Accept()
		if err != nil {
			fmt.Println("accept err:", err)
			continue
		}

		go handler(conn)

	}
}

func handler(conn net.Conn) {

	go func() {
		for  {
      // 指定从buffer中读取数据的最大容量
			var buf = make([]byte, 1024)
      // 从buffer中读取数据并保存到buf中,n代表实际返回的数据大小
			n, err := conn.Read(buf)
			if err != nil {
        // 客户端关闭会触发EOF
				if err == io.EOF {
					conn.Close()
					return
				}
				fmt.Println("read err:", err)
				return
			}

			fmt.Println("read data ", n, ":", string(buf))
		}
	}()


	curTime := time.Now().String()
  // 数据写到缓冲区
	_, err := conn.Write([]byte(curTime))
	if err != nil {
		fmt.Println("write err:", err)
		return
	}
	fmt.Println("send data:", curTime)

}

首先要明白,操作系统内核会为每个连接的客户端和服务端分配发送缓冲区接收缓冲区

  1. 当客户端需要发送数据到服务端,调用conn.Write从客户端缓冲区发送数据到操作系统内核的发送缓冲区。实际所做的事情是把数据从应用程序缓冲区中拷贝到操作系统内核的发送缓冲区中,并不一定是把数据通过套接字写出去。
  2. 数据通过tcp发送到服务端的接收缓冲区,然后服务端的程序从接收缓冲区读取数据。

非阻塞I/O,当应用程序调用非阻塞 I/O 完成某个操作时,内核立即返回,不会把 CPU 时间切换给其他进程,应用程序在返回后,可以得到足够的 CPU 时间继续完成其他事情。

读操作:如果套接字对应的接收缓冲区没有数据可读,在非阻塞情况下 read 调用会立即返回,一般返回 EWOULDBLOCK 或 EAGAIN 出错信息。

写操作:在非阻塞 I/O 的情况下,如果套接字的发送缓冲区已达到了极限,不能容纳更多的字节,那么操作系统内核会尽最大可能从应用程序拷贝数据到发送缓冲区中,并立即从 write 等函数调用中返回。可想而知,在拷贝动作发生的瞬间,有可能一个字符也没拷贝,有可能所有请求字符都被拷贝完成,那么这个时候就需要返回一个数值,告诉应用程序到底有多少数据被成功拷贝到了发送缓冲区中,应用程序需要再次调用 write 函数,以输出未完成拷贝的字节。

非阻塞 I/O 操作:拷贝→返回→再拷贝→再返回。

阻塞 I/O 操作:拷贝→直到所有数据拷贝至发送缓冲区完成→返回。

golang中底层使用的还是非阻塞的I/O,但是在代码层面做了一些处理,让用户感觉是以阻塞方式调用的。

...

for {
		n, err := ignoringEINTRIO(syscall.Read, fd.Sysfd, p)
		if err != nil {
			n = 0
      // 非阻塞方式调用,如果遇到syscall.EAGAIN报错,代表没拿到数据,继续循环
			if err == syscall.EAGAIN && fd.pd.pollable() {
				if err = fd.pd.waitRead(fd.isFile); err == nil {
					continue
				}
			}
		}
		err = fd.eofError(n, err)
		return n, err
	}

...

6、golang中的关闭

在socket中,当客户端调用close()方法时,其实就是发送一个FIN标志位,意思就是我要主动关闭TCP连接了。Close方法会让对端的所有读写操作结束阻塞,并返回。

在golang中调用Close方法,会让对端的Read读取到EOF的错误,此时就代表我想关闭连接。对端接收到关闭的请求后也可以调用Close方法关闭连接。

客户端:

package main

import (
	"fmt"
	"io"
	"net"
)

func main() {
	serverAddr := ":8330"

	tcpAddr, err := net.ResolveTCPAddr("tcp", serverAddr)
	if err != nil {
		fmt.Println("resolve err:", err)
		return
	}

	conn, err := net.DialTCP("tcp", nil, tcpAddr)
	if err != nil {
		fmt.Println("dial err:", err)
		return
	}


	closed := make(chan bool)
	go func() {
		for  {
			var buf = make([]byte, 1024)
			n, err := conn.Read(buf)
			if err != nil {
        // 读取到EOF,服务端关闭连接
				if err == io.EOF {
					conn.Close()
					closed <- true
					return
				}
				fmt.Println("read err:", err)
				return
			}

			fmt.Println("read data ", n, ":", string(buf))
		}
	}()

	for  {
		select {
		case <-closed:
			fmt.Println("服务端关闭")
			return
		}
	}

}

服务端:

package main

import (
	"fmt"
	"io"
	"net"
	"time"
)

func main() {
	server := ":8330"
	tcpAddr, err := net.ResolveTCPAddr("tcp", server)
	if err != nil {
		fmt.Println("resolve err:", err)
		return
	}

	listen, err := net.ListenTCP("tcp", tcpAddr)
	if err != nil {
		fmt.Println("listen err:", err)
		return
	}
	defer listen.Close()

	for {
		conn, err := listen.Accept()
		if err != nil {
			fmt.Println("accept err:", err)
			continue
		}

		go handler(conn)

	}
}

func handler(conn net.Conn) {

	go func() {
		for  {
			var buf = make([]byte, 1024)
			n, err := conn.Read(buf)
			if err != nil {
        // 读取到EOF,客户端关闭连接
				if err == io.EOF {
					conn.Close()
					return
				}
				fmt.Println("read err:", err)
				return
			}

			fmt.Println("read data ", n, ":", string(buf))
		}
	}()


	curTime := time.Now().String()
	_, err := conn.Write([]byte(curTime))
	if err != nil {
		fmt.Println("write err:", err)
		return
	}
	fmt.Println("send data:", curTime)

}

参考:

《极客时间:网络编程实战》

有关golang中的socket编程的更多相关文章

  1. ruby - 如何从 ruby​​ 中的字符串运行任意对象方法? - 2

    总的来说,我对ruby​​还比较陌生,我正在为我正在创建的对象编写一些rspec测试用例。许多测试用例都非常基础,我只是想确保正确填充和返回值。我想知道是否有办法使用循环结构来执行此操作。不必为我要测试的每个方法都设置一个assertEquals。例如:describeitem,"TestingtheItem"doit"willhaveanullvaluetostart"doitem=Item.new#HereIcoulddotheitem.name.shouldbe_nil#thenIcoulddoitem.category.shouldbe_nilendend但我想要一些方法来使用

  2. ruby - 其他文件中的 Rake 任务 - 2

    我试图在一个项目中使用rake,如果我把所有东西都放到Rakefile中,它会很大并且很难读取/找到东西,所以我试着将每个命名空间放在lib/rake中它自己的文件中,我添加了这个到我的rake文件的顶部:Dir['#{File.dirname(__FILE__)}/lib/rake/*.rake'].map{|f|requiref}它加载文件没问题,但没有任务。我现在只有一个.rake文件作为测试,名为“servers.rake”,它看起来像这样:namespace:serverdotask:testdoputs"test"endend所以当我运行rakeserver:testid时

  3. ruby-on-rails - Ruby net/ldap 模块中的内存泄漏 - 2

    作为我的Rails应用程序的一部分,我编写了一个小导入程序,它从我们的LDAP系统中吸取数据并将其塞入一个用户表中。不幸的是,与LDAP相关的代码在遍历我们的32K用户时泄漏了大量内存,我一直无法弄清楚如何解决这个问题。这个问题似乎在某种程度上与LDAP库有关,因为当我删除对LDAP内容的调用时,内存使用情况会很好地稳定下来。此外,不断增加的对象是Net::BER::BerIdentifiedString和Net::BER::BerIdentifiedArray,它们都是LDAP库的一部分。当我运行导入时,内存使用量最终达到超过1GB的峰值。如果问题存在,我需要找到一些方法来更正我的代

  4. ruby-on-rails - Rails 3 中的多个路由文件 - 2

    Rails2.3可以选择随时使用RouteSet#add_configuration_file添加更多路由。是否可以在Rails3项目中做同样的事情? 最佳答案 在config/application.rb中:config.paths.config.routes在Rails3.2(也可能是Rails3.1)中,使用:config.paths["config/routes"] 关于ruby-on-rails-Rails3中的多个路由文件,我们在StackOverflow上找到一个类似的问题

  5. ruby-openid:执行发现时未设置@socket - 2

    我在使用omniauth/openid时遇到了一些麻烦。在尝试进行身份验证时,我在日志中发现了这一点:OpenID::FetchingError:Errorfetchinghttps://www.google.com/accounts/o8/.well-known/host-meta?hd=profiles.google.com%2Fmy_username:undefinedmethod`io'fornil:NilClass重要的是undefinedmethodio'fornil:NilClass来自openid/fetchers.rb,在下面的代码片段中:moduleNetclass

  6. ruby-on-rails - Rails - 一个 View 中的多个模型 - 2

    我需要从一个View访问多个模型。以前,我的links_controller仅用于提供以不同方式排序的链接资源。现在我想包括一个部分(我假设)显示按分数排序的顶级用户(@users=User.all.sort_by(&:score))我知道我可以将此代码插入每个链接操作并从View访问它,但这似乎不是“ruby方式”,我将需要在不久的将来访问更多模型。这可能会变得很脏,是否有针对这种情况的任何技术?注意事项:我认为我的应用程序正朝着单一格式和动态页面内容的方向发展,本质上是一个典型的网络应用程序。我知道before_filter但考虑到我希望应用程序进入的方向,这似乎很麻烦。最终从任何

  7. ruby-on-rails - Rails 3.2.1 中 ActionMailer 中的未定义方法 'default_content_type=' - 2

    我在我的项目中添加了一个系统来重置用户密码并通过电子邮件将密码发送给他,以防他忘记密码。昨天它运行良好(当我实现它时)。当我今天尝试启动服务器时,出现以下错误。=>BootingWEBrick=>Rails3.2.1applicationstartingindevelopmentonhttp://0.0.0.0:3000=>Callwith-dtodetach=>Ctrl-CtoshutdownserverExiting/Users/vinayshenoy/.rvm/gems/ruby-1.9.3-p0/gems/actionmailer-3.2.1/lib/action_mailer

  8. ruby-on-rails - Rails 应用程序中的 Rails : How are you using application_controller. rb 是新手吗? - 2

    刚入门rails,开始慢慢理解。有人可以解释或给我一些关于在application_controller中编码的好处或时间和原因的想法吗?有哪些用例。您如何为Rails应用程序使用应用程序Controller?我不想在那里放太多代码,因为据我了解,每个请求都会调用此Controller。这是真的? 最佳答案 ApplicationController实际上是您应用程序中的每个其他Controller都将从中继承的类(尽管这不是强制性的)。我同意不要用太多代码弄乱它并保持干净整洁的态度,尽管在某些情况下ApplicationContr

  9. ruby-on-rails - form_for 中不在模型中的自定义字段 - 2

    我想向我的Controller传递一个参数,它是一个简单的复选框,但我不知道如何在模型的form_for中引入它,这是我的观点:{:id=>'go_finance'}do|f|%>Transferirde:para:Entrada:"input",:placeholder=>"Quantofoiganho?"%>Saída:"output",:placeholder=>"Quantofoigasto?"%>Nota:我想做一个额外的复选框,但我该怎么做,模型中没有一个对象,而是一个要检查的对象,以便在Controller中创建一个ifelse,如果没有检查,请帮助我,非常感谢,谢谢

  10. ruby - rspec 需要 .rspec 文件中的 spec_helper - 2

    我注意到像bundler这样的项目在每个specfile中执行requirespec_helper我还注意到rspec使用选项--require,它允许您在引导rspec时要求一个文件。您还可以将其添加到.rspec文件中,因此只要您运行不带参数的rspec就会添加它。使用上述方法有什么缺点可以解释为什么像bundler这样的项目选择在每个规范文件中都需要spec_helper吗? 最佳答案 我不在Bundler上工作,所以我不能直接谈论他们的做法。并非所有项目都checkin.rspec文件。原因是这个文件,通常按照当前的惯例,只

随机推荐