2014-04-19 01:25:11 +04:00
|
|
|
// Copyright 2013 The Gorilla WebSocket Authors. All rights reserved.
|
2013-10-16 20:41:47 +04:00
|
|
|
// Use of this source code is governed by a BSD-style
|
|
|
|
// license that can be found in the LICENSE file.
|
|
|
|
|
|
|
|
// Package websocket implements the WebSocket protocol defined in RFC 6455.
|
|
|
|
//
|
|
|
|
// Overview
|
|
|
|
//
|
2014-04-18 02:40:10 +04:00
|
|
|
// The Conn type represents a WebSocket connection. A server application uses
|
|
|
|
// the Upgrade function from an Upgrader object with a HTTP request handler
|
|
|
|
// to get a pointer to a Conn:
|
|
|
|
//
|
|
|
|
// var upgrader = websocket.Upgrader{
|
|
|
|
// ReadBufferSize: 1024,
|
|
|
|
// WriteBufferSize: 1024,
|
|
|
|
// }
|
2013-10-16 20:41:47 +04:00
|
|
|
//
|
|
|
|
// func handler(w http.ResponseWriter, r *http.Request) {
|
2014-04-18 02:40:10 +04:00
|
|
|
// conn, err := upgrader.Upgrade(w, r, nil)
|
2014-04-20 22:38:35 +04:00
|
|
|
// if err != nil {
|
2013-12-21 03:57:02 +04:00
|
|
|
// log.Println(err)
|
|
|
|
// return
|
|
|
|
// }
|
2013-10-16 20:41:47 +04:00
|
|
|
// ... Use conn to send and receive messages.
|
|
|
|
// }
|
|
|
|
//
|
2015-05-30 02:26:01 +03:00
|
|
|
// Call the connection's WriteMessage and ReadMessage methods to send and
|
2013-10-30 04:43:03 +04:00
|
|
|
// receive messages as a slice of bytes. This snippet of code shows how to echo
|
|
|
|
// messages using these methods:
|
2013-10-16 20:41:47 +04:00
|
|
|
//
|
|
|
|
// for {
|
2013-10-30 04:43:03 +04:00
|
|
|
// messageType, p, err := conn.ReadMessage()
|
2013-10-16 20:41:47 +04:00
|
|
|
// if err != nil {
|
|
|
|
// return
|
|
|
|
// }
|
2013-12-31 15:00:58 +04:00
|
|
|
// if err = conn.WriteMessage(messageType, p); err != nil {
|
2013-10-16 20:41:47 +04:00
|
|
|
// return err
|
|
|
|
// }
|
|
|
|
// }
|
|
|
|
//
|
2013-10-30 04:43:03 +04:00
|
|
|
// In above snippet of code, p is a []byte and messageType is an int with value
|
|
|
|
// websocket.BinaryMessage or websocket.TextMessage.
|
|
|
|
//
|
|
|
|
// An application can also send and receive messages using the io.WriteCloser
|
|
|
|
// and io.Reader interfaces. To send a message, call the connection NextWriter
|
|
|
|
// method to get an io.WriteCloser, write the message to the writer and close
|
|
|
|
// the writer when done. To receive a message, call the connection NextReader
|
|
|
|
// method to get an io.Reader and read until io.EOF is returned. This snippet
|
2016-02-11 03:22:44 +03:00
|
|
|
// shows how to echo messages using the NextWriter and NextReader methods:
|
2013-10-16 20:41:47 +04:00
|
|
|
//
|
|
|
|
// for {
|
2013-10-30 04:43:03 +04:00
|
|
|
// messageType, r, err := conn.NextReader()
|
2013-10-16 20:41:47 +04:00
|
|
|
// if err != nil {
|
|
|
|
// return
|
|
|
|
// }
|
2013-10-30 04:43:03 +04:00
|
|
|
// w, err := conn.NextWriter(messageType)
|
|
|
|
// if err != nil {
|
|
|
|
// return err
|
|
|
|
// }
|
|
|
|
// if _, err := io.Copy(w, r); err != nil {
|
|
|
|
// return err
|
|
|
|
// }
|
|
|
|
// if err := w.Close(); err != nil {
|
2013-10-16 20:41:47 +04:00
|
|
|
// return err
|
|
|
|
// }
|
|
|
|
// }
|
|
|
|
//
|
|
|
|
// Data Messages
|
|
|
|
//
|
|
|
|
// The WebSocket protocol distinguishes between text and binary data messages.
|
|
|
|
// Text messages are interpreted as UTF-8 encoded text. The interpretation of
|
|
|
|
// binary messages is left to the application.
|
|
|
|
//
|
2013-10-30 04:43:03 +04:00
|
|
|
// This package uses the TextMessage and BinaryMessage integer constants to
|
|
|
|
// identify the two data message types. The ReadMessage and NextReader methods
|
|
|
|
// return the type of the received message. The messageType argument to the
|
|
|
|
// WriteMessage and NextWriter methods specifies the type of a sent message.
|
|
|
|
//
|
|
|
|
// It is the application's responsibility to ensure that text messages are
|
|
|
|
// valid UTF-8 encoded text.
|
2013-10-16 20:41:47 +04:00
|
|
|
//
|
|
|
|
// Control Messages
|
|
|
|
//
|
|
|
|
// The WebSocket protocol defines three types of control messages: close, ping
|
|
|
|
// and pong. Call the connection WriteControl, WriteMessage or NextWriter
|
|
|
|
// methods to send a control message to the peer.
|
|
|
|
//
|
2016-02-17 02:03:54 +03:00
|
|
|
// Connections handle received close messages by sending a close message to the
|
|
|
|
// peer and returning a *CloseError from the the NextReader, ReadMessage or the
|
|
|
|
// message Read method.
|
|
|
|
//
|
2016-03-13 21:57:37 +03:00
|
|
|
// Connections handle received ping and pong messages by invoking callback
|
|
|
|
// functions set with SetPingHandler and SetPongHandler methods. The callback
|
|
|
|
// functions are called from the NextReader, ReadMessage and the message Read
|
|
|
|
// methods.
|
|
|
|
//
|
|
|
|
// The default ping handler sends a pong to the peer. The application's reading
|
|
|
|
// goroutine can block for a short time while the handler writes the pong data
|
|
|
|
// to the connection.
|
|
|
|
//
|
|
|
|
// The application must read the connection to process ping, pong and close
|
|
|
|
// messages sent from the peer. If the application is not otherwise interested
|
|
|
|
// in messages from the peer, then the application should start a goroutine to
|
2016-02-17 02:03:54 +03:00
|
|
|
// read and discard messages from the peer. A simple example is:
|
|
|
|
//
|
|
|
|
// func readLoop(c *websocket.Conn) {
|
|
|
|
// for {
|
|
|
|
// if _, _, err := c.NextReader(); err != nil {
|
|
|
|
// c.Close()
|
|
|
|
// break
|
|
|
|
// }
|
|
|
|
// }
|
|
|
|
// }
|
2013-10-30 04:43:03 +04:00
|
|
|
//
|
|
|
|
// Concurrency
|
|
|
|
//
|
2015-10-20 01:05:01 +03:00
|
|
|
// Connections support one concurrent reader and one concurrent writer.
|
|
|
|
//
|
|
|
|
// Applications are responsible for ensuring that no more than one goroutine
|
|
|
|
// calls the write methods (NextWriter, SetWriteDeadline, WriteMessage,
|
|
|
|
// WriteJSON) concurrently and that no more than one goroutine calls the read
|
|
|
|
// methods (NextReader, SetReadDeadline, ReadMessage, ReadJSON, SetPongHandler,
|
|
|
|
// SetPingHandler) concurrently.
|
2014-11-01 00:51:45 +03:00
|
|
|
//
|
|
|
|
// The Close and WriteControl methods can be called concurrently with all other
|
|
|
|
// methods.
|
2013-10-30 04:43:03 +04:00
|
|
|
//
|
2014-10-12 20:34:51 +04:00
|
|
|
// Origin Considerations
|
|
|
|
//
|
|
|
|
// Web browsers allow Javascript applications to open a WebSocket connection to
|
|
|
|
// any host. It's up to the server to enforce an origin policy using the Origin
|
|
|
|
// request header sent by the browser.
|
|
|
|
//
|
|
|
|
// The Upgrader calls the function specified in the CheckOrigin field to check
|
|
|
|
// the origin. If the CheckOrigin function returns false, then the Upgrade
|
|
|
|
// method fails the WebSocket handshake with HTTP status 403.
|
|
|
|
//
|
|
|
|
// If the CheckOrigin field is nil, then the Upgrader uses a safe default: fail
|
|
|
|
// the handshake if the Origin request header is present and not equal to the
|
|
|
|
// Host request header.
|
|
|
|
//
|
|
|
|
// An application can allow connections from any origin by specifying a
|
|
|
|
// function that always returns true:
|
|
|
|
//
|
2016-02-11 03:22:44 +03:00
|
|
|
// var upgrader = websocket.Upgrader{
|
2014-10-12 20:34:51 +04:00
|
|
|
// CheckOrigin: func(r *http.Request) bool { return true },
|
2016-02-11 03:22:44 +03:00
|
|
|
// }
|
2014-10-12 20:34:51 +04:00
|
|
|
//
|
2014-11-01 00:51:45 +03:00
|
|
|
// The deprecated Upgrade function does not enforce an origin policy. It's the
|
2014-10-12 20:34:51 +04:00
|
|
|
// application's responsibility to check the Origin header before calling
|
|
|
|
// Upgrade.
|
2013-10-16 20:41:47 +04:00
|
|
|
package websocket
|