51. Socket服务端和客户端使用TCP协议通讯 | 厚土Go学习笔记
51. Socket服务端和客户端使用TCP协议通讯 | 厚土Go学习笔记
李海彬
发布于 2018-03-16 17:09:37
发布于 2018-03-16 17:09:37
Socket服务器是网络服务中常用的服务器。使用 go 语言实现这个业务场景是很容易的。
这样的网络通讯，需要一个服务端和至少一个客户端。
我们计划构建一个这样的通讯过程。服务端启动后等待客户端的访问。客户端发送一段信息给服务端。服务端接收信息后，再回馈给客户端一段信息。
首先要建立服务端。 服务端最先要做的事情就是“建立Socket端口监听”
netListen, err := net.Listen("tcp", "localhost:1024")这样代码，表明监听的本机端口是 1024，而使用的通讯协议是 TCP。
当监听结束，模块任务完成时，最后要close这个netListen
defer netListen.Close()使用日志功能，让服务端窗口能看到服务已经运行了。
Log("Waiting for clients ...")之后使用一个 for 循环，无尽的等待那些不知何时来访问的客户端信息。
for 循环体内，要监听 netListen 的信息接收情况
conn, err := netListen.Accept()当有来自客户端的访问时，接收访问。并且在服务端的日志记录已经有客户端连接成功了。
Log(conn.RemoteAddr().String(), "tcp connect success")conn.RemoteAddr().String() 表示的就是远程客户端。
然后，我们开启一个 goroutine 处理连接任务。
go handleConnection(conn)处理过程就是接收客户端信息和反馈给客户端信息
n, err := conn.Read(buffer)
conn.Write([]byte(strTemp))服务端代码示例：
package main
import (
"net"
"fmt"
"os"
"log"
"time"
)
func main() {
//建立socket端口监听
netListen, err := net.Listen("tcp", "localhost:1024")
CheckError(err)
defer netListen.Close()
Log("Waiting for clients ...")
//等待客户端访问
for{
conn, err := netListen.Accept() //监听接收
if err != nil{
continue //如果发生错误，继续下一个循环。
}
Log(conn.RemoteAddr().String(), "tcp connect success") //tcp连接成功
go handleConnection(conn)
}
}
//处理连接
func handleConnection(conn net.Conn) {
buffer := make([]byte, 2048) //建立一个slice
for{
n, err := conn.Read(buffer) //读取客户端传来的内容
if err != nil{
Log(conn.RemoteAddr().String(), "connection error: ", err)
return //当远程客户端连接发生错误（断开）后，终止此协程。
}
Log(conn.RemoteAddr().String(), "receive data string:\n", string(buffer[:n]))
//返回给客户端的信息
strTemp := "CofoxServer got msg \""+string(buffer[:n])+"\" at "+time.Now().String()
conn.Write([]byte(strTemp))
}
}
//日志处理
func Log(v ...interface{}) {
log.Println(v...)
}
//错误处理
func CheckError(err error) {
if err != nil{
fmt.Fprintf(os.Stderr, "Fatal error: %s", err.Error())
}
}客户端的业务逻辑是，发送信息给服务端，然后接收服务端的反馈。
conn, err := net.DialTCP("tcp", nil, tcpAddr)用 TCP 协议拨号（Dial）到服务端。如果没有发生错误，就说明拨通了。于是在客户端日志记录连接成功
fmt.Println("connection success")然后在这个已经通畅的连接里，进行发送和接收信息的任务。
conn.Write([]byte(words)) 是发送信息
conn.Read(buffer) 是接收信息
如果接收发生错误，就记录错误
Log(conn.RemoteAddr().String(), "waiting server back msg error: ", err)并且中断进程。
如果没有发生错误，就把接收到的信息在日志中记录。
Log(conn.RemoteAddr().String(), "receive server back msg: ", string(buffer[:n]))客户端代码示例：
package main
import (
"net"
"fmt"
"log"
"os"
)
//发送信息
func sender(conn net.Conn) {
words := "Hello Server!"
conn.Write([]byte(words))
fmt.Println("send over")
//接收服务端反馈
buffer := make([]byte, 2048)
n, err := conn.Read(buffer)
if err != nil {
Log(conn.RemoteAddr().String(), "waiting server back msg error: ", err)
return
}
Log(conn.RemoteAddr().String(), "receive server back msg: ", string(buffer[:n]))
}
//日志
func Log(v ...interface{}) {
log.Println(v...)
}
func main() {
server := "127.0.0.1:1024"
tcpAddr, err := net.ResolveTCPAddr("tcp4", server)
if err != nil {
fmt.Fprintf(os.Stderr, "Fatal error: %s", err.Error())
os.Exit(1)
}
conn, err := net.DialTCP("tcp", nil, tcpAddr)
if err != nil {
fmt.Fprintf(os.Stderr, "Fatal error: %s", err.Error())
os.Exit(1)
}
fmt.Println("connection success")
sender(conn)
}
服务端运行效果
客户端运行效果
本文参与 腾讯云自媒体分享计划，分享自微信公众号。
原始发表：2018-02-20，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读