我正在构建一个基于命令行的 Go 机器人,它与 Instagram API 交互。
Instagram API 基于 OAuth,因此不太适合基于命令行的应用程序。
为了解决这个问题,我在浏览器中打开适当的授权 URL,并使用我为重定向 URI 启动的本地服务器 - 这样我就可以捕获并优雅地显示访问令牌,而不是用户需要从手动输入网址。
到目前为止一切顺利,应用程序可以成功打开浏览器访问授权 URL,您对其进行授权,它会将您重定向到本地 HTTP 服务器。
现在,在向用户显示访问令牌后,我不需要 HTTP 服务器,因此我想在执行此操作后手动关闭服务器。
为此,我从中汲取了灵感answer并鼓吹以下内容:
package main
import (
"fmt"
"io"
"log"
"net/http"
"os/exec"
"runtime"
"time"
)
var client_id = "my_client_id"
var client_secret = "my_client_secret"
var redirect_url = "http://localhost:8000/instagram/callback"
func main() {
srv := startHttpServer()
openbrowser(fmt.Sprintf("https://api.instagram.com/oauth/authorize/?client_id=%v&redirect_uri=%v&response_type=code", client_id, redirect_url))
// Backup to gracefully shutdown the server
time.Sleep(20 * time.Second)
if err := srv.Shutdown(nil); err != nil {
panic(err) // failure/timeout shutting down the server gracefully
}
}
func showTokenToUser(w http.ResponseWriter, r *http.Request, srv *http.Server) {
io.WriteString(w, fmt.Sprintf("Your access token is: %v", r.URL.Query().Get("code")))
if err := srv.Shutdown(nil); err != nil {
log.Fatal(err) // failure/timeout shutting down the server gracefully
}
}
func startHttpServer() *http.Server {
srv := &http.Server{Addr: ":8000"}
http.HandleFunc("/instagram/callback", func(w http.ResponseWriter, r *http.Request) {
showTokenToUser(w, r, srv)
})
go func() {
if err := srv.ListenAndServe(); err != nil {
// cannot panic, because this probably is an intentional close
log.Printf("Httpserver: ListenAndServe() error: %s", err)
}
}()
// returning reference so caller can call Shutdown()
return srv
}
func openbrowser(url string) {
var err error
switch runtime.GOOS {
case "linux":
err = exec.Command("xdg-open", url).Start()
case "windows":
err = exec.Command("rundll32", "url.dll,FileProtocolHandler", url).Start()
case "darwin":
err = exec.Command("open", url).Start()
default:
err = fmt.Errorf("unsupported platform")
}
if err != nil {
log.Fatal(err)
}
}
但是,上述情况会导致此错误:
2017/11/23 16:02:03 Httpserver: ListenAndServe() 错误:http:服务器关闭
2017/11/23 16:02:03 http: 恐慌服务 [::1]:61793: 运行时错误: 无效的内存地址或 nil 指针取消引用
如果我在处理程序中注释掉这些行,那么它可以完美地工作,尽管当我点击回调路由时不会关闭服务器:
if err := srv.Shutdown(nil); err != nil {
log.Fatal(err) // failure/timeout shutting down the server gracefully
}
我哪里出错了?我需要更改什么,以便在向用户显示文本后,当我点击回调路由时可以关闭服务器。