Cleaned up the directories

elitebot/lib/python3.11/site-packages/mysql/connector/fabric/balancing.py

@@ -0,0 +1,159 @@
+# MySQL Connector/Python - MySQL driver written in Python.
+# Copyright (c) 2013, 2014, Oracle and/or its affiliates. All rights reserved.
+
+# MySQL Connector/Python is licensed under the terms of the GPLv2
+# <http://www.gnu.org/licenses/old-licenses/gpl-2.0.html>, like most
+# MySQL Connectors. There are special exceptions to the terms and
+# conditions of the GPLv2 as it is applied to this software, see the
+# FOSS License Exception
+# <http://www.mysql.com/about/legal/licensing/foss-exception.html>.
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
+
+"""Implementing load balancing"""
+
+
+import decimal
+
+
+def _calc_ratio(part, whole):
+    """Calculate ratio
+
+    Returns int
+    """
+    return int((part/whole*100).quantize(
+        decimal.Decimal('1'), rounding=decimal.ROUND_HALF_DOWN))
+
+
+class BaseScheduling(object):
+
+    """Base class for all scheduling classes dealing with load balancing"""
+
+    def __init__(self):
+        """Initialize"""
+        self._members = []
+        self._ratios = []
+
+    def set_members(self, *args):
+        """Set members and ratios
+
+        This methods sets the members using the arguments passed. Each
+        argument must be a sequence where the second item is the weight.
+        The first element is an identifier. For example:
+
+            ('server1', 0.6), ('server2', 0.8)
+
+        Setting members means that the load will be reset. If the members
+        are the same as previously set, nothing will be reset or set.
+
+        If no arguments were given the members will be set to an empty
+        list.
+
+        Raises ValueError when weight can't be converted to a Decimal.
+        """
+        raise NotImplementedError
+
+    def get_next(self):
+        """Returns the next member"""
+        raise NotImplementedError
+
+    @property
+    def members(self):
+        """Returns the members of this loadbalancer"""
+        return self._members
+
+    @property
+    def ratios(self):
+        """Returns the ratios for all members"""
+        return self._ratios
+
+
+class WeightedRoundRobin(BaseScheduling):
+
+    """Class for doing Weighted Round Robin balancing"""
+
+    def __init__(self, *args):
+        """Initializing"""
+        super(WeightedRoundRobin, self).__init__()
+        self._load = []
+        self._next_member = 0
+        self._nr_members = 0
+
+        if args:
+            self.set_members(*args)
+
+    @property
+    def load(self):
+        """Returns the current load"""
+        return self._load
+
+    def set_members(self, *args):
+        if not args:
+            # Reset members if nothing was given
+            self._members = []
+            return
+        new_members = []
+        for member in args:
+            member = list(member)
+            try:
+                member[1] = decimal.Decimal(str(member[1]))
+            except decimal.InvalidOperation:
+                raise ValueError("Member '{member}' is invalid".format(
+                    member=member))
+            new_members.append(tuple(member))
+
+        new_members.sort(key=lambda x: x[1], reverse=True)
+        if self._members == new_members:
+            return
+        self._members = new_members
+        self._nr_members = len(new_members)
+
+        min_weight = min(i[1] for i in self._members)
+        self._ratios = []
+        for _, weight in self._members:
+            self._ratios.append(int(weight/min_weight * 100))
+        self.reset()
+
+    def reset(self):
+        """Reset the load"""
+        self._next_member = 0
+        self._load = [0] * self._nr_members
+
+    def get_next(self):
+        """Returns the next member"""
+        if self._ratios == self._load:
+            self.reset()
+
+        # Figure out the member to return
+        current = self._next_member
+
+        while self._load[current] == self._ratios[current]:
+            current = (current + 1) % self._nr_members
+
+        # Update the load and set next member
+        self._load[current] += 1
+        self._next_member = (current + 1) % self._nr_members
+
+        # Return current
+        return self._members[current]
+
+    def __repr__(self):
+        return "{class_}(load={load}, ratios={ratios})".format(
+            class_=self.__class__,
+            load=self.load,
+            ratios=self.ratios
+        )
+
+    def __eq__(self, other):
+        return self._members == other.members