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

                   
                                                                                     
                          

                                   
             
                            

                                                 
 


                                                                         

                                                                      
 
                                                                                               

                                                                          
           
 
                                             
                                              
                                                                                                                                              
             
           
 
                                             



                                                       
                                                                        
                           

                             

                                                                          
 
               
           


       
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(message) }
            @_internal_subscriptions ||= []
            @_internal_subscriptions << [ internal_channel, callback ]

            Concurrent.global_io_executor.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| Concurrent.global_io_executor.post { pubsub.unsubscribe(channel, callback) } }
          end
        end

        def process_internal_message(message)
          message = ActiveSupport::JSON.decode(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