49 lines
897 B
Go
49 lines
897 B
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
func main() {
|
|
url := "ws://localhost:8080/ws"
|
|
log.Printf("来都来了")
|
|
conn, _, err := websocket.DefaultDialer.Dial(url, nil)
|
|
if err != nil {
|
|
log.Fatal("哟，今天不营业呢", err)
|
|
}
|
|
defer conn.Close()
|
|
log.Println("好好好，进去了")
|
|
|
|
go func() {
|
|
for {
|
|
_, message, err := conn.ReadMessage()
|
|
if err != nil {
|
|
log.Println("润了润了")
|
|
break
|
|
}
|
|
log.Printf("掌柜说的是 %s", message)
|
|
}
|
|
}()
|
|
|
|
scanner := bufio.NewScanner(os.Stdin)
|
|
fmt.Print("说点什么：")
|
|
|
|
for scanner.Scan() {
|
|
message := scanner.Text()
|
|
err := conn.WriteMessage(websocket.TextMessage, []byte(message))
|
|
if err != nil {
|
|
log.Println("话没送出去", err)
|
|
break
|
|
}
|
|
fmt.Print("说点什么：")
|
|
}
|
|
|
|
if err := scanner.Err(); err != nil {
|
|
log.Println("输入错误:", err)
|
|
}
|
|
}
|