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.
103 lines
3.0 KiB
103 lines
3.0 KiB
package msggateway
|
|
|
|
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
|
|
//
|
|
GenerateLongConn(w http.ResponseWriter, r *http.Request) error
|
|
}
|
|
type GWebSocket struct {
|
|
protocolType int
|
|
conn *websocket.Conn
|
|
handshakeTimeout time.Duration
|
|
readBufferSize, WriteBufferSize int
|
|
}
|
|
|
|
func newGWebSocket(protocolType int, handshakeTimeout time.Duration, readBufferSize int) *GWebSocket {
|
|
return &GWebSocket{protocolType: protocolType, handshakeTimeout: handshakeTimeout, readBufferSize: readBufferSize}
|
|
}
|
|
|
|
func (d *GWebSocket) Close() error {
|
|
return d.conn.Close()
|
|
}
|
|
func (d *GWebSocket) GenerateLongConn(w http.ResponseWriter, r *http.Request) error {
|
|
upgrader := &websocket.Upgrader{
|
|
HandshakeTimeout: d.handshakeTimeout,
|
|
ReadBufferSize: d.readBufferSize,
|
|
CheckOrigin: func(r *http.Request) bool { return true },
|
|
}
|
|
conn, err := upgrader.Upgrade(w, r, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
d.conn = conn
|
|
return nil
|
|
|
|
}
|
|
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
|
|
//}
|