Exemple #1
0
func ClientConnection(clientsock *websocket.Conn) {
	packet := pnet.NewPacket()
	buffer := make([]uint8, pnet.PACKET_MAXSIZE)
	recv, err := clientsock.Read(buffer)
	if err == nil {
		copy(packet.Buffer[0:recv], buffer[0:recv])
		packet.GetHeader()
		g_server.ParseFirstMessage(clientsock, packet)
	} else {
		if err.Error() != "EOF" {
			logger.Println("Client connection error: " + err.Error())
		}
	}
}
Exemple #2
0
func echoHandler(ws *websocket.Conn) {
	buff := make([]byte, 150)
	fmt.Println("connected to:", ws.Config().Location.Host)
	for {
		_, err := ws.Read(buff)
		if err != nil {
			break
		}
		fmt.Println("got:", string(buff))
		_, err = ws.Write(buff)
		if err != nil {
			break
		}
	}
	fmt.Println("lost connection with:", ws.Config().Location.Host)
	availableNodes <- reverseHostMap[ws.Config().Location.Host]
}
Exemple #3
0
func ChatroomServer(ws *websocket.Conn) {
	defer ws.Close()
	auth := false
	var user *User
	ch := NewChatHome()
	// user.DoNothing()

	item_ws := WSConns.PushBack(ws)
	defer WSConns.Remove(item_ws)

	// SendMessage(nil, fmt.Sprintf("welcome %s join\n", "name"))

	r := bufio.NewReader(ws)
	log.Println("Connected ")
	for {
		data, err := r.ReadBytes('\n')
		if err != nil {
			SendMessage(item_ws, `{"Code":"msg","Data":"`+user.Name+` offline"}`)
			if b, _ := ch.IsOnline(user.Name); b {
				ch.RmOnlineUser(user.Name)
				ch.WaittingToOnline()
				SendMessage(item_ws, `{"Code":"OK","Data":"Welcome `+user.Name+`"}`)
			} else {
				ch.RmWaittingUser(user.Name)
			}
			SendMessage(nil, `{"Code":"online_user_count","Data": `+strconv.Itoa(ch.GetOnlineCount())+`}`)
			SendMessage(nil, `{"Code":"waitting_user_count","Data": `+strconv.Itoa(ch.GetWaittingCount())+`}`)
			// SendMessage(nil, `{"Code":"online_user_list","Data": "`+ch.OnlineListToBase64()+`"}`)
			// SendMessage(nil, `{"Code":"waitting_user_list","Data": "`+ch.WaittingListToBase64()+`"}`)
			break
		}
		log.Println("Received: " + string(data))
		log.Println(`{"Code":"online_user_list","Data": "` + ch.OnlineListToBase64() + `"}`)
		var msg Msg
		err = json.Unmarshal(data, &msg)
		if err != nil {
			SendMessage(item_ws, `{"Code":"error","Data":"JSON Error","Desc":"Nothing"}`)
		}
		if msg.Code == "cookie" {
			for item_user := ConnUsers.Front(); item_user != nil; item_user = item_user.Next() {
				user, _ = item_user.Value.(*User)
				if user.Cookie == msg.Data {
					auth = true
					if ch.GetOnlineCount() < ch.MAX_ONLINE_COUNT && ch.GetWaittingCount() == 0 {
						ch.AddOnlineUser(user.Name)
						SendMessage(item_ws, `{"Code":"OK","Data":"Welcome `+user.Name+`"}`)
					} else {
						ch.AddWaittingUser(user.Name)
						SendMessage(nil, `{"Code":"wait","Data": `+strconv.Itoa(ch.GetWaittingCount())+`}`)
					}
					SendMessage(nil, `{"Code":"online_user_count","Data": `+strconv.Itoa(ch.GetOnlineCount())+`}`)
					SendMessage(nil, `{"Code":"waitting_user_count","Data": `+strconv.Itoa(ch.GetWaittingCount())+`}`)
					// SendMessage(nil, `{"Code":"online_user_list","Data": "`+ch.OnlineListToBase64()+`"}`)
					// SendMessage(nil, `{"Code":"waitting_user_list","Data": "`+ch.WaittingListToBase64()+`"}`)

					break
				}
			}
		} else {
			if !auth {
				SendMessage(item_ws, `{"Code":"error","Data":"Auth Error","Desc":"RELOGIN"}`)
			} else {
				// SendMessage(item_ws, `{"Code":"OK","Data":"Auth Error","Desc":"RELOGIN"}`)
				ControlMsg(item_ws, msg, user)
			}
		}
	}
	log.Println("Listenning Over")
}