Socket服務(wù)器是網(wǎng)絡(luò)服務(wù)中常用的服務(wù)器。使用go語言實現(xiàn)這個業(yè)務(wù)場景是很容易的。
這樣的網(wǎng)絡(luò)通訊,需要一個服務(wù)端和至少一個客戶端。
我們計劃構(gòu)建一個這樣的通訊工程。服務(wù)端啟動后等待客戶端的訪問??蛻舳税l(fā)送一段信息給服務(wù)端。服務(wù)端接收到信息后,再回饋給客戶端一段信息。
首先要建立服務(wù)端。服務(wù)端最先要做的事情就是"建立Socket端口監(jiān)聽"。
netListen, err := net.Listen("tcp", "localhost:1024")
上面的代碼,表名監(jiān)聽的是本機端口1024,而使用的通訊協(xié)議是TCP。
當(dāng)監(jiān)聽結(jié)束,模塊任務(wù)完成后,最后要close這個netListen。
defer netListen.Close()
使用日志功能,讓服務(wù)端窗口能看到服務(wù)端已經(jīng)運行了。
Log("Waiting for clients ...")
之后使用一個for循環(huán),無盡的等待那些不知何時來訪問的客戶端信息。
for循環(huán)體內(nèi),要監(jiān)聽netListen的信息接收情況:
conn, err := netListen.Accept()
當(dāng)有來自客戶端的訪問時,接受訪問。并在服務(wù)端的日志記錄已經(jīng)有客戶端連接成功了。
Log(conn.RemoteAddr().String(), "tcp connect success")
conn.RemoteAddr().String()表示的就是遠程客戶端。
然后,我們開啟一個goroutine處理連接任務(wù)。
go handleConnection(conn)
處理過程就是接收客戶端信息和反饋給客戶端信息。
n, err := conn.Read(buffer)
conn.Write([]byte(strTemp))
服務(wù)端代碼示例
package main
import (
"net"
"fmt"
"os"
"log"
"time"
)
func main() {
//建立socket端口監(jiān)聽
netListen, err := net.Listen("tcp", "localhost:1024")
CheckError(err)
defer netListen.Close()
Log("Waiting for clients ...")
//等待客戶端訪問
for{
conn, err := netListen.Accept() //監(jiān)聽接收
if err != nil{
continue //如果發(fā)生錯誤,繼續(xù)下一個循環(huán)。
}
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) //讀取客戶端傳來的內(nèi)容
if err != nil{
Log(conn.RemoteAddr().String(), "connection error: ", err)
return //當(dāng)遠程客戶端連接發(fā)生錯誤(斷開)后,終止此協(xié)程。
}
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())
}
}
客戶端的業(yè)務(wù)邏輯是,發(fā)送信息給服務(wù)端,然后接收服務(wù)端的反饋。
conn, err := net.DialTCP("tcp", nil, tcpAddr)
用TCP協(xié)議撥號(Dial)到服務(wù)端。如果沒有發(fā)生錯誤,就說明撥通了。于是在客戶端日志記錄連接成功
fmt.Println("connection success")
然后在這個已經(jīng)通暢的連接里,進行發(fā)送和接收信息的任務(wù)。conn.Write([]byte(words))是發(fā)送信息;conn.Read(buffer)是接收信息。如果接收發(fā)生錯誤,就記錄錯誤:
Log(conn.RemoteAddr().String(), "waiting server back msg error: ", err)
并且中斷進程。
如果沒有發(fā)生錯誤,酒吧接收到的信息在日志中記錄。
Log(conn.RemoteAddr().String(), "receive server back msg: ", string(buffer[:n]))
客戶端代碼示例
package main
import (
"net"
"fmt"
"log"
"os"
)
//發(fā)送信息
func sender(conn net.Conn) {
words := "Hello Server!"
conn.Write([]byte(words))
fmt.Println("send over")
//接收服務(wù)端反饋
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)
}
以上為個人經(jīng)驗,希望能給大家一個參考,也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方,望不吝賜教。
您可能感興趣的文章:- golang 實現(xiàn)tcp server端和client端,并計算RTT時間操作
- 使用Go基于WebSocket構(gòu)建千萬級視頻直播彈幕系統(tǒng)的代碼詳解
- golang websocket 服務(wù)端的實現(xiàn)
- golang socket斷點續(xù)傳大文件的實現(xiàn)方法
- golang基于websocket實現(xiàn)的簡易聊天室程序
- golang網(wǎng)絡(luò)socket粘包問題的解決方法