aboutsummaryrefslogblamecommitdiffstats
path: root/actioncable/lib/action_cable/connection/internal_channel.rb
blob: f03904137b6db3b63f1c93916325a6be33ae2cb9 (plain) (tree)
1
2
3
4
5
6
7
8
9

                             

                   
                                                                                     
                          

                                   
             
                            

                                                 
 

                                           
                                                                                

                                                                      
 
                                                                                   

                                                                          
           
 
                                             
                                              
                                                                                                                                  
             
           
 
                                             

                              
                                                                        
                           

                             

                                                                          
 
               
           


       
# frozen_string_literal: true

module ActionCable
  module Connection
    # Makes it possible for the RemoteConnection to disconnect a specific connection.
    module InternalChannel
      extend ActiveSupport::Concern

      private
        def internal_channel
          "action_cable/#{connection_identifier}"
        end

        def subscribe_to_internal_channel
          if connection_identifier.present?
            callback = -> (message) { process_internal_message decode(message) }
            @_internal_subscriptions ||= []
            @_internal_subscriptions << [ internal_channel, callback ]

            server.event_loop.post { pubsub.subscribe(internal_channel, callback) }
            logger.info "Registered connection (#{connection_identifier})"
          end
        end

        def unsubscribe_from_internal_channel
          if @_internal_subscriptions.present?
            @_internal_subscriptions.each { |channel, callback| server.event_loop.post { pubsub.unsubscribe(channel, callback) } }
          end
        end

        def process_internal_message(message)
          case message["type"]
          when "disconnect"
            logger.info "Removing connection (#{connection_identifier})"
            websocket.close
          end
        rescue Exception => e
          logger.error "There was an exception - #{e.class}(#{e.message})"
          logger.error e.backtrace.join("\n")

          close
        end
    end
  end
end