|
本文实例讲述了GO语言实现简单TCP服务的方法。分享给大家供大家参考。具体实现方法如下：
复制代码 代码如下:
package main
import (
"net"
"fmt"
)
var (nbsp;nbsp; maxRead = 1100
nbsp;nbsp;nbsp; msgStopnbsp;nbsp; = []byte("cmdStop")
nbsp;nbsp;nbsp; msgStartnbsp; = []byte("cmdContinue")
nbsp;nbsp;nbsp; )
func main() {
nbsp;
nbsp;nbsp;nbsp; hostAndPort := "localhost:54321"
nbsp;nbsp;nbsp; listener := initServer(hostAndPort)
nbsp;nbsp;nbsp; for {
nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp; conn, err := listener.Accept()
nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp; checkError(err, "Accept: ")
nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp; go connectionHandler(conn)
nbsp;nbsp;nbsp; }
}
func initServer(hostAndPort string) *net.TCPListener {
nbsp;nbsp;nbsp; serverAddr, err := net.ResolveTCPAddr("tcp", hostAndPort)
nbsp;nbsp;nbsp; checkError(err, "Resolving address:port failed: '" + hostAndPort + "'")
nbsp;nbsp;nbsp; listener, err := net.ListenTCP("tcp", serverAddr)
nbsp;nbsp;nbsp; checkError(err, "ListenTCP: ")
nbsp;nbsp;nbsp; println("Listening to: ", listener.Addr().String())
nbsp;nbsp;nbsp; return listener
}
func connectionHandler(conn net.Conn) {
nbsp;nbsp;nbsp; connFrom := conn.RemoteAddr().String()
nbsp;nbsp;nbsp; println("Connection from: ", connFrom)
nbsp;nbsp;nbsp; talktoclients(conn)
nbsp;nbsp;nbsp; for {
nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp; var ibuf []byte = make([]byte, maxRead + 1)
nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp; length, err := conn.Read(ibuf[0:maxRead])
nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp; ibuf[maxRead] = 0 // to prevent overflow
nbsp;nbsp;nbsp; switch err {
nbsp;nbsp;nbsp; case nil:
nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp; handleMsg(length, err, ibuf)
nbsp;nbsp;nbsp; default:
nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp; goto DISCONNECT
nbsp;nbsp;nbsp; }
nbsp;nbsp;nbsp; }
nbsp;nbsp;nbsp; DISCONNECT:
nbsp;nbsp;nbsp; err := conn.Close()
nbsp;nbsp;nbsp; println("Closed connection:" , connFrom)
nbsp;nbsp;nbsp; checkError(err, "Close:" )
nbsp;nbsp;nbsp; }
func talktoclients(to net.Conn) {
nbsp;nbsp;nbsp; wrote, err := to.Write(msgStart)
nbsp;nbsp;nbsp; checkError(err, "Write: wrote " + string(wrote) + " bytes.")
}
func handleMsg(length int, err error, msg []byte) {
nbsp;nbsp;nbsp; if length gt; 0 {
nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;
nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp; for i := 0; ; i++ {
nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp; if msg[i] == 0 {
nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp; break
nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp; }
nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp; }
nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp; fmt.Printf("Received data: %v", string(msg[0:length]))
nbsp;nbsp;nbsp;nbsp;nbsp;nbsp;nbsp; fmt.Println("nbsp;nbsp; length:",length)
nbsp;nbsp;nbsp; }
}
func checkError(error error, info string) {
nbsp;nbsp;nbsp; if error != nil {
panic("ERROR: " + info + " " + error.Error()) // terminate program
}
}
希望本文所述对大家的Go语言程序设计有所帮助。 （编辑：源码网）
【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!
|
