Add a makara flag to enable usage with makara to hit the primary

basecamp · May 10, 2022 · 57f4be5 · 57f4be5
1 parent d673384
commit 57f4be5
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/lib/easymon/checks/active_record_mysql_writeable_check.rb b/lib/easymon/checks/active_record_mysql_writeable_check.rb
@@ -2,8 +2,11 @@ module Easymon
   class ActiveRecordMysqlWriteableCheck
     attr_accessor :klass
 
-    def initialize(klass)
+    def initialize(klass, makara = false)
       self.klass = klass
+      @query = "SELECT @@read_only"
+      # Trick makara into using the primary db
+      @query += " for UPDATE" if makara
     end
 
     def check
@@ -18,7 +21,7 @@ def check
 
     private
       def database_writeable?
-        klass.connection.execute("SELECT @@read_only").entries.flatten.first == 0
+        klass.connection.execute(@query).entries.flatten.first == 0
       rescue
         false
       end