connection.coffee 2.4 KB
Newer Older
1
# Encapsulate the cable connection held by the consumer. This is an internal class not intended for direct user manipulation.
2

3
{message_types} = ActionCable.INTERNAL
4

5
class ActionCable.Connection
J
Javan Makhmali 已提交
6 7
  @reopenDelay: 500

J
Javan Makhmali 已提交
8
  constructor: (@consumer) ->
9
    @open()
10 11

  send: (data) ->
12
    if @isOpen()
J
Javan Makhmali 已提交
13
      @webSocket.send(JSON.stringify(data))
14 15 16 17
      true
    else
      false

18 19
  open: =>
    if @webSocket and not @isState("closed")
20
      console.log("[cable] Attemped to open WebSocket, but existing socket is #{@getState()}", Date.now())
21
      throw new Error("Existing connection must be closed before opening")
22
    else
23
      console.log("[cable] Opening WebSocket", Date.now())
24 25
      @webSocket = new WebSocket(@consumer.url)
      @installEventHandlers()
26
      true
27

28
  close: ->
J
Javan Makhmali 已提交
29
    @webSocket?.close()
30 31

  reopen: ->
32
    console.log("[cable] Reopening WebSocket, current state is #{@getState()}", Date.now())
J
Javan Makhmali 已提交
33 34 35 36 37 38
    if @isState("closed")
      @open()
    else
      try
        @close()
      finally
39
        console.log("[cable] Failed to reopen WebSocket, retrying in #{@constructor.reopenDelay}ms", Date.now())
J
Javan Makhmali 已提交
40
        setTimeout(@open, @constructor.reopenDelay)
41 42

  isOpen: ->
43 44
    @isState("open")

J
Javan Makhmali 已提交
45 46
  # Private

47 48
  isState: (states...) ->
    @getState() in states
49

50
  getState: ->
J
Javan Makhmali 已提交
51
    return state.toLowerCase() for state, value of WebSocket when value is @webSocket?.readyState
52
    null
53

J
Javan Makhmali 已提交
54 55
  installEventHandlers: ->
    for eventName of @events
56 57
      handler = @events[eventName].bind(this)
      @webSocket["on#{eventName}"] = handler
J
Javan Makhmali 已提交
58
    return
59

J
Javan Makhmali 已提交
60 61
  events:
    message: (event) ->
62 63
      {identifier, message, type} = JSON.parse(event.data)

64
      switch type
65
        when message_types.confirmation
66
          @consumer.subscriptions.notify(identifier, "connected")
67
        when message_types.rejection
68
          @consumer.subscriptions.reject(identifier)
69 70
        else
          @consumer.subscriptions.notify(identifier, "received", message)
71

J
Javan Makhmali 已提交
72
    open: ->
73
      console.log("[cable] WebSocket onopen event", Date.now())
74
      @disconnected = false
75
      @consumer.subscriptions.reload()
76

J
Javan Makhmali 已提交
77
    close: ->
78
      console.log("[cable] WebSocket onclose event", Date.now())
79
      @disconnect()
80

J
Javan Makhmali 已提交
81
    error: ->
82
      console.log("[cable] WebSocket onerror event", Date.now())
83 84 85 86 87 88
      @disconnect()

  disconnect: ->
    return if @disconnected
    @disconnected = true
    @consumer.subscriptions.notifyAll("disconnected")