go-packages/iris
1
0
Fork 0
mirror of https://github.com/kataras/iris.git synced 2025-03-21 23:06:27 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

websocket: expose Connection#Write in favor of https://github.com/kataras/iris/issues/976

Browse Source 
Former-commit-id: 74cba9b2bfb4d5cc874eba3330dc9c7bad2ffac5
This commit is contained in:
Gerasimos Maropoulos 2018-04-25 13:40:40 +03:00
parent 59f138963b
commit 5dc8cbcefc
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
websocket/connection.go
View File

@ -149,6 +149,10 @@ type (
// Its connection-relative operations are safe for use. // Its connection-relative operations are safe for use.
Server() *Server Server() *Server
// Write writes a raw websocket message with a specific type to the client
// used by ping messages and any CloseMessage types.
Write(websocketMessageType int, data []byte) error
// Context returns the (upgraded) context.Context of this connection // Context returns the (upgraded) context.Context of this connection
// avoid using it, you normally don't need it, // avoid using it, you normally don't need it,
// websocket has everything you need to authenticate the user BUT if it's necessary // websocket has everything you need to authenticate the user BUT if it's necessary
@ -240,6 +244,8 @@ type (
var _ Connection = &connection{} var _ Connection = &connection{}
const CloseMessage = websocket.CloseMessage
func newConnection(ctx context.Context, s *Server, underlineConn UnderlineConnection, id string) *connection { func newConnection(ctx context.Context, s *Server, underlineConn UnderlineConnection, id string) *connection {
c := &connection{ c := &connection{
underline: underlineConn, underline: underlineConn,
@ -271,9 +277,9 @@ func (c *connection) Err() error {
return c.err return c.err
} }
// write writes a raw websocket message with a specific type to the client // Write writes a raw websocket message with a specific type to the client
// used by ping messages and any CloseMessage types. // used by ping messages and any CloseMessage types.
func (c *connection) write(websocketMessageType int, data []byte) error { func (c *connection) Write(websocketMessageType int, data []byte) error {
// for any-case the app tries to write from different goroutines, // for any-case the app tries to write from different goroutines,
// we must protect them because they're reporting that as bug... // we must protect them because they're reporting that as bug...
c.writerMu.Lock() c.writerMu.Lock()
@ -295,7 +301,7 @@ func (c *connection) write(websocketMessageType int, data []byte) error {
// writeDefault is the same as write but the message type is the configured by c.messageType // writeDefault is the same as write but the message type is the configured by c.messageType
// if BinaryMessages is enabled then it's raw []byte as you expected to work with protobufs // if BinaryMessages is enabled then it's raw []byte as you expected to work with protobufs
func (c *connection) writeDefault(data []byte) { func (c *connection) writeDefault(data []byte) {
c.write(c.messageType, data) c.Write(c.messageType, data)
} }
const ( const (
@ -332,7 +338,7 @@ func (c *connection) startPinger() {
//fire all OnPing methods //fire all OnPing methods
c.fireOnPing() c.fireOnPing()
// try to ping the client, if failed then it disconnects // try to ping the client, if failed then it disconnects
err := c.write(websocket.PingMessage, []byte{}) err := c.Write(websocket.PingMessage, []byte{})
if err != nil { if err != nil {
// must stop to exit the loop and finish the go routine // must stop to exit the loop and finish the go routine
break break