75 lines
1.6 KiB
Go
75 lines
1.6 KiB
Go
|
package engineio
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/googollee/go-socket.io/engineio/session"
|
||
|
|
"net/http"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/googollee/go-socket.io/engineio/transport"
|
||
|
|
"github.com/googollee/go-socket.io/engineio/transport/polling"
|
||
|
|
"github.com/googollee/go-socket.io/engineio/transport/websocket"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Options is options to create a server.
|
||
|
|
type Options struct {
|
||
|
|
PingTimeout time.Duration
|
||
|
|
PingInterval time.Duration
|
||
|
|
|
||
|
|
Transports []transport.Transport
|
||
|
|
SessionIDGenerator session.IDGenerator
|
||
|
|
|
||
|
|
RequestChecker CheckerFunc
|
||
|
|
ConnInitor ConnInitorFunc
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Options) getRequestChecker() CheckerFunc {
|
||
|
|
if c != nil && c.RequestChecker != nil {
|
||
|
|
return c.RequestChecker
|
||
|
|
}
|
||
|
|
return defaultChecker
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Options) getConnInitor() ConnInitorFunc {
|
||
|
|
if c != nil && c.ConnInitor != nil {
|
||
|
|
return c.ConnInitor
|
||
|
|
}
|
||
|
|
return defaultInitor
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Options) getPingTimeout() time.Duration {
|
||
|
|
if c != nil && c.PingTimeout != 0 {
|
||
|
|
return c.PingTimeout
|
||
|
|
}
|
||
|
|
return time.Minute
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Options) getPingInterval() time.Duration {
|
||
|
|
if c != nil && c.PingInterval != 0 {
|
||
|
|
return c.PingInterval
|
||
|
|
}
|
||
|
|
return time.Second * 20
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Options) getTransport() []transport.Transport {
|
||
|
|
if c != nil && len(c.Transports) != 0 {
|
||
|
|
return c.Transports
|
||
|
|
}
|
||
|
|
return []transport.Transport{
|
||
|
|
polling.Default,
|
||
|
|
websocket.Default,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Options) getSessionIDGenerator() session.IDGenerator {
|
||
|
|
if c != nil && c.SessionIDGenerator != nil {
|
||
|
|
return c.SessionIDGenerator
|
||
|
|
}
|
||
|
|
return &session.DefaultIDGenerator{}
|
||
|
|
}
|
||
|
|
|
||
|
|
func defaultChecker(*http.Request) (http.Header, error) {
|
||
|
|
return nil, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func defaultInitor(*http.Request, Conn) {}
|