116 lines
2.7 KiB
Ruby
116 lines
2.7 KiB
Ruby
# Copyright (C) 2012-2021 Zammad Foundation, http://zammad-foundation.org/
|
|
|
|
class LongPollingController < ApplicationController
|
|
skip_before_action :session_update # prevent race conditions
|
|
prepend_before_action :authentication_check_only
|
|
|
|
# GET /api/v1/message_send
|
|
def message_send
|
|
new_connection = false
|
|
|
|
# check client id
|
|
client_id = client_id_verify
|
|
if !client_id
|
|
new_connection = true
|
|
client_id = client_id_gen
|
|
log 'new client connection', client_id
|
|
end
|
|
data = params['data'].permit!.to_h
|
|
session_data = {}
|
|
if current_user&.id
|
|
session_data = { 'id' => current_user.id }
|
|
end
|
|
|
|
# spool messages for new connects
|
|
if data['spool']
|
|
Sessions.spool_create(data)
|
|
end
|
|
if data['event'] == 'login'
|
|
Sessions.create(client_id, session_data, { type: 'ajax' })
|
|
elsif data['event']
|
|
message = Sessions::Event.run(
|
|
event: data['event'],
|
|
payload: data,
|
|
session: session_data,
|
|
client_id: client_id,
|
|
clients: {},
|
|
options: {},
|
|
)
|
|
if message
|
|
Sessions.send(client_id, message)
|
|
end
|
|
else
|
|
log "unknown message '#{data.inspect}'", client_id
|
|
end
|
|
|
|
if new_connection
|
|
result = { client_id: client_id }
|
|
render json: result
|
|
return
|
|
end
|
|
render json: {}
|
|
end
|
|
|
|
# GET /api/v1/message_receive
|
|
def message_receive
|
|
|
|
# check client id
|
|
client_id = client_id_verify
|
|
raise Exceptions::UnprocessableEntity, 'Invalid client_id receive!' if !client_id
|
|
|
|
# check queue to send
|
|
begin
|
|
# update last ping
|
|
4.times do
|
|
sleep 0.25
|
|
end
|
|
# sleep 1
|
|
Sessions.touch(client_id) # rubocop:disable Rails/SkipsModelValidations
|
|
|
|
# set max loop time to 24 sec. because of 30 sec. timeout of mod_proxy
|
|
count = 3
|
|
if Rails.env.production?
|
|
count = 12
|
|
end
|
|
loop do
|
|
count -= 1
|
|
queue = Sessions.queue(client_id)
|
|
if queue && queue[0]
|
|
logger.debug { "send #{queue.inspect} to #{client_id}" }
|
|
render json: queue
|
|
return
|
|
end
|
|
8.times do
|
|
sleep 0.25
|
|
end
|
|
# sleep 2
|
|
if count.zero?
|
|
render json: { event: 'pong' }
|
|
return
|
|
end
|
|
end
|
|
rescue
|
|
raise Exceptions::UnprocessableEntity, 'Invalid client_id in receive loop!'
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def client_id_gen
|
|
SecureRandom.uuid
|
|
end
|
|
|
|
def client_id_verify
|
|
return if !params[:client_id]
|
|
|
|
sessions = Sessions.sessions
|
|
return if sessions.exclude?(params[:client_id].to_s)
|
|
|
|
params[:client_id].to_s
|
|
end
|
|
|
|
def log(data, client_id = '-')
|
|
logger.info "client(#{client_id}) #{data}"
|
|
end
|
|
end
|