从零到一：用Golang编写机器人
DustyFitzsi
8年前
<p><img src="https://simg.open-open.com/show/42b3f7b967060197cd5850c2c1939d99.jpg"></p> <p>从一个有趣的项目来入门新的语言，再合适不过了。</p> <p>本人也是通过编写一个可以聊天，可以设置备忘/定时提醒，可以搜索美剧/电影资源等等功能的小机器人，迅速掌握了Go这门语言并喜欢上它。</p> <h2><strong>1. 先定个小目标——从对话开始</strong></h2> <p>让机器人“开口说话”是首要的，这里先采用第三方服务提供的API，本地通过HTTP请求获取回答并返回。</p> <p style="text-align:center"><img src="https://simg.open-open.com/show/1f35110feb30140ff9f11186b991a17d.png"></p> <p>此阶段基本没有难点，以Go为例，关键部分数十行代码解决：</p> <pre> <code class="language-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"` }</code></pre> <h2><strong>2. 独乐不如众乐——分享给好友</strong></h2> <p>通过第一步，机器人已经具备了基础的对话功能，此时可以开始拉上好友一起调戏了。</p> <p>虽然Go语言可以编译出多个平台的可执行文件用来分享（包括但不限于Linux、Windows、Mac OS），但我们有许多更方便更优雅的方式。</p> <h3><strong>2.1 微信公众号</strong></h3> <p>通过微信开发者平台注册一个公众号，加上上述的一点点代码便可让其具备对话功能：</p> <p style="text-align:center"><img src="https://simg.open-open.com/show/67a8f203e8d46733e938127954f7009a.jpg"></p> <h3>2.2 网页分享</h3> <p>通过一些简单的前端技术，可以让对话更人性化：</p> <p style="text-align:center"><img src="https://simg.open-open.com/show/8c40cb0ee001a55bb6f84c978113f746.gif"></p> <p>此处前后端采用 WebSocket 来通信：</p> <pre> <code class="language-go">//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("")) } }</code></pre> <p>通过WebSocket的推送功能，让对话返回效果(分段、限速)变得可控。</p> <h2><strong>3. 增加技能点</strong></h2> <p>仅有对话功能，只能称为聊天机器人，所以开始添加新的功能。</p> <p>以简单的添加备忘为例：</p> <p style="text-align:center"><img src="https://simg.open-open.com/show/0805e00f83334ffc2c70c1f283dd8999.png"></p> <pre> <code class="language-go">// 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 }</code></pre> <p>交互模式下，先判断用户交互状态来给出不同回复。</p> <p>使用 go conn.CreateMemo(user, when, memo) 异步生成 <a href="/misc/goto?guid=4958185538616997143" rel="nofollow,noindex">Redis</a> 记录。：</p> <pre> <code class="language-go">// 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) }</code></pre> <h2><strong>4. 最后</strong></h2> <p>至此，一个机器人的编写就结束了，从零到一创造出一个小机器人固然是有趣的，而真正有趣的是从一到N的过程，脑洞有多大，小机器人的能力就有多大。</p> <p> </p> <p> </p> <p>来自：http://www.jianshu.com/p/46c7d36a95d2</p> <p> </p>