You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
84 lines
2.3 KiB
84 lines
2.3 KiB
package new
|
|
|
|
import (
|
|
"github.com/gorilla/websocket"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
type LongConn interface {
|
|
//Close this connection
|
|
Close() error
|
|
//Write message to connection,messageType means data type,can be set binary(2) and text(1).
|
|
WriteMessage(messageType int, message []byte) error
|
|
//Read message from connection.
|
|
ReadMessage() (int, []byte, error)
|
|
//SetReadTimeout sets the read deadline on the underlying network connection,
|
|
//after a read has timed out, will return an error.
|
|
SetReadTimeout(timeout int) error
|
|
//SetWriteTimeout sets the write deadline when send message,when read has timed out,will return error.
|
|
SetWriteTimeout(timeout int) error
|
|
//Try to dial a connection,url must set auth args,header can control compress data
|
|
Dial(urlStr string, requestHeader http.Header) (*http.Response, error)
|
|
//Whether the connection of the current long connection is nil
|
|
IsNil() bool
|
|
//Set the connection of the current long connection to nil
|
|
SetConnNil()
|
|
//Check the connection of the current and when it was sent are the same
|
|
CheckSendConnDiffNow() bool
|
|
}
|
|
type GWebSocket struct {
|
|
protocolType int
|
|
conn *websocket.Conn
|
|
}
|
|
|
|
func NewDefault(protocolType int) *GWebSocket {
|
|
return &GWebSocket{protocolType: protocolType}
|
|
}
|
|
func (d *GWebSocket) Close() error {
|
|
return d.conn.Close()
|
|
}
|
|
|
|
func (d *GWebSocket) WriteMessage(messageType int, message []byte) error {
|
|
d.setSendConn(d.conn)
|
|
return d.conn.WriteMessage(messageType, message)
|
|
}
|
|
|
|
func (d *GWebSocket) setSendConn(sendConn *websocket.Conn) {
|
|
d.sendConn = sendConn
|
|
}
|
|
|
|
func (d *GWebSocket) ReadMessage() (int, []byte, error) {
|
|
return d.conn.ReadMessage()
|
|
}
|
|
func (d *GWebSocket) SetReadTimeout(timeout int) error {
|
|
return d.conn.SetReadDeadline(time.Now().Add(time.Duration(timeout) * time.Second))
|
|
}
|
|
|
|
func (d *GWebSocket) SetWriteTimeout(timeout int) error {
|
|
return d.conn.SetWriteDeadline(time.Now().Add(time.Duration(timeout) * time.Second))
|
|
}
|
|
|
|
func (d *GWebSocket) Dial(urlStr string, requestHeader http.Header) (*http.Response, error) {
|
|
conn, httpResp, err := websocket.DefaultDialer.Dial(urlStr, requestHeader)
|
|
if err == nil {
|
|
d.conn = conn
|
|
}
|
|
return httpResp, err
|
|
|
|
}
|
|
|
|
func (d *GWebSocket) IsNil() bool {
|
|
if d.conn != nil {
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
func (d *GWebSocket) SetConnNil() {
|
|
d.conn = nil
|
|
}
|
|
func (d *GWebSocket) CheckSendConnDiffNow() bool {
|
|
return d.conn == d.sendConn
|
|
}
|