golang 实现一个 tcp服务器 / 客户端
原创
©著作权归作者所有:来自51CTO博客作者冰糖豆豆的原创作品,请联系作者获取转载授权,否则将追究法律责任
源码说明:
"fmt"包是格式化打印相关的包
"log"包是日志服务的包
"net"是网络相关的包,实现了tcp、http、https、udp等包
"time"是系统时间相关的包
项目功能描述:
服务器在127.0.0.1:8000地址接受客户端的连接请求。连接成功后,为每一个连接创建一个go程,go程里面以固定定时时间(5s)发送心跳包,兄跳包内容为当前戳。
客户端单独开启一个go程,延时3秒后,以127.0.0.1:8000为目标地址,开始连接到回显服务器。进入for循环(5次),10秒一次从echo server获取包,并打印。 向echo server发送一个http空包:“HEAD / HTTP/1.1\r\n\r\n”。
使用vscode调试,客户端,服务器写指一个源码文件中,分别占一个go程去运行。源码如下:
package main
import (
"fmt"
"log"
"net"
"time"
)
// tcp echo server,向客户端以固定周期发送时间戳字符串
func echo(conn *net.TCPConn) {
tick := time.Tick(5 * time.Second) // 五秒的心跳间隔
for now := range tick {
n, err := conn.Write([]byte(now.String()))
if err != nil {
log.Println(err)
conn.Close()
return
}
fmt.Printf("send %d bytes to %s\n", n, conn.RemoteAddr())
}
}
// tcp客户端:读取echo server发送的时间戳
func client() {
time.Sleep(3 * time.Second)
/*if len(os.Args) != 2 {
log.Fatalf("Usage: %s host:port", os.Args[0])
}*/
//service := os.Args[1]
service := "127.0.0.1:8000"
tcpAddr, err := net.ResolveTCPAddr("tcp4", service)
if err != nil {
log.Fatal(err)
}
conn, err := net.DialTCP("tcp4", nil, tcpAddr)
if err != nil {
log.Fatal(err)
}
for i := 0; i < 5; i++ {
var buf [512]byte
n, err := conn.Read(buf[0:])
if err != nil {
log.Fatal(err)
fmt.Println("client write message error")
} else {
fmt.Println("read from server, message:", string(buf[:]))
}
n, err = conn.Write([]byte("HEAD / HTTP/1.1\r\n\r\n"))
if err != nil {
log.Fatal(err)
fmt.Println("client write message error")
}
time.Sleep(10 * time.Second)
//log.Fatal(n)
fmt.Println(n)
}
}
func main() {
go client()
address := net.TCPAddr{
IP: net.ParseIP("127.0.0.1"), // 把字符串IP地址转换为net.IP类型
Port: 8000,
}
listener, err := net.ListenTCP("tcp4", &address) // 创建TCP4服务器端监听器
if err != nil {
log.Fatal(err) // Println + os.Exit(1)
}
for {
conn, err := listener.AcceptTCP()
if err != nil {
log.Fatal(err) // 错误直接退出
}
fmt.Println("remote address:", conn.RemoteAddr())
go echo(conn)
}
}
调试效果如下截图: