trabajo-afectivo/lib/web_socket.rb

540 lines
16 KiB
Ruby
Raw Normal View History

2012-07-23 22:22:23 +00:00
require 'json'
module Session
@path = '/tmp/websocket'
2012-08-03 22:46:05 +00:00
@@user_threads = {}
@@client_threads = {}
2012-07-23 22:22:23 +00:00
def self.create( client_id, session )
path = @path + '/' + client_id.to_s
FileUtils.mkpath path
File.open( path + '/session', 'w' ) { |file|
user = { :id => session['id'] }
file.puts Marshal.dump(user)
}
end
def self.get( client_id )
session_file = @path + '/' + client_id.to_s + '/session'
data = nil
return if !File.exist? session_file
File.open( session_file, 'r' ) { |file|
all = ''
2012-08-03 22:46:05 +00:00
while line = file.gets
all = all + line
2012-07-23 22:22:23 +00:00
end
2012-08-03 22:46:05 +00:00
begin
2012-07-23 22:22:23 +00:00
data = Marshal.load( all )
rescue
return
end
}
return data
end
def self.transaction( client_id, data )
2012-08-03 23:35:21 +00:00
path = @path + '/' + client_id.to_s + '/'
filename = 'transaction-' + Time.new().to_i.to_s + '-' + rand(999999).to_s
if File::exists?( path + filename )
filename = filename + '-1'
if File::exists?( path + filename )
filename = filename + '-1'
if File::exists?( path + filename )
filename = filename + '-1'
if File::exists?( path + filename )
filename = filename + '-' + rand(999999).to_s
2012-07-23 22:22:23 +00:00
end
end
end
end
2012-08-03 23:35:21 +00:00
return false if !File.directory? path
File.open( path + 'a-' + filename, 'w' ) { |file|
2012-07-23 22:22:23 +00:00
file.puts data.to_json
}
2012-08-03 23:35:21 +00:00
FileUtils.mv( path + 'a-' + filename, path + filename)
2012-07-23 22:22:23 +00:00
return true
end
def self.jobs
2012-08-03 22:46:05 +00:00
Thread.abort_on_exception = true
2012-07-23 22:22:23 +00:00
while true
client_ids = self.sessions
client_ids.each { |client_id|
2012-08-07 21:53:43 +00:00
# connection already open
next if @@client_threads[client_id]
2012-07-23 22:22:23 +00:00
# get current user
user_session = Session.get( client_id )
next if !user_session
next if !user_session[:id]
user = User.find( user_session[:id] )
2012-08-03 22:46:05 +00:00
next if !user
# start user thread
start_user_thread = false
2012-08-03 22:46:05 +00:00
if !@@user_threads[user.id]
start_user_thread = true
2012-08-03 22:46:05 +00:00
@@user_threads[user.id] = Thread.new {
UserState.new(user.id)
@@user_threads[user.id] = nil
# raise "Exception from thread"
}
2012-07-23 22:22:23 +00:00
end
# wait with client thread unil user thread has done some little work
if start_user_thread
2012-08-07 21:53:43 +00:00
sleep 0.5
end
2012-08-03 22:46:05 +00:00
# start client thread
if !@@client_threads[client_id]
@@client_threads[client_id] = Thread.new {
ClientState.new(client_id)
@@client_threads[client_id] = nil
# raise "Exception from thread"
}
end
2012-07-23 22:22:23 +00:00
}
2012-08-07 06:43:15 +00:00
# system settings
sleep 0.4
2012-07-23 22:22:23 +00:00
end
end
2012-08-03 22:46:05 +00:00
def self.sessions
path = @path + '/'
data = []
Dir.foreach( path ) do |entry|
if entry != '.' && entry != '..'
data.push entry
end
2012-07-30 14:50:54 +00:00
end
2012-08-03 22:46:05 +00:00
return data
2012-07-30 14:50:54 +00:00
end
2012-08-03 22:46:05 +00:00
def self.queue( client_id )
path = @path + '/' + client_id.to_s + '/'
data = []
Dir.foreach( path ) do |entry|
if /^transaction/.match( entry )
2012-08-03 23:35:21 +00:00
data.push Session.queue_file( path, entry )
2012-08-03 22:46:05 +00:00
end
2012-07-30 14:50:54 +00:00
end
2012-08-03 22:46:05 +00:00
return data
end
2012-07-30 14:50:54 +00:00
2012-08-03 23:35:21 +00:00
def self.queue_file( path, filename )
file_old = path + filename
file_new = path + 'a-' + filename
FileUtils.mv( file_old, file_new )
2012-08-03 22:46:05 +00:00
data = nil
2012-08-03 23:35:21 +00:00
all = ''
File.open( file_new, 'r' ) { |file|
2012-08-03 22:46:05 +00:00
while line = file.gets
all = all + line
end
}
2012-08-03 23:35:21 +00:00
File.delete( file_new )
data = JSON.parse( all )
2012-08-03 22:46:05 +00:00
return data
end
2012-07-30 14:50:54 +00:00
2012-08-03 22:46:05 +00:00
def self.destory( client_id )
path = @path + '/' + client_id.to_s
FileUtils.rm_rf path
2012-07-30 14:50:54 +00:00
end
2012-08-03 22:46:05 +00:00
end
2012-07-30 14:50:54 +00:00
2012-08-03 22:46:05 +00:00
module CacheIn
@@data = {}
2012-08-07 06:43:15 +00:00
@@data_time = {}
2012-08-03 22:46:05 +00:00
@@expires_in = {}
@@expires_in_ttl = {}
2012-08-07 06:43:15 +00:00
2012-08-03 22:46:05 +00:00
def self.set( key, value, params = {} )
# puts 'CacheIn.set:' + key + '-' + value.inspect
if params[:expires_in]
@@expires_in[key] = Time.now + params[:expires_in]
@@expires_in_ttl[key] = params[:expires_in]
2012-07-30 14:50:54 +00:00
end
2012-08-03 22:46:05 +00:00
@@data[ key ] = value
2012-08-07 06:43:15 +00:00
@@data_time[ key ] = Time.now
2012-08-03 22:46:05 +00:00
end
2012-08-07 06:43:15 +00:00
def self.expired( key, params = {} )
# expire if value never was set
return true if !@@data.include? key
2012-08-07 07:30:29 +00:00
# ignore_expire
return false if params[:ignore_expire]
2012-08-07 06:43:15 +00:00
# set re_expire
if params[:re_expire]
if @@expires_in[key]
@@expires_in[key] = Time.now + @@expires_in_ttl[key]
end
return false
end
# check if expired
2012-08-03 22:46:05 +00:00
if @@expires_in[key]
return true if @@expires_in[key] < Time.now
return false
end
2012-08-07 06:43:15 +00:00
# return false if key was set without expires_in
return false
end
def self.get_time( key, params = {} )
data = self.get( key, params )
if data
return @@data_time[key]
end
2012-08-07 07:30:29 +00:00
return nil
2012-08-03 22:46:05 +00:00
end
2012-08-07 06:43:15 +00:00
2012-08-03 22:46:05 +00:00
def self.get( key, params = {} )
# puts 'CacheIn.get:' + key + '-' + @@data[ key ].inspect
2012-08-07 06:43:15 +00:00
return if self.expired( key, params )
2012-08-03 22:46:05 +00:00
@@data[ key ]
2012-07-30 14:50:54 +00:00
end
2012-08-03 22:46:05 +00:00
end
2012-07-30 14:50:54 +00:00
2012-08-03 22:46:05 +00:00
class UserState
def initialize( user_id )
@user_id = user_id
@data = {}
@cache_key = 'user_' + user_id.to_s
self.log "---user started user state"
2012-07-30 14:50:54 +00:00
2012-08-03 22:46:05 +00:00
CacheIn.set( 'last_run_' + user_id.to_s , true, { :expires_in => 20.seconds } )
2012-07-30 14:50:54 +00:00
2012-08-03 22:46:05 +00:00
self.fetch
end
def fetch
user = User.find( @user_id )
return if !user
2012-07-30 14:50:54 +00:00
2012-08-03 22:46:05 +00:00
while true
2012-08-02 12:42:50 +00:00
2012-08-03 22:46:05 +00:00
# check if user is still with min one open connection
if !CacheIn.get( 'last_run_' + user.id.to_s )
self.log "---user - closeing thread - no open user connection"
return
end
2012-08-02 12:42:50 +00:00
2012-08-03 22:46:05 +00:00
self.log "---user - fetch user data"
# overview
cache_key = @cache_key + '_overview'
if CacheIn.expired(cache_key)
overview = Ticket.overview(
2012-09-04 21:28:49 +00:00
:current_user => user,
2012-08-03 22:46:05 +00:00
)
overview_cache = CacheIn.get( cache_key, { :re_expire => true } )
self.log 'fetch overview - ' + cache_key
if overview != overview_cache
self.log 'fetch overview changed - ' + cache_key
# puts overview.inspect
# puts '------'
# puts overview_cache.inspect
CacheIn.set( cache_key, overview, { :expires_in => 3.seconds } )
end
end
2012-07-30 14:50:54 +00:00
2012-08-03 22:46:05 +00:00
# overview lists
overviews = Ticket.overview_list(
2012-09-04 21:28:49 +00:00
:current_user => user,
2012-08-03 22:46:05 +00:00
)
overviews.each { |overview|
cache_key = @cache_key + '_overview_data_' + overview.meta[:url]
if CacheIn.expired(cache_key)
overview_data = Ticket.overview(
2012-09-04 21:28:49 +00:00
:view => overview.meta[:url],
# :view_mode => params[:view_mode],
:current_user => user,
:array => true,
2012-08-03 22:46:05 +00:00
)
overview_data_cache = CacheIn.get( cache_key, { :re_expire => true } )
self.log 'fetch overview_data - ' + cache_key
if overview_data != overview_data_cache
self.log 'fetch overview_data changed - ' + cache_key
CacheIn.set( cache_key, overview_data, { :expires_in => 5.seconds } )
end
end
}
2012-07-30 14:50:54 +00:00
2012-08-03 22:46:05 +00:00
# create_attributes
cache_key = @cache_key + '_ticket_create_attributes'
if CacheIn.expired(cache_key)
ticket_create_attributes = Ticket.create_attributes(
:current_user_id => user.id,
)
ticket_create_attributes_cache = CacheIn.get( cache_key, { :re_expire => true } )
self.log 'fetch ticket_create_attributes - ' + cache_key
if ticket_create_attributes != ticket_create_attributes_cache
self.log 'fetch ticket_create_attributes changed - ' + cache_key
CacheIn.set( cache_key, ticket_create_attributes, { :expires_in => 2.minutes } )
end
end
2012-07-30 14:50:54 +00:00
2012-08-03 22:46:05 +00:00
# recent viewed
cache_key = @cache_key + '_recent_viewed'
if CacheIn.expired(cache_key)
recent_viewed = History.recent_viewed(user)
recent_viewed_cache = CacheIn.get( cache_key, { :re_expire => true } )
self.log 'fetch recent_viewed - ' + cache_key
if recent_viewed != recent_viewed_cache
self.log 'fetch recent_viewed changed - ' + cache_key
2012-07-30 14:50:54 +00:00
2012-08-03 22:46:05 +00:00
recent_viewed_full = History.recent_viewed_fulldata(user)
2012-08-04 10:34:47 +00:00
CacheIn.set( cache_key, recent_viewed, { :expires_in => 5.seconds } )
2012-08-03 22:46:05 +00:00
CacheIn.set( cache_key + '_push', recent_viewed_full )
end
end
2012-08-02 12:42:50 +00:00
2012-08-03 22:46:05 +00:00
# activity steam
cache_key = @cache_key + '_activity_stream'
if CacheIn.expired(cache_key)
activity_stream = History.activity_stream( user )
activity_stream_cache = CacheIn.get( cache_key, { :re_expire => true } )
self.log 'fetch activity_stream - ' + cache_key
if activity_stream != activity_stream_cache
self.log 'fetch activity_stream changed - ' + cache_key
2012-08-02 12:42:50 +00:00
2012-08-03 22:46:05 +00:00
activity_stream_full = History.activity_stream_fulldata( user )
2012-08-04 10:34:47 +00:00
CacheIn.set( cache_key, activity_stream, { :expires_in => 0.75.minutes } )
2012-08-03 22:46:05 +00:00
CacheIn.set( cache_key + '_push', activity_stream_full )
end
end
2012-07-30 14:50:54 +00:00
2012-08-03 22:46:05 +00:00
# rss
cache_key = @cache_key + '_rss'
if CacheIn.expired(cache_key)
url = 'http://www.heise.de/newsticker/heise-atom.xml'
rss_items = RSS.fetch( url, 8 )
rss_items_cache = CacheIn.get( cache_key, { :re_expire => true } )
self.log 'fetch rss - ' + cache_key
if rss_items != rss_items_cache
self.log 'fetch rss changed - ' + cache_key
CacheIn.set( cache_key, rss_items, { :expires_in => 2.minutes } )
CacheIn.set( cache_key + '_push', {
head: 'Heise ATOM',
items: rss_items,
})
end
end
self.log "---/user-"
2012-08-04 10:01:26 +00:00
sleep 1
2012-07-30 14:50:54 +00:00
end
end
2012-08-03 22:46:05 +00:00
def log( data )
puts "#{Time.now}:user_id(#{ @user_id }) #{ data }"
end
end
2012-07-30 14:50:54 +00:00
2012-08-03 22:46:05 +00:00
class ClientState
def initialize( client_id )
@client_id = client_id
@data = {}
@pushed = {}
self.log "---client start ws connection---"
self.fetch
self.log "---client exiting ws connection---"
end
def fetch
2012-08-06 04:37:08 +00:00
loop_count = 0
2012-08-03 22:46:05 +00:00
while true
2012-07-30 14:50:54 +00:00
2012-08-03 22:46:05 +00:00
# get connection user
user_session = Session.get( @client_id )
return if !user_session
return if !user_session[:id]
user = User.find( user_session[:id] )
return if !user
2012-08-06 04:37:08 +00:00
loop_count += 1
2012-08-03 22:46:05 +00:00
self.log "---client - looking for data of user #{user.id}"
# remember last run
CacheIn.set( 'last_run_' + user.id.to_s , true, { :expires_in => 20.seconds } )
# overview
cache_key = 'user_' + user.id.to_s + '_overview'
2012-08-07 07:30:29 +00:00
overview_time = CacheIn.get_time( cache_key, { :ignore_expire => true } )
2012-08-07 06:43:15 +00:00
if overview_time && @data[:overview_time] != overview_time
@data[:overview_time] = overview_time
2012-08-07 07:30:29 +00:00
overview = CacheIn.get( cache_key, { :ignore_expire => true } )
2012-08-07 06:43:15 +00:00
2012-08-03 22:46:05 +00:00
self.log "push overview for user #{user.id}"
# send update to browser
self.transaction({
:event => 'navupdate_ticket_overview',
:data => overview,
})
end
2012-08-02 12:42:50 +00:00
2012-08-03 22:46:05 +00:00
# overview_data
overviews = Ticket.overview_list(
2012-09-04 21:28:49 +00:00
:current_user => user,
2012-08-02 12:42:50 +00:00
)
2012-08-03 22:46:05 +00:00
overviews.each { |overview|
cache_key = 'user_' + user.id.to_s + '_overview_data_' + overview.meta[:url]
2012-08-07 07:30:29 +00:00
overview_data_time = CacheIn.get_time( cache_key, { :ignore_expire => true } )
2012-08-07 06:43:15 +00:00
if overview_data_time && @data[cache_key] != overview_data_time
@data[cache_key] = overview_data_time
2012-08-07 07:30:29 +00:00
overview_data = CacheIn.get( cache_key, { :ignore_expire => true } )
2012-08-03 22:46:05 +00:00
self.log "push overview_data for user #{user.id}"
users = {}
tickets = []
overview_data[:tickets].each {|ticket_id|
self.ticket( ticket_id, tickets, users )
}
# send update to browser
self.transaction({
:data => {
:overview => overview_data[:overview],
:ticket_list => overview_data[:tickets],
:tickets_count => overview_data[:tickets_count],
:collections => {
:User => users,
:Ticket => tickets,
}
},
:event => [ 'loadCollection', 'ticket_overview_rebuild' ],
:collection => 'ticket_overview_' + overview.meta[:url].to_s,
})
end
}
2012-07-30 14:50:54 +00:00
2012-08-03 22:46:05 +00:00
# ticket_create_attributes
cache_key = 'user_' + user.id.to_s + '_ticket_create_attributes'
2012-08-07 07:30:29 +00:00
ticket_create_attributes_time = CacheIn.get_time( cache_key, { :ignore_expire => true } )
2012-08-07 06:43:15 +00:00
if ticket_create_attributes_time && @data[:ticket_create_attributes_time] != ticket_create_attributes_time
@data[:ticket_create_attributes_time] = ticket_create_attributes_time
2012-08-07 07:30:29 +00:00
ticket_create_attributes = CacheIn.get( cache_key, { :ignore_expire => true } )
2012-08-03 22:46:05 +00:00
self.log "push ticket_create_attributes for user #{user.id}"
# send update to browser
self.transaction({
:collection => 'ticket_create_attributes',
:data => ticket_create_attributes,
})
end
2012-07-30 14:50:54 +00:00
2012-08-03 22:46:05 +00:00
# recent viewed
cache_key = 'user_' + user.id.to_s + '_recent_viewed'
2012-08-07 07:30:29 +00:00
recent_viewed_time = CacheIn.get_time( cache_key, { :ignore_expire => true } )
2012-08-07 06:43:15 +00:00
if recent_viewed_time && @data[:recent_viewed_time] != recent_viewed_time
@data[:recent_viewed_time] = recent_viewed_time
2012-08-07 07:30:29 +00:00
recent_viewed = CacheIn.get( cache_key, { :ignore_expire => true } )
2012-08-03 22:46:05 +00:00
self.log "push recent_viewed for user #{user.id}"
# send update to browser
2012-08-07 07:30:29 +00:00
r = CacheIn.get( cache_key + '_push', { :ignore_expire => true } )
2012-08-03 22:46:05 +00:00
self.transaction({
:event => 'update_recent_viewed',
:data => r,
})
end
2012-07-30 14:50:54 +00:00
2012-08-03 22:46:05 +00:00
# activity stream
cache_key = 'user_' + user.id.to_s + '_activity_stream'
2012-08-07 07:30:29 +00:00
activity_stream_time = CacheIn.get_time( cache_key, { :ignore_expire => true } )
2012-08-07 06:43:15 +00:00
if activity_stream_time && @data[:activity_stream_time] != activity_stream_time
@data[:activity_stream_time] = activity_stream_time
2012-08-07 07:30:29 +00:00
activity_stream = CacheIn.get( cache_key, { :ignore_expire => true } )
2012-08-03 22:46:05 +00:00
self.log "push activity_stream for user #{user.id}"
# send update to browser
2012-08-07 07:30:29 +00:00
r = CacheIn.get( cache_key + '_push', { :ignore_expire => true } )
2012-08-03 22:46:05 +00:00
self.transaction({
:event => 'activity_stream_rebuild',
:collection => 'activity_stream',
:data => r,
})
end
2012-08-02 12:42:50 +00:00
2012-08-03 22:46:05 +00:00
# rss
cache_key = 'user_' + user.id.to_s + '_rss'
2012-08-07 07:30:29 +00:00
rss_items_time = CacheIn.get_time( cache_key, { :ignore_expire => true } )
2012-08-07 06:43:15 +00:00
if rss_items_time && @data[:rss_time] != rss_items_time
@data[:rss_time] = rss_items_time
2012-08-07 07:30:29 +00:00
rss_items = CacheIn.get( cache_key, { :ignore_expire => true } )
2012-08-03 22:46:05 +00:00
self.log "push rss for user #{user.id}"
# send update to browser
2012-08-07 07:30:29 +00:00
r = CacheIn.get( cache_key + '_push', { :ignore_expire => true } )
2012-08-03 22:46:05 +00:00
self.transaction({
:event => 'rss_rebuild',
:collection => 'dashboard_rss',
:data => r,
})
end
self.log "---/client-"
2012-08-06 04:37:08 +00:00
# start faster in the beginnig
if loop_count < 20
2012-08-07 21:53:43 +00:00
sleep 0.6
2012-08-06 04:37:08 +00:00
else
sleep 1
end
2012-08-02 12:42:50 +00:00
end
2012-08-03 22:46:05 +00:00
end
2012-08-02 12:42:50 +00:00
2012-08-03 22:46:05 +00:00
# add ticket if needed
def ticket( ticket_id, tickets, users )
if !@pushed[:tickets]
@pushed[:tickets] = {}
2012-08-02 12:42:50 +00:00
end
2012-08-03 22:46:05 +00:00
ticket = Ticket.full_data(ticket_id)
if @pushed[:tickets][ticket_id] != ticket
@pushed[:tickets][ticket_id] = ticket
tickets.push ticket
2012-08-02 12:42:50 +00:00
end
2012-08-03 22:46:05 +00:00
# add users if needed
self.user( ticket['owner_id'], users )
self.user( ticket['customer_id'], users )
self.user( ticket['created_by_id'], users )
2012-08-02 12:42:50 +00:00
end
2012-08-03 22:46:05 +00:00
# add user if needed
def user( user_id, users )
if !@pushed[:users]
@pushed[:users] = {}
end
2012-08-02 12:42:50 +00:00
2012-08-03 22:46:05 +00:00
# get user
user = User.user_data_full( user_id )
2012-08-02 12:42:50 +00:00
2012-08-03 22:46:05 +00:00
# user is already on client and not changed
return if @pushed[:users][ user_id ] == user
@pushed[:users][user_id] = user
2012-08-02 12:42:50 +00:00
2012-08-03 22:46:05 +00:00
# user not on client or different
self.log 'push user ... ' + user['login']
users[ user_id ] = user
2012-07-23 22:22:23 +00:00
end
2012-08-03 22:46:05 +00:00
# send update to browser
def transaction( data )
Session.transaction( @client_id, data )
2012-07-23 22:22:23 +00:00
end
2012-08-03 22:46:05 +00:00
def log( data )
puts "#{Time.now}:client(#{ @client_id }) #{ data }"
2012-07-23 22:22:23 +00:00
end
2012-08-03 22:46:05 +00:00
end