對(duì)長(zhǎng)、短連接的處理策略（模擬心跳）
作為一個(gè)可能會(huì)和很多Client進(jìn)行通訊交互的Server，首先要保證的就是整個(gè)Server運(yùn)行狀態(tài)的穩(wěn)定性，因此在和Client建立連接通訊的時(shí)候，確保連接的及時(shí)斷開(kāi)非常重要，否則一旦和多個(gè)客戶端建立不關(guān)閉的長(zhǎng)連接，對(duì)于服務(wù)器資源的占用是很可怕的。因此，我們需要針對(duì)可能出現(xiàn)的短連接和長(zhǎng)連接，設(shè)定不同的限制策略。
針對(duì)短連接，我們可以使用golang中的net包自帶的timeout函數(shù)，一共有三個(gè)，分別是：
復(fù)制代碼 代碼如下:
func (*IPConn) SetDeadline
func (c *IPConn) SetDeadline(t time.Time) error
func (*IPConn) SetReadDeadline
func (c *IPConn) SetReadDeadline(t time.Time) error
func (*IPConn) SetWriteDeadline
func (c *IPConn) SetWriteDeadline(t time.Time) error
如果想要給服務(wù)器設(shè)置短連接的timeout,我們就可以這么寫(xiě)：
復(fù)制代碼 代碼如下:
netListen, err := net.Listen("tcp", Port)
Log("Waiting for clients")
for {
conn, err := netListen.Accept()
if err != nil {
continue
}
conn.SetReadDeadline(time.Now().Add(time.Duration(10) * time.Second))
這里的三個(gè)函數(shù)都是用于設(shè)置每次socket連接能夠維持的最長(zhǎng)時(shí)間，一旦超過(guò)設(shè)置的timeout后，便會(huì)在Server端自動(dòng)斷開(kāi)連接。其中SetReadline, SetWriteline設(shè)置的是讀取和寫(xiě)入的最長(zhǎng)持續(xù)時(shí)間，而SetDeadline則同時(shí)包含了 SetReadline, SetWriteline兩個(gè)函數(shù)。
通過(guò)這樣設(shè)定，每個(gè)和Server通訊的Client連接時(shí)長(zhǎng)最長(zhǎng)也不會(huì)超過(guò)10s了~~
搞定短連接后，接下來(lái)就是針對(duì)長(zhǎng)連接的處理策略了~~
作為長(zhǎng)連接，由于我們往往很難確定什么時(shí)候會(huì)中斷連接，因此并不能像處理短連接那樣簡(jiǎn)單粗暴的設(shè)定一個(gè)timeout就可以搞定，而在Golang的net包中，并沒(méi)有針對(duì)長(zhǎng)連接的函數(shù)，因此需要我們自己設(shè)計(jì)并實(shí)現(xiàn)針對(duì)長(zhǎng)連接的處理策略啦~
針對(duì)socke長(zhǎng)連接，常見(jiàn)的做法是在Server和Socket之間設(shè)計(jì)通訊機(jī)制，當(dāng)兩者之間沒(méi)有信息交互時(shí)，雙方便會(huì)定時(shí)發(fā)送數(shù)據(jù)包(心跳)，以維持連接狀態(tài)。
這種方法是目前使用相對(duì)比較多的做法，但是開(kāi)銷(xiāo)相對(duì)也較大，特別是當(dāng)Server和多個(gè)client保持長(zhǎng)連接的時(shí)候，并發(fā)會(huì)比較高，考慮到公司的業(yè)務(wù)需求，我最后選擇了邏輯相對(duì)簡(jiǎn)單，開(kāi)銷(xiāo)相對(duì)較小的策略：
當(dāng)Server每次收到Client發(fā)到的信息之后，便會(huì)開(kāi)始心跳計(jì)時(shí)，如果在心跳計(jì)時(shí)結(jié)束之前沒(méi)有再次收到Client發(fā)來(lái)的信息，那么便會(huì)斷開(kāi)跟Client的連接。而一旦在設(shè)定時(shí)間內(nèi)再次收到Client發(fā)來(lái)的信息，那么Server便會(huì)重置計(jì)時(shí)器，再次重新進(jìn)行心跳計(jì)時(shí)，直到超時(shí)斷開(kāi)連接為止。
下面就是實(shí)現(xiàn)該計(jì)時(shí)的代碼：
復(fù)制代碼 代碼如下:
//長(zhǎng)連接入口
func handleConnection(conn net.Conn,timeout int) {
buffer := make([]byte, 2048)
for {
n, err := conn.Read(buffer)
if err != nil {
LogErr(conn.RemoteAddr().String(), " connection error: ", err)
return
}
Data :=(buffer[:n])
messnager := make(chan byte)
postda :=make(chan byte)
//心跳計(jì)時(shí)
go HeartBeating(conn,messnager,timeout)
//檢測(cè)每次Client是否有數(shù)據(jù)傳來(lái)
go GravelChannel(Data,messnager)
Log( "receive data length:",n)
Log(conn.RemoteAddr().String(), "receive data string:", string(Data
}
}
//心跳計(jì)時(shí)，根據(jù)GravelChannel判斷Client是否在設(shè)定時(shí)間內(nèi)發(fā)來(lái)信息
func HeartBeating(conn net.Conn, readerChannel chan byte,timeout int) {
select {
case fk := -readerChannel:
Log(conn.RemoteAddr().String(), "receive data string:", string(fk))
conn.SetDeadline(time.Now().Add(time.Duration(timeout) * time.Second))
//conn.SetReadDeadline(time.Now().Add(time.Duration(5) * time.Second))
break
case -time.After(time.Second*5):
Log("It's really weird to get Nothing!!!")
conn.Close()
}
}
func GravelChannel(n []byte,mess chan byte){
for _ , v := range n{
mess - v
}
close(mess)
}
func Log(v ...interface{}) {
log.Println(v...)
}
這樣，就可以成功實(shí)現(xiàn)對(duì)于長(zhǎng)連接的處理了~~，我們可以這么進(jìn)行測(cè)試：
func sender(conn net.Conn) {
for i := 0; i 5; i++ {
words:= strconv.Itoa(i)+"This is a test for long conn"
conn.Write([]byte(words))
time.Sleep(2*time.Second)
}
fmt.Println("send over")
}
可以發(fā)現(xiàn)，Sender函數(shù)中time.Sleep阻塞的時(shí)間設(shè)定的比Server中的timeout短的時(shí)候，Client端的信息可以自由的發(fā)送到循環(huán)結(jié)束，而當(dāng)我們?cè)O(shè)定Sender函數(shù)的阻塞時(shí)間較長(zhǎng)時(shí)，就只能發(fā)出第一次循環(huán)的信息。
將運(yùn)行參數(shù)放入配置文件（XML/YAML）
為了將我們寫(xiě)好的Server發(fā)布到服務(wù)器上，就要將我們的代碼進(jìn)行build打包，這樣如果以后想要修改一些代碼的話，需要重新給代碼進(jìn)行編譯打包并上傳到服務(wù)器上。
顯然，這么做過(guò)于繁瑣。。。因此常見(jiàn)的做法都是將Server運(yùn)行中可能會(huì)頻繁變更的變量、數(shù)值寫(xiě)入配置文件中，這樣直接讓程序從配置文件讀取參數(shù)，避免對(duì)代碼頻繁的操作。
關(guān)于配置文件的格式，在這里推薦YAML 和XML~ XML是傳統(tǒng)的配置文件寫(xiě)法，不過(guò)本人比較推薦yaml，他比XML要更加人性化，也更好寫(xiě)，關(guān)于yaml的詳細(xì)信息可以參考： yaml官網(wǎng)
比如我們可以將Server監(jiān)聽(tīng)的端口作為變量，寫(xiě)入配置文件 config.yaml 和 config.xml，放入代碼的根目錄下，這樣當(dāng)我們想要更換服務(wù)器端口的時(shí)候，只要在配置文件中修改port對(duì)應(yīng)的值就可以拉。 config.xml內(nèi)容如下：
?xml version="1.0" encoding="UTF-8"?>
Config1>GetConfig/Config1>
Config2>THE/Config2>
Config3>Information/Config3>
Feature1>HereIsTEST1/Feature1>
Feature2>1024/Feature2>
Feature3>Feature23333/Feature3>
config.yaml內(nèi)容如下：
Address: 172.168.0.1
Config1: Easy
Config2:
Feature1: 2
Feature2: [3, 4]
Port: :6060
Config4: IS
Config5: ATest
接下來(lái)就是解析他們了，目前golang官方還沒(méi)有解析yaml的庫(kù)，因此我推薦使用第三方的go-yaml包，
地址如下： go-yaml ，go get安裝該包后，我們就可以通過(guò)他解析文件啦：
復(fù)制代碼 代碼如下:
//解析文件，取出所有參數(shù)
func GetYamlConfig() map[interface{}]interface{}{
data, err := ioutil.ReadFile("config.yaml")
//將解析出的參數(shù)轉(zhuǎn)為map的形式
m := make(map[interface{}]interface{})
if err != nil {
LogErr("error: %v", err)
}
err = yaml.Unmarshal([]byte(data), m)
return m
}
//根據(jù)需求取出對(duì)應(yīng)值
func GetElement(key string,themap map[interface{}]interface{})string {
if value,ok:=themap[key];ok {
return value.(string)
}
LogErr("Can't find the *.yaml")
return ""
}
這里同樣給出解析xml配置文件的代碼：
復(fù)制代碼 代碼如下:
func GetXMLConfig() map[string]string {
var t xml.Token
var err error
Keylst := make([]string,6)
Valuelst:=make([]string,6)
//將解析出的元素填入map中，便于查找
map1:=make(map[string]string)
content, err := ioutil.ReadFile("config.xml")
CheckError(err)
decoder := xml.NewDecoder(bytes.NewBuffer(content))
i:=0
j:=0
for t, err = decoder.Token(); err == nil; t, err = decoder.Token() {
switch token := t.(type) {
// 處理元素
case xml.StartElement:
name := token.Name.Local
Keylst[i]=string(name)
i=i+1
case xml.CharData:
content1 := string([]byte(token))
//Valuelst=append(Valuelst,content1)
Valuelst[j]=content1
j=j+1
}
}
for count:=0;countlen(Keylst);count++{
map1[Keylst[count]]=Valuelst[count]
}
return map1
}
//取出map的函數(shù)跟yaml中的差不多，此處略過(guò)
運(yùn)行效果如下：
您可能感興趣的文章:- Go語(yǔ)言編譯程序從后臺(tái)運(yùn)行,不出現(xiàn)dos窗口的操作
- 解決運(yùn)行django程序出錯(cuò)問(wèn)題 ''str''object has no attribute''_meta''
- 在Apache服務(wù)器上同時(shí)運(yùn)行多個(gè)Django程序的方法
- 在Colaboratory上運(yùn)行Go程序的詳細(xì)過(guò)程
