7583568be8
Also: * remove ShareablesFromPerson evil-query * improve multi-stream and aspect-stream queries * fix logging for recieve * don't add last 100 public posts to users streams after sharing * delete share visibility when shareable is deleted
77 lines
2.2 KiB
Ruby
77 lines
2.2 KiB
Ruby
# Copyright (c) 2010-2011, Diaspora Inc. This file is
|
|
# licensed under the Affero General Public License version 3 or later. See
|
|
# the COPYRIGHT file.
|
|
|
|
class Postzord::Receiver::LocalBatch < Postzord::Receiver
|
|
|
|
attr_reader :object, :recipient_user_ids, :users
|
|
|
|
def initialize(object, recipient_user_ids)
|
|
@object = object
|
|
@recipient_user_ids = recipient_user_ids
|
|
@users = User.where(:id => @recipient_user_ids)
|
|
|
|
end
|
|
|
|
def receive!
|
|
logger.info "receiving local batch for #{@object.inspect}"
|
|
if @object.respond_to?(:relayable?)
|
|
receive_relayable
|
|
else
|
|
create_share_visibilities
|
|
end
|
|
notify_mentioned_users if @object.respond_to?(:mentions)
|
|
|
|
# 09/27/11 this is slow
|
|
notify_users
|
|
|
|
logger.info "receiving local batch completed for #{@object.inspect}"
|
|
end
|
|
|
|
# NOTE(copied over from receiver public)
|
|
# @return [void]
|
|
def receive_relayable
|
|
if @object.parent_author.local?
|
|
# receive relayable object only for the owner of the parent object
|
|
@object.receive(@object.parent_author.owner)
|
|
end
|
|
end
|
|
|
|
# Batch import post visibilities for the recipients of the given @object
|
|
# @note performs a bulk insert into mySQL
|
|
# @return [void]
|
|
def create_share_visibilities
|
|
ShareVisibility.batch_import(@recipient_user_ids, object)
|
|
end
|
|
|
|
# Notify any mentioned users within the @object's text
|
|
# @return [void]
|
|
def notify_mentioned_users
|
|
@object.mentions.each do |mention|
|
|
mention.notify_recipient
|
|
end
|
|
end
|
|
|
|
#NOTE(these methods should be in their own module, included in this class)
|
|
# Notify users of the new object
|
|
# return [void]
|
|
def notify_users
|
|
return unless @object.respond_to?(:notification_type)
|
|
@users.find_each do |user|
|
|
Notification.notify(user, @object, @object.author)
|
|
end
|
|
if @object.respond_to?(:target)
|
|
additional_subscriber = @object.target.author.owner
|
|
elsif @object.respond_to?(:post)
|
|
additional_subscriber = @object.post.author.owner
|
|
end
|
|
|
|
Notification.notify(additional_subscriber, @object, @object.author) if needs_notification?(additional_subscriber)
|
|
end
|
|
|
|
private
|
|
|
|
def needs_notification?(person)
|
|
person && person != @object.author.owner && !@users.exists?(person.id)
|
|
end
|
|
end
|