package main

import (
	"flag"
	"github.com/gorilla/websocket"
	client2 "go-websocket/client"
	"log"
	"net/http"
	"time"
)

func main() {
	var addr = flag.String("addr", ":8088", "http service address")
	http.HandleFunc("/", echo)
	log.Fatal(http.ListenAndServe(*addr, nil))
}
func echo(w http.ResponseWriter, r *http.Request) {
	key := r.Header["Sec-Websocket-Protocol"][0]
	var upGrader = websocket.Upgrader{
		CheckOrigin: func(r *http.Request) bool {
			return true
		},
		Subprotocols: []string{key}, //chrome处理
	}
	conn, err := upGrader.Upgrade(w, r, nil)
	//放至客户端
	var client = client2.Client{
		Key:           key,
		Conn:          conn,
		HeartbeatTime: time.Now().Unix(),
	}
	if err != nil {
		log.Println(err)
		return
	}

	//注册到用户连接管理
	client2.Start(&client)
}