|
@@ -2,14 +2,24 @@ 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.3:12090/bus/v1/homePage"
|
|
|
d := websocket.Dialer{}
|
|
|
conn, _, err := d.Dial(wsURL, nil)
|
|
@@ -17,15 +27,15 @@ func main() {
|
|
|
panic(err.Error())
|
|
|
}
|
|
|
go clientReadMessage(conn)
|
|
|
- writeToken(conn)
|
|
|
+ writeToken(conn, *token)
|
|
|
|
|
|
go clientWriteMessage(conn)
|
|
|
|
|
|
select {}
|
|
|
}
|
|
|
|
|
|
-func writeToken(conn *websocket.Conn) {
|
|
|
- token := "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJVVUlEIjoiYWM3NzgyMDMtNTg3ZC00ZjBjLWEzZDktYzgwNDA1NTBjNDllIiwiVXNlcklEIjoyLCJVc2VybmFtZSI6Inl1bmxvbmciLCJOaWNrTmFtZSI6IuadjuS6kem-mSIsIlJvbGVJRHMiOlszXSwiRGVwdElEIjoxLCJSb2xlTmFtZXMiOm51bGwsImlzcyI6IkpXVCIsImV4cCI6MTY3MzQxNjA1MSwibmJmIjoxNjczMzI5NjUxfQ.QxEgaPcTBmP8JTn0d7CVRs0aXl1C3M3Dynh4KFqMWh8"
|
|
|
+func writeToken(conn *websocket.Conn, token string) {
|
|
|
+
|
|
|
homeMsg := HomeMsg{
|
|
|
MsgType: 11,
|
|
|
Data: token,
|