proxysqlweightlifter 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4083bbfbafb2c33b3237ab1344346b3a8c38c2f1855a618226c6bf037ff008d8
-  data.tar.gz: 0d090098060d10351ca4f06b935ee6332848dd9412863be9072920000268a574
+  metadata.gz: d20c202dbb5e6fd963e4d5b58f52a4ac7766bfcd43009e79cbf8b10b0dc7272e
+  data.tar.gz: bb9faa6666c5babf550175365aae062fd11d9e5853596039cdd4a2fd7354a300
 SHA512:
-  metadata.gz: 03d8850273413d02f80c9b364dc9bb40f87ce5f4d439c134fdf0ebb4e1d85bca979d27f74e7f6c91e718ae9f8f4ea7f82a6ec9a208a0b3c7ce7d4bc1c01fd9a3
-  data.tar.gz: 260ec3a0299528edd9a8c121728464e6d3dddff2d0904bb29def9c4411bc85eb79f949c907c22fadc16bc51990d72cba26b9bebc28d27ed4be5acac99bbe314b
+  metadata.gz: f05d6b3a374739a7aa845f1c5b7ce778fb1dc8dd3e125b6a7e79d79133a83328c211002b844f5213a73127e9ac76e4a6f21c319348df4e694bd97836201a47e0
+  data.tar.gz: e762f7d8f221ce83d08215f391952dc494489ea1fad5136d78033b793bd012ab2cfd7dd965d88c84b5b9eb225d266b84a27d07a2b6e357aa51c881984e43f246

data/lib/proxysqlweightlifter.rb

@@ -1,12 +1,12 @@
 require 'mysql2'
-require "proxysqlweightlifter/version"
-require "proxysqlweightlifter/lift"
+require 'proxysqlweightlifter/version'
+require 'proxysqlweightlifter/lift'
 
 module Proxysqlweightlifter
   class Error < StandardError; end
 
   def self.run(argv)
-    host, port, user, pass, weight = argv
+    host, port, user, pass, weight, custom_weight = argv
     Lift.new(
       Mysql2::Client.new(
         host: host,
@@ -15,7 +15,8 @@ module Proxysqlweightlifter
         password: pass,
         database: 'main'
       ),
-      weight || 9
+      weight || 9,
+      custom_weight
     ).run
   end
 end

data/lib/proxysqlweightlifter/lift.rb

@@ -2,11 +2,12 @@ module Proxysqlweightlifter
   # Lift is responsible for resolving current slaves from
   # mysql_replication_hostgroups table
   class Lift
-    attr_reader :db, :weight
+    attr_reader :db, :default_weight
 
-    def initialize(database, weight = 10)
+    def initialize(database, default_weight = 10, custom_weight = '')
       @db = database
-      @weight = weight
+      @default_weight = default_weight
+      @custom_weight = custom_weight.to_s
     end
 
     def run
@@ -30,6 +31,16 @@ module Proxysqlweightlifter
       end
     end
 
+    def hostgroup_weights
+      @hostgroup_weights ||= @custom_weight.split(',').map do |cs|
+        hostgroup_id, group_id, weight = cs.split(':')
+        raise 'Invalid or unsupported format: hostgroup_id:9100:70' if hostgroup_id != 'hostgroup_id'
+        {
+          group_id.to_i => weight.to_i
+        }
+      end.reduce(&:merge) || {}
+    end
+
     def hostgroups
       db.query(%(
         SELECT writer_hostgroup,reader_hostgroup
@@ -69,9 +80,10 @@ module Proxysqlweightlifter
     end
 
     def update_weight(hostname:, port:, hostgroup_id:)
+      weight = hostgroup_weights[hostgroup_id] || default_weight || 1
       db.query(%(
         UPDATE mysql_servers
-        SET weight = #{weight || 1}
+        SET weight = #{weight}
         WHERE hostname = "#{hostname}" AND port = "#{port}" AND hostgroup_id = "#{hostgroup_id}";
       ))
     end

data/lib/proxysqlweightlifter/version.rb

@@ -1,3 +1,3 @@
 module Proxysqlweightlifter
-  VERSION = "0.1.0"
+  VERSION = '0.2.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: proxysqlweightlifter
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Ernestas Poskus
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-11-12 00:00:00.000000000 Z
+date: 2019-01-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: mysql2