123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899 |
- package main
- import (
- "encoding/json"
- "flag"
- "fmt"
- "github.com/gorilla/websocket"
- "github.com/sirupsen/logrus"
- "time"
- )
- var token = flag.String("token", "", "user token")
- func main() {
- flag.Parse()
- fmt.Println(token)
- if *token == "" {
- fmt.Println("请输入token")
- return
- }
- wsURL := "ws://10.0.0.230:12090/bus/v1/homePage"
- d := websocket.Dialer{}
- conn, _, err := d.Dial(wsURL, nil)
- if err != nil {
- panic(err.Error())
- }
- go clientReadMessage(conn)
- writeToken(conn, *token)
- go clientWriteMessage(conn)
- select {}
- }
- func writeToken(conn *websocket.Conn, token string) {
- homeMsg := HomeMsg{
- MsgType: 11,
- Data: token,
- }
- homeBytes, err := json.Marshal(homeMsg)
- if err != nil {
- fmt.Println("marshal err")
- return
- }
- err = conn.WriteMessage(websocket.TextMessage, homeBytes)
- if err != nil {
- fmt.Printf("Write Message error:%s\n", err)
- return
- }
- fmt.Println("Write token success")
- }
- func clientReadMessage(conn *websocket.Conn) {
- for {
- msgType, msg, err := conn.ReadMessage()
- if err != nil {
- if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway, websocket.CloseAbnormalClosure) {
- logrus.Warnf("readMessage err:%s", err)
- } else {
- logrus.Info("readMessage conn closed")
- }
- break
- }
- logrus.WithFields(logrus.Fields{"msgType": msgType, "data": string(msg), "err": err}).Info("read success")
- }
- }
- type HomeMsg struct {
- DeptID int `json:"deptId"`
- MsgType uint8 `json:"msgType"`
- Data string `json:"data"`
- }
- func clientWriteMessage(conn *websocket.Conn) {
- ticker := time.NewTicker(5 * time.Second)
- for {
- select {
- case <-ticker.C:
- homeMsg := HomeMsg{
- MsgType: 1,
- Data: "",
- }
- homeBytes, err := json.Marshal(homeMsg)
- if err != nil {
- fmt.Println("marshal err")
- return
- }
- err = conn.WriteMessage(websocket.TextMessage, homeBytes)
- if err != nil {
- fmt.Printf("Write Message error:%s\n", err)
- return
- }
- fmt.Println("Write Message success")
- }
- }
- }
|