client端
一个TCP客户端进行TCP通信的流程如下:
- 向指定ip和端口拨号,请求建立连接(Tcp或Udp)
- 进行数据收发(从终端和从服务端读数据)
- 关闭链接
package main
import (
"bufio"
"fmt"
"net"
"os"
"strings"
)
func main() {
conn, err := net.Dial("tcp", "127.0.0.1:8888")
if err != nil {
fmt.Println("client dial err=", err)
return
}
reader := bufio.NewReader(os.Stdin)
for {
//从终端读取数据(会以文件的形式)
line, err := reader.ReadString('\n')
if err != nil {
fmt.Println("readstring err=", err)
}
//如多用户输入exit就退出
line = strings.Trim(line, "\r\n")
if line == "exit" {
fmt.Println("客户端退出")
return
}
//将line发给服务器
n, err := conn.Write([]byte(line + "\n"))
if err != nil {
fmt.Println("conn.Write err=", err)
}
fmt.Printf("发送了%d个字节\n",n)
//创建切片
buf := make([]byte, 1024)
//1 等待客户端通过conn发送信息
//2 如果没有writer发送就一直阻塞在这
n, err = conn.Read(buf)
if err != nil {
fmt.Println("服务器read err=", err) //出错退出
return
}
//3. 显示读取内容到终端
fmt.Print(string(buf[:n]))
}
}
server端
server端TCP服务端程序的处理流程:
- 监听端口
- 接收客户端请求建立链接
- 创建goroutine处理链接(数据的收发)。
package main
import (
"bufio"
"fmt"
"net"
"os"
"strings"
)
func process(conn net.Conn) {
//循环接受客户端发送数据,不让协程退出
defer conn.Close() //关闭conn连接
for {
//创建切片
buf := make([]byte, 1024)
//1 等待客户端通过conn发送信息
//2 如果没有writer发送就一直阻塞
n, err := conn.Read(buf)
if err != nil {
fmt.Println("服务器read err=", err) //出错退出
return
}
//3. 显示读取内容到终端
fmt.Print(string(buf[:n]))
//发送消息给客户端
reader := bufio.NewReader(os.Stdin)
//从终端读取数据(会以文件的形式)
line, err := reader.ReadString('\n')
if err != nil {
fmt.Println("readstring err=", err)
}
//如多用户输入exit就退出
line = strings.Trim(line, "\r\n")
if line == "exit" {
fmt.Println("客户端退出")
return
}
//将line发给服务器
_, err = conn.Write([]byte(line + "\n"))
if err != nil {
fmt.Println("conn.Write err=", err)
}
}
}
func main() {
fmt.Println("服务器开始监听------")
listen, err := net.Listen("tcp", "0.0.0.0:8888")
if err != nil {
fmt.Println("listen err=", err)
return
}
defer listen.Close() //延时关闭listen
for {
fmt.Println("-----等待客户连接----")
conn, err := listen.Accept()
if err != nil {
fmt.Println("Accept err=", err)
} else {
fmt.Printf("Accept success con类型%T,客户ip%v\n", conn, conn.RemoteAddr().String())
}
//这里要起一个协程
go process(conn)
}
}
上面的client和server只是对tcp通信进行了简单的模拟,旨在联系tcp的通信过程。
代码还有个bug,就是client和server都只能收一句、发一句、收一句...交替进行,因为程序中收发是顺序进行的,例如client,发送一次数据之后,程序就阻塞到了 n, err := conn.Read(buf) 这个位置,所以只有再server发送来数据并读取后,才可以再次在终端输入数据发送。
|
请发表评论