從一個(gè)有趣的項(xiàng)目來(lái)入門(mén)新的語(yǔ)言,再合適不過(guò)了雪隧。
本人也是通過(guò)編寫(xiě)一個(gè)可以聊天这溅,可以設(shè)置備忘/定時(shí)提醒,可以搜索美劇/電影資源等等功能的小機(jī)器人,迅速掌握了Go這門(mén)語(yǔ)言并喜歡上它讯柔。
文末會(huì)給出小機(jī)器人源碼及文中實(shí)例代碼鏈接抡蛙。
1. 先定個(gè)小目標(biāo)——從對(duì)話(huà)開(kāi)始
讓機(jī)器人“開(kāi)口說(shuō)話(huà)”是首要的,這里先采用第三方服務(wù)提供的API魂迄,本地通過(guò)HTTP請(qǐng)求獲取回答并返回粗截。
此階段基本沒(méi)有難點(diǎn),以Go為例捣炬,關(guān)鍵部分?jǐn)?shù)十行代碼解決:
//get reply from tlAI
func tlAI(info string) string {
tuLingURL := fmt.Sprintf("http://www.tuling123.com/openapi/api?key=%s&info=%s", tlKey, url.QueryEscape(info))
resp, err := http.Get(tuLingURL)
if err != nil {
log.Println(err)
return ""
}
defer resp.Body.Close()
reply := new(tlReply)
decoder := json.NewDecoder(resp.Body) //decode reply from response body
decoder.Decode(reply)
return reply.Text
}
type tlReply struct {
code int
Text string `json:"text"`
}
2. 獨(dú)樂(lè)不如眾樂(lè)——分享給好友
通過(guò)第一步熊昌,機(jī)器人已經(jīng)具備了基礎(chǔ)的對(duì)話(huà)功能,此時(shí)可以開(kāi)始拉上好友一起調(diào)戲了遥金。
雖然Go語(yǔ)言可以編譯出多個(gè)平臺(tái)的可執(zhí)行文件用來(lái)分享(包括但不限于Linux浴捆、Windows、Mac OS)稿械,但我們有許多更方便更優(yōu)雅的方式选泻。
2.1 微信公眾號(hào)
通過(guò)微信開(kāi)發(fā)者平臺(tái)注冊(cè)一個(gè)公眾號(hào),加上上述的一點(diǎn)點(diǎn)代碼便可讓其具備對(duì)話(huà)功能:
2.2 網(wǎng)頁(yè)分享
通過(guò)一些簡(jiǎn)單的前端技術(shù)美莫,可以讓對(duì)話(huà)更人性化:
在線演示地址 (瀏覽器給予權(quán)限的話(huà)可支持語(yǔ)音輸入)
此處前后端采用WebSocket來(lái)通信:
//used by web samaritan robot
func socketHandler(w http.ResponseWriter, r *http.Request) {
c, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Print("upgrade:", err)
return
}
defer c.Close()
for {
mt, in, err := c.ReadMessage() // read from user input
if err != nil {
log.Println("read:", err)
break
}
ret := tlAI(string(in)) // get reply from tl AI robot
for i := range ret {
c.WriteMessage(mt, []byte(ret[i]))
time.Sleep(time.Second)
}
c.WriteMessage(mt, []byte(""))
}
}
通過(guò)WebSocket的推送功能页眯,讓對(duì)話(huà)返回效果(分段、限速)變得可控厢呵。
3. 增加技能點(diǎn)
僅有對(duì)話(huà)功能窝撵,只能稱(chēng)為聊天機(jī)器人,所以開(kāi)始添加新的功能襟铭。
以簡(jiǎn)單的添加備忘為例:
// SaveMemo create a memo for user, saved in redis
// command '/memo'
func (rb *Robot) SaveMemo(update tgbotapi.Update, step int) (ret string) {
user := update.Message.Chat.UserName
tmpAction := userAction[user]
switch step {
case 0:
tmpAction.ActionStep++
userAction[user] = tmpAction
ret = "Ok, what do you want to save?"
case 1:
defer delete(userAction, user)
when := time.Now().Format("2006-1-02 15:04")
memo := update.Message.Text
go conn.CreateMemo(user, when, memo)
ret = "Ok, type '/memos' to see all your memos"
}
return
}
交互模式下碌奉,先判斷用戶(hù)交互狀態(tài)來(lái)給出不同回復(fù)。
使用go conn.CreateMemo(user, when, memo)
異步生成Redis記錄寒砖。:
// CreateMemo saves a memo
func CreateMemo(user, when, memo string) {
c := Pool.Get()
defer c.Close()
var setMemoLua = `
local id = redis.call("INCR", "memoIncrId")
redis.call("RPUSH", KEYS[1]..":memos", id)
redis.call("HMSET", "memo:"..id, "time", KEYS[2], "content", KEYS[3])
`
script := redis.NewScript(3, setMemoLua)
script.Do(c, user, when, memo)
}
4. 最后
至此赐劣,一個(gè)機(jī)器人的編寫(xiě)就結(jié)束了,從零到一創(chuàng)造出一個(gè)小機(jī)器人固然是有趣的哩都,而真正有趣的是從一到N的過(guò)程魁兼,腦洞有多大,小機(jī)器人的能力就有多大漠嵌。
文中代碼以及自己寫(xiě)的小機(jī)器人代碼鏈接: github/evolsnow/robot
如果有其他有趣的點(diǎn)子咐汞,歡迎一起來(lái)開(kāi)發(fā)玩耍。
補(bǔ)了一篇講解機(jī)器人自動(dòng)找資源的文章儒鹿,拋磚引玉之用~:一步一步教你的機(jī)器人尋找資源鏈接