Golang實現簡單tcp服務器02 -- 實現echo服務器/客戶端

用Golang實現 echo服務器/客戶端

本節咱們就從實現一個簡單的echo的服務端/客戶端來入手, 瞭解golang的實現tcp長鏈接服務器的具體細節.

首先, 咱們先列一下服務端的實現思路及步驟:

1. 建立一個套接字對象, 指定其IP以及端口.
2. 開始監聽套接字指定的端口.
3. 若有新的客戶端鏈接請求, 則創建一個goroutine, 在goroutine中, 讀取客戶端消息, 並轉發回去, 直到客戶端斷開鏈接
4. 主進程繼續監聽端口.

咱們能夠在實驗環境的主文件夾中, 創建一個名爲server.go的文件, 在其中編寫服務器端程序代碼
服務端程序清單以下:

server.go

package main

import (
	"bufio"
	"fmt"
	"net"
	"time"
)

func main() {
	var tcpAddr *net.TCPAddr

	tcpAddr, _ = net.ResolveTCPAddr("tcp", "127.0.0.1:9999")

	tcpListener, _ := net.ListenTCP("tcp", tcpAddr)

	defer tcpListener.Close()

	for {
		tcpConn, err := tcpListener.AcceptTCP()
		if err != nil {
			continue
		}

		fmt.Println("A client connected : " + tcpConn.RemoteAddr().String())
		go tcpPipe(tcpConn)
	}

}

func tcpPipe(conn *net.TCPConn) {
	ipStr := conn.RemoteAddr().String()
	defer func() {
		fmt.Println("disconnected :" + ipStr)
		conn.Close()
	}()
	reader := bufio.NewReader(conn)

	for {
		message, err := reader.ReadString('\n')
		if err != nil {
			return
		}

		fmt.Println(string(message))
		msg := time.Now().String() + "\n"
		b := []byte(msg)
		conn.Write(b)
	}
}

接着, 咱們打開終端, 編譯服務端程序:

go build server.go

編譯成功的話, 會在主目錄中看到編譯成功的server程序

接下來, 是客戶端的代碼實現步驟:

1. 建立一個套接字對象, ip與端口指定到上面咱們實現的服務器的ip與端口上.
2. 使用建立好的套接字對象鏈接服務器.
3. 鏈接成功後, 開啓一個goroutine, 在這個goroutine內, 定時的向服務器發送消息, 並接受服務器的返回消息, 直到錯誤發生或斷開鏈接.

程序清單以下:

client.go

package main

import (
	"bufio"
	"fmt"
	"net"
	"time"
)

var quitSemaphore chan bool

func main() {
	var tcpAddr *net.TCPAddr
	tcpAddr, _ = net.ResolveTCPAddr("tcp", "127.0.0.1:9999")

	conn, _ := net.DialTCP("tcp", nil, tcpAddr)
	defer conn.Close()
	fmt.Println("connected!")

	go onMessageRecived(conn)

	b := []byte("time\n")
	conn.Write(b)

	<-quitSemaphore
}

func onMessageRecived(conn *net.TCPConn) {
	reader := bufio.NewReader(conn)
	for {
		msg, err := reader.ReadString('\n')
		fmt.Println(msg)
		if err != nil {
			quitSemaphore <- true
			break
		}
		time.Sleep(time.Second)
		b := []byte(msg)
		conn.Write(b)
	}
}

編譯客戶端:

go build client.go

最後, 開啓兩個終端, 分別運行server和client

能夠看到如下相似的輸出:

connected! 2015-03-19 23:42:08.4875559 +0800 CST

2015-03-19 23:42:09.4896132 +0800 CST

2015-03-19 23:42:10.4906704 +0800 CST

2015-03-19 23:42:11.4917277 +0800 CST

2015-03-19 23:42:12.4927849 +0800 CST

2015-03-19 23:42:13.4938422 +0800 CST

2015-03-19 23:42:14.4948995 +0800 CST

2015-03-19 23:42:15.4959567 +0800 CST

2015-03-19 23:42:16.497014 +0800 CST

2015-03-19 23:42:17.4980712 +0800 CST

2015-03-19 23:42:18.4991285 +0800 CST

2015-03-19 23:42:19.5001857 +0800 CST

這樣, 一個簡單的echo服務器/客戶端就實現了

相關源碼: https://git.oschina.net/victoriest/go-simple-tcp-server.git