mirror of
https://github.com/codeninjasllc/discourse.git
synced 2024-11-27 09:36:19 -05:00
Don't blow up if Redis switches to READONLY
This commit is contained in:
parent
8129cce104
commit
5b3f99aa50
56 changed files with 194 additions and 147 deletions
|
@ -54,7 +54,7 @@ class Admin::UsersController < Admin::AdminController
|
|||
@user.suspended_at = DateTime.now
|
||||
@user.save!
|
||||
StaffActionLogger.new(current_user).log_user_suspend(@user, params[:reason])
|
||||
MessageBus.publish "/logout", @user.id, user_ids: [@user.id]
|
||||
DiscourseBus.publish "/logout", @user.id, user_ids: [@user.id]
|
||||
render nothing: true
|
||||
end
|
||||
|
||||
|
@ -71,7 +71,7 @@ class Admin::UsersController < Admin::AdminController
|
|||
if @user
|
||||
@user.auth_token = nil
|
||||
@user.save!
|
||||
MessageBus.publish "/logout", @user.id, user_ids: [@user.id]
|
||||
DiscourseBus.publish "/logout", @user.id, user_ids: [@user.id]
|
||||
render json: success_json
|
||||
else
|
||||
render json: {error: I18n.t('admin_js.admin.users.id_not_found')}, status: 404
|
||||
|
@ -350,7 +350,7 @@ class Admin::UsersController < Admin::AdminController
|
|||
end
|
||||
|
||||
def refresh_browser(user)
|
||||
MessageBus.publish "/file-change", ["refresh"], user_ids: [user.id]
|
||||
DiscourseBus.publish "/file-change", ["refresh"], user_ids: [user.id]
|
||||
end
|
||||
|
||||
end
|
||||
|
|
|
@ -4,4 +4,4 @@ class Admin::VersionsController < Admin::AdminController
|
|||
def show
|
||||
render json: DiscourseUpdates.check_version
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
|
@ -222,7 +222,7 @@ SQL
|
|||
end
|
||||
|
||||
def publish_categories_list
|
||||
MessageBus.publish('/categories', {categories: ActiveModel::ArraySerializer.new(Category.latest).as_json})
|
||||
DiscourseBus.publish('/categories', {categories: ActiveModel::ArraySerializer.new(Category.latest).as_json})
|
||||
end
|
||||
|
||||
def parent_category_validator
|
||||
|
|
|
@ -97,7 +97,7 @@ class ColorScheme < ActiveRecord::Base
|
|||
end
|
||||
|
||||
def publish_discourse_stylesheet
|
||||
MessageBus.publish("/discourse_stylesheet", self.name)
|
||||
DiscourseBus.publish("/discourse_stylesheet", self.name)
|
||||
DiscourseStylesheets.cache.clear
|
||||
end
|
||||
|
||||
|
|
|
@ -44,4 +44,4 @@ module Roleable
|
|||
save_and_refresh_staff_groups!
|
||||
end
|
||||
|
||||
end
|
||||
end
|
||||
|
|
|
@ -97,7 +97,7 @@ class Post < ActiveRecord::Base
|
|||
# special failsafe for posts missing topics
|
||||
# consistency checks should fix, but message
|
||||
# is safe to skip
|
||||
MessageBus.publish("/topic/#{topic_id}", {
|
||||
DiscourseBus.publish("/topic/#{topic_id}", {
|
||||
id: id,
|
||||
post_number: post_number,
|
||||
updated_at: Time.now,
|
||||
|
|
|
@ -56,7 +56,7 @@ class PostAction < ActiveRecord::Base
|
|||
|
||||
$redis.set('posts_flagged_count', posts_flagged_count)
|
||||
user_ids = User.staff.pluck(:id)
|
||||
MessageBus.publish('/flagged_counts', { total: posts_flagged_count }, { user_ids: user_ids })
|
||||
DiscourseBus.publish('/flagged_counts', { total: posts_flagged_count }, { user_ids: user_ids })
|
||||
end
|
||||
|
||||
def self.flagged_posts_count
|
||||
|
|
|
@ -30,7 +30,7 @@ class QueuedPost < ActiveRecord::Base
|
|||
|
||||
def self.broadcast_new!
|
||||
msg = { post_queue_new_count: QueuedPost.new_count }
|
||||
MessageBus.publish('/queue_counts', msg, user_ids: User.staff.pluck(:id))
|
||||
DiscourseBus.publish('/queue_counts', msg, user_ids: User.staff.pluck(:id))
|
||||
end
|
||||
|
||||
def reject!(rejected_by)
|
||||
|
|
|
@ -34,11 +34,11 @@ class SiteCustomization < ActiveRecord::Base
|
|||
after_save do
|
||||
remove_from_cache!
|
||||
if stylesheet_changed? || mobile_stylesheet_changed?
|
||||
MessageBus.publish "/file-change/#{key}", SecureRandom.hex
|
||||
MessageBus.publish "/file-change/#{SiteCustomization::ENABLED_KEY}", SecureRandom.hex
|
||||
DiscourseBus.publish "/file-change/#{key}", SecureRandom.hex
|
||||
DiscourseBus.publish "/file-change/#{SiteCustomization::ENABLED_KEY}", SecureRandom.hex
|
||||
end
|
||||
MessageBus.publish "/header-change/#{key}", header if header_changed?
|
||||
MessageBus.publish "/footer-change/#{key}", footer if footer_changed?
|
||||
DiscourseBus.publish "/header-change/#{key}", header if header_changed?
|
||||
DiscourseBus.publish "/footer-change/#{key}", footer if footer_changed?
|
||||
DiscourseStylesheets.cache.clear
|
||||
end
|
||||
|
||||
|
@ -109,7 +109,7 @@ class SiteCustomization < ActiveRecord::Base
|
|||
end
|
||||
|
||||
def self.remove_from_cache!(key, broadcast = true)
|
||||
MessageBus.publish('/site_customization', key: key) if broadcast
|
||||
DiscourseBus.publish('/site_customization', key: key) if broadcast
|
||||
clear_cache!
|
||||
end
|
||||
|
||||
|
|
|
@ -636,7 +636,7 @@ class Topic < ActiveRecord::Base
|
|||
self.add_moderator_post(user, I18n.t("archetypes.banner.message.make"))
|
||||
self.save
|
||||
|
||||
MessageBus.publish('/site/banner', banner)
|
||||
DiscourseBus.publish('/site/banner', banner)
|
||||
end
|
||||
|
||||
def remove_banner!(user)
|
||||
|
@ -644,7 +644,7 @@ class Topic < ActiveRecord::Base
|
|||
self.add_moderator_post(user, I18n.t("archetypes.banner.message.remove"))
|
||||
self.save
|
||||
|
||||
MessageBus.publish('/site/banner', nil)
|
||||
DiscourseBus.publish('/site/banner', nil)
|
||||
end
|
||||
|
||||
def banner
|
||||
|
|
|
@ -33,7 +33,7 @@ class TopicTrackingState
|
|||
|
||||
group_ids = topic.category && topic.category.secure_group_ids
|
||||
|
||||
MessageBus.publish("/new", message.as_json, group_ids: group_ids)
|
||||
DiscourseBus.publish("/new", message.as_json, group_ids: group_ids)
|
||||
publish_read(topic.id, 1, topic.user_id)
|
||||
end
|
||||
|
||||
|
@ -51,7 +51,7 @@ class TopicTrackingState
|
|||
}
|
||||
|
||||
group_ids = topic.category && topic.category.secure_group_ids
|
||||
MessageBus.publish("/latest", message.as_json, group_ids: group_ids)
|
||||
DiscourseBus.publish("/latest", message.as_json, group_ids: group_ids)
|
||||
end
|
||||
|
||||
def self.publish_unread(post)
|
||||
|
@ -77,7 +77,7 @@ class TopicTrackingState
|
|||
}
|
||||
}
|
||||
|
||||
MessageBus.publish("/unread/#{tu.user_id}", message.as_json, group_ids: group_ids)
|
||||
DiscourseBus.publish("/unread/#{tu.user_id}", message.as_json, group_ids: group_ids)
|
||||
end
|
||||
|
||||
end
|
||||
|
@ -97,7 +97,7 @@ class TopicTrackingState
|
|||
}
|
||||
}
|
||||
|
||||
MessageBus.publish("/unread/#{user_id}", message.as_json, user_ids: [user_id])
|
||||
DiscourseBus.publish("/unread/#{user_id}", message.as_json, user_ids: [user_id])
|
||||
|
||||
end
|
||||
|
||||
|
|
|
@ -40,7 +40,7 @@ class TopicUser < ActiveRecord::Base
|
|||
notifications_reason_id: reason
|
||||
)
|
||||
|
||||
MessageBus.publish("/topic/#{topic_id}", {
|
||||
DiscourseBus.publish("/topic/#{topic_id}", {
|
||||
notification_level_change: notification_levels[:tracking],
|
||||
notifications_reason_id: reason
|
||||
}, user_ids: [user_id])
|
||||
|
@ -113,7 +113,7 @@ class TopicUser < ActiveRecord::Base
|
|||
end
|
||||
|
||||
if attrs[:notification_level]
|
||||
MessageBus.publish("/topic/#{topic_id}",
|
||||
DiscourseBus.publish("/topic/#{topic_id}",
|
||||
{notification_level_change: attrs[:notification_level]}, user_ids: [user_id])
|
||||
end
|
||||
|
||||
|
@ -196,7 +196,7 @@ class TopicUser < ActiveRecord::Base
|
|||
end
|
||||
|
||||
if before != after
|
||||
MessageBus.publish("/topic/#{topic_id}", {notification_level_change: after}, user_ids: [user.id])
|
||||
DiscourseBus.publish("/topic/#{topic_id}", {notification_level_change: after}, user_ids: [user.id])
|
||||
end
|
||||
end
|
||||
|
||||
|
@ -219,7 +219,7 @@ class TopicUser < ActiveRecord::Base
|
|||
WHERE ftu.user_id = :user_id and ftu.topic_id = :topic_id)",
|
||||
args)
|
||||
|
||||
MessageBus.publish("/topic/#{topic_id}", {notification_level_change: args[:new_status]}, user_ids: [user.id])
|
||||
DiscourseBus.publish("/topic/#{topic_id}", {notification_level_change: args[:new_status]}, user_ids: [user.id])
|
||||
end
|
||||
end
|
||||
|
||||
|
|
|
@ -206,7 +206,7 @@ class User < ActiveRecord::Base
|
|||
# tricky, we need our bus to be subscribed from the right spot
|
||||
def sync_notification_channel_position
|
||||
@unread_notifications_by_type = nil
|
||||
self.notification_channel_position = MessageBus.last_id("/notification/#{id}")
|
||||
self.notification_channel_position = DiscourseBus.last_id("/notification/#{id}")
|
||||
end
|
||||
|
||||
def invited_by
|
||||
|
@ -298,7 +298,7 @@ class User < ActiveRecord::Base
|
|||
end
|
||||
|
||||
def publish_notifications_state
|
||||
MessageBus.publish("/notification/#{id}",
|
||||
DiscourseBus.publish("/notification/#{id}",
|
||||
{unread_notifications: unread_notifications,
|
||||
unread_private_messages: unread_private_messages,
|
||||
total_unread_notifications: total_unread_notifications},
|
||||
|
|
|
@ -224,7 +224,7 @@ SQL
|
|||
end
|
||||
|
||||
if action.user
|
||||
MessageBus.publish("/users/#{action.user.username.downcase}", action.id, user_ids: [user_id], group_ids: group_ids)
|
||||
DiscourseBus.publish("/users/#{action.user.username.downcase}", action.id, user_ids: [user_id], group_ids: group_ids)
|
||||
end
|
||||
|
||||
action
|
||||
|
@ -240,7 +240,7 @@ SQL
|
|||
require_parameters(hash, :action_type, :user_id, :acting_user_id, :target_topic_id, :target_post_id)
|
||||
if action = UserAction.find_by(hash.except(:created_at))
|
||||
action.destroy
|
||||
MessageBus.publish("/user/#{hash[:user_id]}", {user_action_id: action.id, remove: true})
|
||||
DiscourseBus.publish("/user/#{hash[:user_id]}", {user_action_id: action.id, remove: true})
|
||||
end
|
||||
|
||||
update_like_count(hash[:user_id], hash[:action_type], -1)
|
||||
|
|
|
@ -5,4 +5,4 @@ class ColorSchemeSerializer < ApplicationSerializer
|
|||
def base
|
||||
object.is_base || false
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
|
@ -2,4 +2,4 @@ class DiscourseVersionCheckSerializer < ApplicationSerializer
|
|||
attributes :latest_version, :critical_updates, :installed_version, :installed_sha, :missing_versions_count, :updated_at
|
||||
|
||||
self.root = false
|
||||
end
|
||||
end
|
||||
|
|
|
@ -42,4 +42,4 @@ class SpamRule::AutoBlock
|
|||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
|
@ -37,4 +37,4 @@ class UserBlocker
|
|||
end
|
||||
end
|
||||
|
||||
end
|
||||
end
|
||||
|
|
|
@ -77,7 +77,7 @@ class UserDestroyer
|
|||
end
|
||||
|
||||
StaffActionLogger.new(@actor == user ? Discourse.system_user : @actor).log_user_deletion(user, opts.slice(:context))
|
||||
MessageBus.publish "/file-change", ["refresh"], user_ids: [user.id]
|
||||
DiscourseBus.publish "/file-change", ["refresh"], user_ids: [user.id]
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
3
config/initializers/04-discourse_bus.rb
Normal file
3
config/initializers/04-discourse_bus.rb
Normal file
|
@ -0,0 +1,3 @@
|
|||
require_dependency 'discourse_bus'
|
||||
|
||||
DiscourseBus.start!
|
|
@ -1,47 +0,0 @@
|
|||
MessageBus.site_id_lookup do
|
||||
RailsMultisite::ConnectionManagement.current_db
|
||||
end
|
||||
|
||||
MessageBus.extra_response_headers_lookup do |env|
|
||||
{
|
||||
"Access-Control-Allow-Origin" => Discourse.base_url_no_prefix,
|
||||
"Access-Control-Allow-Methods" => "GET, POST",
|
||||
"Access-Control-Allow-Headers" => "X-SILENCE-LOGGER, X-Shared-Session-Key"
|
||||
}
|
||||
end
|
||||
|
||||
MessageBus.user_id_lookup do |env|
|
||||
user = CurrentUser.lookup_from_env(env)
|
||||
user.id if user
|
||||
end
|
||||
|
||||
MessageBus.group_ids_lookup do |env|
|
||||
user = CurrentUser.lookup_from_env(env)
|
||||
user.groups.select('groups.id').map{|g| g.id} if user
|
||||
end
|
||||
|
||||
MessageBus.on_connect do |site_id|
|
||||
RailsMultisite::ConnectionManagement.establish_connection(db: site_id)
|
||||
end
|
||||
|
||||
MessageBus.on_disconnect do |site_id|
|
||||
ActiveRecord::Base.connection_handler.clear_active_connections!
|
||||
end
|
||||
|
||||
# Point at our redis
|
||||
MessageBus.redis_config = YAML.load(ERB.new(File.new("#{Rails.root}/config/redis.yml").read).result)[Rails.env].symbolize_keys
|
||||
|
||||
MessageBus.long_polling_enabled = SiteSetting.enable_long_polling
|
||||
MessageBus.long_polling_interval = SiteSetting.long_polling_interval
|
||||
|
||||
MessageBus.is_admin_lookup do |env|
|
||||
user = CurrentUser.lookup_from_env(env)
|
||||
if user && user.admin
|
||||
true
|
||||
else
|
||||
false
|
||||
end
|
||||
end
|
||||
|
||||
MessageBus.cache_assets = !Rails.env.development?
|
||||
MessageBus.enable_diagnostics
|
|
@ -1,3 +1,3 @@
|
|||
require 'rack/protection'
|
||||
|
||||
Rails.configuration.middleware.use Rack::Protection::FrameOptions
|
||||
Rails.configuration.middleware.use Rack::Protection::FrameOptions
|
||||
|
|
|
@ -1,2 +1,2 @@
|
|||
# Not sure why it's not using this by default!
|
||||
MultiJson.engine = :oj
|
||||
MultiJson.engine = :oj
|
||||
|
|
|
@ -2,4 +2,4 @@ require_dependency 'twitter_api'
|
|||
|
||||
Onebox.options = {
|
||||
twitter_client: TwitterApi
|
||||
}
|
||||
}
|
||||
|
|
|
@ -20,4 +20,4 @@ if Rails.env.test?
|
|||
end
|
||||
end
|
||||
|
||||
end
|
||||
end
|
||||
|
|
|
@ -1 +1 @@
|
|||
ActiveRecord::Base.send(:include, ActiveModel::ForbiddenAttributesProtection)
|
||||
ActiveRecord::Base.send(:include, ActiveModel::ForbiddenAttributesProtection)
|
||||
|
|
|
@ -28,4 +28,4 @@ END
|
|||
|
||||
raise "Assets have not been precompiled"
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
|
@ -8,4 +8,4 @@ class AddMissingIdColumns < ActiveRecord::Migration
|
|||
remove_column :category_featured_topics, :id
|
||||
remove_column :topic_users, :id
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
|
@ -30,4 +30,4 @@ class MigrateWordCounts < ActiveRecord::Migration
|
|||
|
||||
end
|
||||
|
||||
end
|
||||
end
|
||||
|
|
|
@ -2,4 +2,4 @@ class AddProfileBackgroundToUser < ActiveRecord::Migration
|
|||
def change
|
||||
add_column :users, :profile_background, :string, limit: 255
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
|
@ -2,4 +2,4 @@ class AddExternalAvatarUrlToSingleSignOnRecord < ActiveRecord::Migration
|
|||
def change
|
||||
add_column :single_sign_on_records, :external_avatar_url, :string
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
|
@ -85,7 +85,7 @@ class Auth::DefaultCurrentUserProvider
|
|||
if SiteSetting.log_out_strict && (user = current_user)
|
||||
user.auth_token = nil
|
||||
user.save!
|
||||
MessageBus.publish "/logout", user.id, user_ids: [user.id]
|
||||
DiscourseBus.publish "/logout", user.id, user_ids: [user.id]
|
||||
end
|
||||
cookies[TOKEN_COOKIE] = nil
|
||||
end
|
||||
|
|
|
@ -64,4 +64,4 @@ class Auth::GoogleOAuth2Authenticator < Auth::Authenticator
|
|||
|
||||
h
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
|
@ -62,7 +62,7 @@ module BackupRestore
|
|||
|
||||
def self.logs
|
||||
id = start_logs_message_id
|
||||
MessageBus.backlog(LOGS_CHANNEL, id).map { |m| m.data }
|
||||
DiscourseBus.backlog(LOGS_CHANNEL, id).map { |m| m.data }
|
||||
end
|
||||
|
||||
def self.current_version
|
||||
|
@ -142,7 +142,7 @@ module BackupRestore
|
|||
end
|
||||
|
||||
def self.save_start_logs_message_id
|
||||
id = MessageBus.last_id(LOGS_CHANNEL)
|
||||
id = DiscourseBus.last_id(LOGS_CHANNEL)
|
||||
$redis.set(start_logs_message_id_key, id)
|
||||
end
|
||||
|
||||
|
|
|
@ -336,7 +336,7 @@ module BackupRestore
|
|||
def publish_log(message, timestamp)
|
||||
return unless @publish_to_message_bus
|
||||
data = { timestamp: timestamp, operation: "backup", message: message }
|
||||
MessageBus.publish(BackupRestore::LOGS_CHANNEL, data, user_ids: [@user_id])
|
||||
DiscourseBus.publish(BackupRestore::LOGS_CHANNEL, data, user_ids: [@user_id])
|
||||
end
|
||||
|
||||
def save_log(message, timestamp)
|
||||
|
|
|
@ -354,7 +354,7 @@ module BackupRestore
|
|||
def publish_log(message, timestamp)
|
||||
return unless @publish_to_message_bus
|
||||
data = { timestamp: timestamp, operation: "restore", message: message }
|
||||
MessageBus.publish(BackupRestore::LOGS_CHANNEL, data, user_ids: [@user_id])
|
||||
DiscourseBus.publish(BackupRestore::LOGS_CHANNEL, data, user_ids: [@user_id])
|
||||
end
|
||||
|
||||
def save_log(message, timestamp)
|
||||
|
|
|
@ -12,4 +12,4 @@ module ConfigurableUrls
|
|||
SiteSetting.privacy_policy_url.blank? ? "#{Discourse::base_uri}/privacy" : SiteSetting.privacy_policy_url
|
||||
end
|
||||
|
||||
end
|
||||
end
|
||||
|
|
|
@ -95,10 +95,10 @@ module Discourse
|
|||
digest = Digest::MD5.hexdigest(ActionView::Base.assets_manifest.assets.values.sort.join)
|
||||
|
||||
channel = "/global/asset-version"
|
||||
message = MessageBus.last_message(channel)
|
||||
message = DiscourseBus.last_message(channel)
|
||||
|
||||
unless message && message.data == digest
|
||||
MessageBus.publish channel, digest
|
||||
DiscourseBus.publish channel, digest
|
||||
end
|
||||
digest
|
||||
end
|
||||
|
@ -169,7 +169,7 @@ module Discourse
|
|||
|
||||
def self.enable_readonly_mode
|
||||
$redis.set(readonly_mode_key, 1)
|
||||
MessageBus.publish(readonly_channel, true)
|
||||
DiscourseBus.publish(readonly_channel, true)
|
||||
keep_readonly_mode
|
||||
true
|
||||
end
|
||||
|
@ -186,7 +186,7 @@ module Discourse
|
|||
|
||||
def self.disable_readonly_mode
|
||||
$redis.del(readonly_mode_key)
|
||||
MessageBus.publish(readonly_channel, false)
|
||||
DiscourseBus.publish(readonly_channel, false)
|
||||
true
|
||||
end
|
||||
|
||||
|
@ -197,9 +197,9 @@ module Discourse
|
|||
def self.request_refresh!
|
||||
# Causes refresh on next click for all clients
|
||||
#
|
||||
# This is better than `MessageBus.publish "/file-change", ["refresh"]` because
|
||||
# This is better than `DiscourseBus.publish "/file-change", ["refresh"]` because
|
||||
# it spreads the refreshes out over a time period
|
||||
MessageBus.publish '/global/asset-version', 'clobber'
|
||||
DiscourseBus.publish '/global/asset-version', 'clobber'
|
||||
end
|
||||
|
||||
def self.git_version
|
||||
|
@ -274,7 +274,7 @@ module Discourse
|
|||
def self.after_fork
|
||||
current_db = RailsMultisite::ConnectionManagement.current_db
|
||||
RailsMultisite::ConnectionManagement.establish_connection(db: current_db)
|
||||
MessageBus.after_fork
|
||||
DiscourseBus.after_fork
|
||||
SiteSetting.after_fork
|
||||
$redis.client.reconnect
|
||||
Rails.cache.reconnect
|
||||
|
|
73
lib/discourse_bus.rb
Normal file
73
lib/discourse_bus.rb
Normal file
|
@ -0,0 +1,73 @@
|
|||
# Proxies MessageBus and noops READONLY errors.
|
||||
module DiscourseBus
|
||||
|
||||
def self.start!
|
||||
MessageBus.site_id_lookup do
|
||||
RailsMultisite::ConnectionManagement.current_db
|
||||
end
|
||||
|
||||
MessageBus.extra_response_headers_lookup do |env|
|
||||
{
|
||||
"Access-Control-Allow-Origin" => Discourse.base_url_no_prefix,
|
||||
"Access-Control-Allow-Methods" => "GET, POST",
|
||||
"Access-Control-Allow-Headers" => "X-SILENCE-LOGGER, X-Shared-Session-Key"
|
||||
}
|
||||
end
|
||||
|
||||
MessageBus.user_id_lookup do |env|
|
||||
user = CurrentUser.lookup_from_env(env)
|
||||
user.id if user
|
||||
end
|
||||
|
||||
MessageBus.group_ids_lookup do |env|
|
||||
user = CurrentUser.lookup_from_env(env)
|
||||
user.groups.select('groups.id').map{|g| g.id} if user
|
||||
end
|
||||
|
||||
MessageBus.on_connect do |site_id|
|
||||
RailsMultisite::ConnectionManagement.establish_connection(db: site_id)
|
||||
end
|
||||
|
||||
MessageBus.on_disconnect do |site_id|
|
||||
ActiveRecord::Base.connection_handler.clear_active_connections!
|
||||
end
|
||||
|
||||
# Point at our redis
|
||||
MessageBus.redis_config = YAML.load(ERB.new(File.new("#{Rails.root}/config/redis.yml").read).result)[Rails.env].symbolize_keys
|
||||
|
||||
MessageBus.long_polling_enabled = SiteSetting.enable_long_polling
|
||||
MessageBus.long_polling_interval = SiteSetting.long_polling_interval
|
||||
|
||||
MessageBus.is_admin_lookup do |env|
|
||||
user = CurrentUser.lookup_from_env(env)
|
||||
if user && user.admin
|
||||
true
|
||||
else
|
||||
false
|
||||
end
|
||||
end
|
||||
|
||||
MessageBus.cache_assets = !Rails.env.development?
|
||||
MessageBus.enable_diagnostics
|
||||
end
|
||||
|
||||
def self.proxy_readonly(*methods)
|
||||
methods.each do |name|
|
||||
define_singleton_method(name) do |*args, &block|
|
||||
DiscourseRedis.ignore_readonly { MessageBus.send(name, *args, &block) }
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
proxy_readonly :publish,
|
||||
:subscribe,
|
||||
:unsubscribe,
|
||||
:last_id,
|
||||
:last_message,
|
||||
:backlog,
|
||||
:after_fork,
|
||||
:on_connect,
|
||||
:on_disconnect,
|
||||
:track_publish
|
||||
|
||||
end
|
|
@ -20,8 +20,8 @@ class DiscourseRedis
|
|||
"redis://#{(':' + config['password'] + '@') if config['password']}#{config['host']}:#{config['port']}/#{config['db']}"
|
||||
end
|
||||
|
||||
def initialize
|
||||
@config = DiscourseRedis.config
|
||||
def initialize(config=nil)
|
||||
@config = config || DiscourseRedis.config
|
||||
@redis = DiscourseRedis.raw_connection(@config)
|
||||
end
|
||||
|
||||
|
@ -34,10 +34,20 @@ class DiscourseRedis
|
|||
self.class.url(@config)
|
||||
end
|
||||
|
||||
def self.ignore_readonly
|
||||
yield
|
||||
rescue Redis::CommandError => ex
|
||||
if ex.message =~ /READONLY/
|
||||
STDERR.puts "WARN: Redis is in a readonly state. Performed a noop"
|
||||
else
|
||||
raise ex
|
||||
end
|
||||
end
|
||||
|
||||
# prefix the key with the namespace
|
||||
def method_missing(meth, *args, &block)
|
||||
if @redis.respond_to?(meth)
|
||||
@redis.send(meth, *args, &block)
|
||||
DiscourseRedis.ignore_readonly { @redis.send(meth, *args, &block) }
|
||||
else
|
||||
super
|
||||
end
|
||||
|
@ -54,28 +64,36 @@ class DiscourseRedis
|
|||
:zremrangebyscore, :zrevrange, :zrevrangebyscore, :zrevrank, :zrangebyscore].each do |m|
|
||||
define_method m do |*args|
|
||||
args[0] = "#{DiscourseRedis.namespace}:#{args[0]}"
|
||||
@redis.send(m, *args)
|
||||
DiscourseRedis.ignore_readonly { @redis.send(m, *args) }
|
||||
end
|
||||
end
|
||||
|
||||
def del(k)
|
||||
k = "#{DiscourseRedis.namespace}:#{k}"
|
||||
@redis.del k
|
||||
DiscourseRedis.ignore_readonly do
|
||||
k = "#{DiscourseRedis.namespace}:#{k}"
|
||||
@redis.del k
|
||||
end
|
||||
end
|
||||
|
||||
def keys(pattern=nil)
|
||||
len = DiscourseRedis.namespace.length + 1
|
||||
@redis.keys("#{DiscourseRedis.namespace}:#{pattern || '*'}").map{
|
||||
|k| k[len..-1]
|
||||
}
|
||||
DiscourseRedis.ignore_readonly do
|
||||
len = DiscourseRedis.namespace.length + 1
|
||||
@redis.keys("#{DiscourseRedis.namespace}:#{pattern || '*'}").map{
|
||||
|k| k[len..-1]
|
||||
}
|
||||
end
|
||||
end
|
||||
|
||||
def delete_prefixed(prefix)
|
||||
keys("#{prefix}*").each { |k| $redis.del(k) }
|
||||
DiscourseRedis.ignore_readonly do
|
||||
keys("#{prefix}*").each { |k| $redis.del(k) }
|
||||
end
|
||||
end
|
||||
|
||||
def flushdb
|
||||
keys.each{|k| del(k)}
|
||||
DiscourseRedis.ignore_readonly do
|
||||
keys.each{|k| del(k)}
|
||||
end
|
||||
end
|
||||
|
||||
def reconnect
|
||||
|
|
|
@ -51,7 +51,7 @@ class DistributedCache
|
|||
return if @subscribed
|
||||
@lock.synchronize do
|
||||
return if @subscribed
|
||||
MessageBus.subscribe(channel_name) do |message|
|
||||
DiscourseBus.subscribe(channel_name) do |message|
|
||||
@lock.synchronize do
|
||||
process_message(message)
|
||||
end
|
||||
|
@ -63,7 +63,7 @@ class DistributedCache
|
|||
def self.publish(hash, message)
|
||||
message[:origin] = hash.object_id
|
||||
message[:hash_key] = hash.key
|
||||
MessageBus.publish(channel_name, message, { user_ids: [-1] })
|
||||
DiscourseBus.publish(channel_name, message, { user_ids: [-1] })
|
||||
end
|
||||
|
||||
def self.set(hash, key, value)
|
||||
|
|
|
@ -3,4 +3,4 @@ class Arel::Table
|
|||
def hash
|
||||
@name.hash
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
|
@ -8,4 +8,4 @@ class ActiveRecord::Base
|
|||
def blank?
|
||||
false
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
|
@ -11,7 +11,7 @@ class MessageBusDiags
|
|||
end
|
||||
|
||||
def self.establish_peer_names
|
||||
MessageBus.publish "/server-name", {channel: "/server-name-reply/#{my_id}"}
|
||||
DiscourseBus.publish "/server-name", {channel: "/server-name-reply/#{my_id}"}
|
||||
end
|
||||
|
||||
def self.seen_hosts
|
||||
|
@ -20,12 +20,12 @@ class MessageBusDiags
|
|||
|
||||
unless @subscribed
|
||||
|
||||
MessageBus.subscribe "/server-name-reply/#{my_id}" do |msg|
|
||||
DiscourseBus.subscribe "/server-name-reply/#{my_id}" do |msg|
|
||||
MessageBusDiags.seen_host(msg.data)
|
||||
end
|
||||
|
||||
MessageBus.subscribe "/server-name" do |msg|
|
||||
MessageBus.publish msg.data["channel"], MessageBusDiags.my_id
|
||||
DiscourseBus.subscribe "/server-name" do |msg|
|
||||
DiscourseBus.publish msg.data["channel"], MessageBusDiags.my_id
|
||||
end
|
||||
@subscribed = true
|
||||
end
|
||||
|
|
|
@ -221,7 +221,7 @@ module SiteSettingExtension
|
|||
|
||||
def ensure_listen_for_changes
|
||||
unless @subscribed
|
||||
MessageBus.subscribe("/site_settings") do |message|
|
||||
DiscourseBus.subscribe("/site_settings") do |message|
|
||||
process_message(message)
|
||||
end
|
||||
@subscribed = true
|
||||
|
@ -233,10 +233,10 @@ module SiteSettingExtension
|
|||
if data["process"] != process_id
|
||||
begin
|
||||
@last_message_processed = message.global_id
|
||||
MessageBus.on_connect.call(message.site_id)
|
||||
DiscourseBus.on_connect.call(message.site_id)
|
||||
refresh!
|
||||
ensure
|
||||
MessageBus.on_disconnect.call(message.site_id)
|
||||
DiscourseBus.on_disconnect.call(message.site_id)
|
||||
end
|
||||
end
|
||||
end
|
||||
|
@ -294,7 +294,7 @@ module SiteSettingExtension
|
|||
end
|
||||
|
||||
def notify_changed!
|
||||
MessageBus.publish('/site_settings', {process: process_id})
|
||||
DiscourseBus.publish('/site_settings', {process: process_id})
|
||||
end
|
||||
|
||||
def has_setting?(name)
|
||||
|
|
|
@ -66,7 +66,7 @@ after_initialize do
|
|||
post.custom_fields["#{VOTES_CUSTOM_FIELD}-#{user_id}"] = votes
|
||||
post.save_custom_fields
|
||||
|
||||
MessageBus.publish("/polls/#{post_id}", { poll: poll })
|
||||
DiscourseBus.publish("/polls/#{post_id}", { poll: poll })
|
||||
|
||||
render json: { poll: poll, vote: options }
|
||||
end
|
||||
|
@ -100,7 +100,7 @@ after_initialize do
|
|||
post.custom_fields[POLLS_CUSTOM_FIELD] = polls
|
||||
post.save_custom_fields
|
||||
|
||||
MessageBus.publish("/polls/#{post_id}", { poll: polls[poll_name] })
|
||||
DiscourseBus.publish("/polls/#{post_id}", { poll: polls[poll_name] })
|
||||
|
||||
render json: { poll: polls[poll_name] }
|
||||
end
|
||||
|
|
|
@ -10,7 +10,7 @@ describe ::DiscoursePoll::PollsController do
|
|||
describe "#vote" do
|
||||
|
||||
it "works" do
|
||||
MessageBus.expects(:publish)
|
||||
DiscourseBus.expects(:publish)
|
||||
|
||||
xhr :put, :vote, { post_id: poll.id, poll_name: "poll", options: ["A"] }
|
||||
|
||||
|
@ -76,7 +76,7 @@ describe ::DiscoursePoll::PollsController do
|
|||
describe "#toggle_status" do
|
||||
|
||||
it "works for OP" do
|
||||
MessageBus.expects(:publish)
|
||||
DiscourseBus.expects(:publish)
|
||||
|
||||
xhr :put, :toggle_status, { post_id: poll.id, poll_name: "poll", status: "closed" }
|
||||
expect(response).to be_success
|
||||
|
@ -86,7 +86,7 @@ describe ::DiscoursePoll::PollsController do
|
|||
|
||||
it "works for staff" do
|
||||
log_in(:moderator)
|
||||
MessageBus.expects(:publish)
|
||||
DiscourseBus.expects(:publish)
|
||||
|
||||
xhr :put, :toggle_status, { post_id: poll.id, poll_name: "poll", status: "closed" }
|
||||
expect(response).to be_success
|
||||
|
|
|
@ -89,7 +89,7 @@ describe Discourse do
|
|||
|
||||
it "adds a key in redis and publish a message through the message bus" do
|
||||
$redis.expects(:set).with(Discourse.readonly_mode_key, 1)
|
||||
MessageBus.expects(:publish).with(Discourse.readonly_channel, true)
|
||||
DiscourseBus.expects(:publish).with(Discourse.readonly_channel, true)
|
||||
Discourse.enable_readonly_mode
|
||||
end
|
||||
|
||||
|
@ -99,7 +99,7 @@ describe Discourse do
|
|||
|
||||
it "removes a key from redis and publish a message through the message bus" do
|
||||
$redis.expects(:del).with(Discourse.readonly_mode_key)
|
||||
MessageBus.expects(:publish).with(Discourse.readonly_channel, false)
|
||||
DiscourseBus.expects(:publish).with(Discourse.readonly_channel, false)
|
||||
Discourse.disable_readonly_mode
|
||||
end
|
||||
|
||||
|
|
|
@ -72,7 +72,7 @@ describe PostCreator do
|
|||
|
||||
it "does not notify on system messages" do
|
||||
admin = Fabricate(:admin)
|
||||
messages = MessageBus.track_publish do
|
||||
messages = DiscourseBus.track_publish do
|
||||
p = PostCreator.create(admin, basic_topic_params.merge(post_type: Post.types[:moderator_action]))
|
||||
PostCreator.create(admin, basic_topic_params.merge(topic_id: p.topic_id, post_type: Post.types[:moderator_action]))
|
||||
end
|
||||
|
@ -93,7 +93,7 @@ describe PostCreator do
|
|||
created_post = nil
|
||||
reply = nil
|
||||
|
||||
messages = MessageBus.track_publish do
|
||||
messages = DiscourseBus.track_publish do
|
||||
created_post = PostCreator.new(admin, basic_topic_params.merge(category: cat.id)).create
|
||||
reply = PostCreator.new(admin, raw: "this is my test reply 123 testing", topic_id: created_post.topic_id).create
|
||||
end
|
||||
|
@ -118,7 +118,7 @@ describe PostCreator do
|
|||
it 'generates the correct messages for a normal topic' do
|
||||
|
||||
p = nil
|
||||
messages = MessageBus.track_publish do
|
||||
messages = DiscourseBus.track_publish do
|
||||
p = creator.create
|
||||
end
|
||||
|
||||
|
|
|
@ -1,3 +1,3 @@
|
|||
Fabricator(:api_key) do
|
||||
key '1dfb7d427400cb8ef18052fd412781af134cceca5725dd74f34bbc6b9e35ddc9'
|
||||
end
|
||||
end
|
||||
|
|
|
@ -2,4 +2,4 @@ Fabricator(:color_scheme) do
|
|||
name { sequence(:name) {|i| "Palette #{i}" } }
|
||||
enabled false
|
||||
color_scheme_colors(count: 2) { |attrs, i| Fabricate.build(:color_scheme_color, color_scheme: nil) }
|
||||
end
|
||||
end
|
||||
|
|
|
@ -2,4 +2,4 @@ Fabricator(:flag, from: :post_action) do
|
|||
user
|
||||
post
|
||||
post_action_type_id PostActionType.types[:spam]
|
||||
end
|
||||
end
|
||||
|
|
|
@ -2,4 +2,4 @@ Fabricator(:like, from: :post_action) do
|
|||
post
|
||||
user
|
||||
post_action_type_id PostActionType.types[:like]
|
||||
end
|
||||
end
|
||||
|
|
|
@ -1,3 +1,3 @@
|
|||
Fabricator(:permalink) do
|
||||
url { sequence(:url) {|i| "my/#{i}/url" } }
|
||||
end
|
||||
end
|
||||
|
|
|
@ -1,3 +1,3 @@
|
|||
Fabricator(:screened_ip_address) do
|
||||
ip_address { sequence(:ip_address) { |n| "123.#{(n*3)%255}.#{(n*2)%255}.#{n%255}" } }
|
||||
end
|
||||
end
|
||||
|
|
|
@ -573,7 +573,7 @@ describe Topic do
|
|||
describe "make_banner!" do
|
||||
|
||||
it "changes the topic archetype to 'banner'" do
|
||||
messages = MessageBus.track_publish do
|
||||
messages = DiscourseBus.track_publish do
|
||||
topic.make_banner!(user)
|
||||
expect(topic.archetype).to eq(Archetype.banner)
|
||||
end
|
||||
|
@ -597,7 +597,7 @@ describe Topic do
|
|||
|
||||
it "resets the topic archetype" do
|
||||
topic.expects(:add_moderator_post)
|
||||
MessageBus.expects(:publish).with("/site/banner", nil)
|
||||
DiscourseBus.expects(:publish).with("/site/banner", nil)
|
||||
topic.remove_banner!(user)
|
||||
expect(topic.archetype).to eq(Archetype.default)
|
||||
end
|
||||
|
|
Loading…
Reference in a new issue