Class: Tina4::WebSocketConnection

Inherits:
Object
  • Object
show all
Defined in:
lib/tina4/websocket.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(id, socket, ws_server: nil, path: "/") ⇒ WebSocketConnection

Returns a new instance of WebSocketConnection.



651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
 
# File 'lib/tina4/websocket.rb', line 651

def initialize(id, socket, ws_server: nil, path: "/")
  @id = id
  @socket = socket
  @params = {}
  @ws_server = ws_server
  @path = path
  # Verified JWT payload on a secured WS route, else nil (public route).
  # Mirrors Python's connection.auth.
  @auth = nil
  @rooms = Set.new
  @on_message_handler = nil
  @on_close_handler = nil
  @on_error_handler = nil
  # Updated on every inbound frame; the idle reaper closes connections that
  # have been silent longer than TINA4_WS_IDLE_TIMEOUT (opt-in).
  @last_activity = Time.now.to_f
end

Instance Attribute Details

#auth ⇒ Object

Returns the value of attribute auth.



648
649
650
 
# File 'lib/tina4/websocket.rb', line 648

def auth
  @auth
end

#id ⇒ Object (readonly)

Returns the value of attribute id.



647
648
649
 
# File 'lib/tina4/websocket.rb', line 647

def id
  @id
end

#last_activity ⇒ Object (readonly)

Returns the value of attribute last_activity.



647
648
649
 
# File 'lib/tina4/websocket.rb', line 647

def last_activity
  @last_activity
end

#on_close_handler ⇒ Object

Returns the value of attribute on_close_handler.



648
649
650
 
# File 'lib/tina4/websocket.rb', line 648

def on_close_handler
  @on_close_handler
end

#on_error_handler ⇒ Object

Returns the value of attribute on_error_handler.



648
649
650
 
# File 'lib/tina4/websocket.rb', line 648

def on_error_handler
  @on_error_handler
end

#on_message_handler ⇒ Object

Returns the value of attribute on_message_handler.



648
649
650
 
# File 'lib/tina4/websocket.rb', line 648

def on_message_handler
  @on_message_handler
end

#params ⇒ Object

Returns the value of attribute params.



648
649
650
 
# File 'lib/tina4/websocket.rb', line 648

def params
  @params
end

#path ⇒ Object

Returns the value of attribute path.



648
649
650
 
# File 'lib/tina4/websocket.rb', line 648

def path
  @path
end

#rooms ⇒ Object (readonly)

Returns the value of attribute rooms.



647
648
649
 
# File 'lib/tina4/websocket.rb', line 647

def rooms
  @rooms
end

Instance Method Details

#broadcast(message, include_self: false) ⇒ Object

Broadcast a message to all other connections on the same path



702
703
704
705
706
707
708
709
710
 
# File 'lib/tina4/websocket.rb', line 702

def broadcast(message, include_self: false)
  return unless @ws_server

  @ws_server.connections.each do |cid, conn|
    next if !include_self && cid == @id
    next if conn.path != @path
    conn.send_text(message)
  end
end

#broadcast_to_room(room_name, message, exclude_self: false) ⇒ Object 



694
695
696
697
698
699
 
# File 'lib/tina4/websocket.rb', line 694

def broadcast_to_room(room_name, message, exclude_self: false)
  return unless @ws_server

  exclude = exclude_self ? @id : nil
  @ws_server.broadcast_to_room(room_name, message, exclude: exclude)
end

#build_frame(opcode, data) ⇒ Object 



780
781
782
 
# File 'lib/tina4/websocket.rb', line 780

def build_frame(opcode, data)
  Tina4.build_frame(opcode, data)
end

#close(code: 1000, reason: "") ⇒ Object 



744
745
746
747
748
749
 
# File 'lib/tina4/websocket.rb', line 744

def close(code: 1000, reason: "")
  payload = [code].pack("n") + reason
  frame = build_frame(0x8, payload)
  @socket.write(frame) rescue nil
  @socket.close rescue nil
end

#closed? ⇒ Boolean

True once a write has failed (broken pipe / closed socket). The manager’s resilient broadcast path uses this to prune dead connections.

Returns:

  • (Boolean) 


714
715
716
 
# File 'lib/tina4/websocket.rb', line 714

def closed?
  @closed == true
end

#join_room(room_name) ⇒ Object 



684
685
686
687
 
# File 'lib/tina4/websocket.rb', line 684

def join_room(room_name)
  @rooms.add(room_name)
  @ws_server&._join_room(@id, room_name)
end

#leave_room(room_name) ⇒ Object 



689
690
691
692
 
# File 'lib/tina4/websocket.rb', line 689

def leave_room(room_name)
  @rooms.delete(room_name)
  @ws_server&._leave_room(@id, room_name)
end

#on_close(&block) ⇒ Object

Register a close handler (decorator style, matching Python).



680
681
682
 
# File 'lib/tina4/websocket.rb', line 680

def on_close(&block)
  @on_close_handler = block
end

#on_message(&block) ⇒ Object

Register a message handler (decorator style, matching Python).



675
676
677
 
# File 'lib/tina4/websocket.rb', line 675

def on_message(&block)
  @on_message_handler = block
end

#read_frame ⇒ Object 



751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
 
# File 'lib/tina4/websocket.rb', line 751

def read_frame
  first_byte = @socket.getbyte
  return nil unless first_byte

  opcode = first_byte & 0x0F
  second_byte = @socket.getbyte
  return nil unless second_byte

  masked = (second_byte & 0x80) != 0
  length = second_byte & 0x7F

  if length == 126
    length = @socket.read(2).unpack1("n")
  elsif length == 127
    length = @socket.read(8).unpack1("Q>")
  end

  mask_key = masked ? @socket.read(4).bytes : nil
  data = @socket.read(length) || ""

  if masked && mask_key
    data = data.bytes.each_with_index.map { |b, i| b ^ mask_key[i % 4] }.pack("C*")
  end

  { opcode: opcode, data: data }
rescue IOError, EOFError
  nil
end

#send(message) ⇒ Object Also known as: send_text 



718
719
720
721
722
723
724
725
726
727
 
# File 'lib/tina4/websocket.rb', line 718

def send(message)
  data = message.is_a?(String) ? message : message.to_s
  # Text frames must be valid UTF-8; binary payloads are sent verbatim.
  data = data.encode("UTF-8") if data.encoding != Encoding::ASCII_8BIT
  frame = build_frame(0x1, data)
  @socket.write(frame)
rescue IOError
  # Connection closed — mark dead so the broadcast path prunes it.
  @closed = true
end

#send_json(data) ⇒ Object

Serialize a Hash/Array (or any JSON-coercible value) and send as a text frame. Matches Python/PHP send_json.



733
734
735
 
# File 'lib/tina4/websocket.rb', line 733

def send_json(data)
  send_text(JSON.generate(data))
end

#send_pong(data) ⇒ Object 



737
738
739
740
741
742
 
# File 'lib/tina4/websocket.rb', line 737

def send_pong(data)
  frame = build_frame(0xA, data || "")
  @socket.write(frame)
rescue IOError
  @closed = true
end

#touch ⇒ Object

Mark inbound activity for the idle reaper.



670
671
672
 
# File 'lib/tina4/websocket.rb', line 670

def touch
  @last_activity = Time.now.to_f
end