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)
	}
}