123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869 |
- package main
- import (
- "flag"
- "fmt"
- "log"
- "net/url"
- "os"
- "os/signal"
- "time"
- "github.com/gorilla/websocket"
- )
- func main() {
- addr := flag.String("addr", "localhost:9080", "http service address")
- flag.Parse()
- fmt.Println("vim-go")
- u := url.URL{Scheme: "ws", Host: *addr, Path: "/ws"}
- log.Printf("connecting to %s", u.String())
- interrupt := make(chan os.Signal, 1)
- signal.Notify(interrupt, os.Interrupt)
- c, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
- if err != nil {
- log.Fatal("dial:", err)
- }
- defer c.Close()
- done := make(chan struct{})
- go func() {
- defer close(done)
- for {
- _, message, err := c.ReadMessage()
- if err != nil {
- log.Println("close:", err)
- return
- }
- fmt.Println(string(message))
- }
- }()
- for {
- select {
- case <-done:
- return
- case <-interrupt:
- log.Println("interrupt")
- // Cleanly close the connection by sending a close message and then waiting (with timeout) for the
- // server to close the connection.
- err := c.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""))
- if err != nil {
- log.Println("write close:", err)
- return
- }
- select {
- case <-done:
- log.Println("Successfully close")
- case <-time.After(1 * time.Second):
- log.Println("Forced close")
- }
- return
- }
- }
- }
|