從一個有趣的項目來入門新的語言,再合適不過了。
本人也是通過編寫一個可以聊天,可以設置備忘/定時提醒,可以搜索美劇/電影資源等等功能的小機器人,迅速掌握了Go這門語言并喜歡上它。
文末會給出小機器人源碼及文中實例代碼鏈接。
1. 先定個小目標——從對話開始
讓機器人“開口說話”是首要的,這里先采用第三方服務提供的API,本地通過HTTP請求獲取回答并返回。
此階段基本沒有難點,以Go為例,關鍵部分數十行代碼解決:
//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. 獨樂不如眾樂——分享給好友
通過第一步,機器人已經具備了基礎的對話功能,此時可以開始拉上好友一起調戲了。
雖然Go語言可以編譯出多個平臺的可執行文件用來分享(包括但不限于Linux、Windows、Mac OS),但我們有許多更方便更優雅的方式。
2.1 微信公眾號
通過微信開發者平臺注冊一個公眾號,加上上述的一點點代碼便可讓其具備對話功能:
2.2 網頁分享
通過一些簡單的前端技術,可以讓對話更人性化:
在線演示地址 (瀏覽器給予權限的話可支持語音輸入)
此處前后端采用WebSocket來通信:
//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(""))
}
}
通過WebSocket的推送功能,讓對話返回效果(分段、限速)變得可控。
3. 增加技能點
僅有對話功能,只能稱為聊天機器人,所以開始添加新的功能。
以簡單的添加備忘為例:
// 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
}
交互模式下,先判斷用戶交互狀態來給出不同回復。
使用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. 最后
至此,一個機器人的編寫就結束了,從零到一創造出一個小機器人固然是有趣的,而真正有趣的是從一到N的過程,腦洞有多大,小機器人的能力就有多大。
文中代碼以及自己寫的小機器人代碼鏈接: github/evolsnow/robot
如果有其他有趣的點子,歡迎一起來開發玩耍。
補了一篇講解機器人自動找資源的文章,拋磚引玉之用~:一步一步教你的機器人尋找資源鏈接