快捷導(dǎo)航

python負(fù)載均衡的簡(jiǎn)單實(shí)現(xiàn)方法

更新時(shí)間：2018年02月04日 09:19:18 作者：熔遁丶螺旋手里劍

這篇文章給大家介紹用python實(shí)現(xiàn)最簡(jiǎn)單的負(fù)載均衡方法，即將請(qǐng)求發(fā)送到未宕機(jī)的服務(wù)器上，感興趣的朋友一起看看吧

提到分發(fā)請(qǐng)求，相信大多數(shù)人首先會(huì)想到Nginx，Nginx作為一種多功能服務(wù)器，不僅提供了反向代理隱藏主機(jī)ip的能力，還擁有簡(jiǎn)單的緩存加速功能。當(dāng)然Nginx最強(qiáng)大的功能還是分發(fā)請(qǐng)求，不僅提供了哈希，一致性哈希，負(fù)載均衡等多種請(qǐng)求分發(fā)模式，還保證了自己服務(wù)的輕量和穩(wěn)定。一臺(tái)Nginx服務(wù)器常年工作在高并發(fā)請(qǐng)求的環(huán)境下，也極少宕機(jī)。

在Nginx負(fù)載均衡模式下，請(qǐng)求會(huì)發(fā)送到壓力最小的未宕機(jī)服務(wù)器上。今天我們不考慮目標(biāo)服務(wù)器的壓力，用python實(shí)現(xiàn)最簡(jiǎn)單的負(fù)載均衡方法，即將請(qǐng)求發(fā)送到未宕機(jī)的服務(wù)器上。

我們想調(diào)用module_b模塊中的接口，module_b服務(wù)在10.10.10.115服務(wù)器上的10081，10082，10083，10084這4個(gè)端口上。

#!/usr/bin/python
# -*- coding: utf-8 -*-
import requests
import random
import os
import sys
import time
import ConnectionError
import Module_bException
module_b = "10.10.10.115:10081,10.10.10.115:10082,10.10.10.115:10083,10.10.10.115:10084"
class Module_b():
  def __init__(self):
    self.url_prefix = [val.strip() for val in module_b.split(',')]
  def _request(self, short_uri, payload):
    res = None
    try_count = 1
    url_prefixs = self.url_prefix[:]
    url_prefixs.sort(key=lambda f: random.randint(0, 100))
　　　　　　
    for curr_url_prefix in url_prefixs:
      url = os.path.join(curr_url_prefix, short_uri)
      try:
        res = requests.post(url, data=payload)
        break
      except ConnectionError as e:
        try_count += 1
        sys.stderr.write('can not connect to Module_b, retry ...\n')
        time.sleep(1)
        if try_count == len(url_prefixs):
          raise e
    if res.status_code != 200:
      raise Module_bException('HTTP ERROR: %s' % res.text)
    result = res.json()
    if result['status'] != '0':
      raise Module_bException(result['errstr'])
    return result['result']

ConnecttionError和Module_bException為封裝好的報(bào)錯(cuò)類無(wú)需在意。

整個(gè)負(fù)載均衡的實(shí)現(xiàn)也很簡(jiǎn)單，傳入api和參數(shù)，然后從所有的module_b地址中隨機(jī)選出一個(gè)，拼接成完整的requests請(qǐng)求，如果無(wú)法訪問(wèn)到module_b服務(wù)，那么將換到另一個(gè)未訪問(wèn)過(guò)的module_b服務(wù)地址，直到訪問(wèn)過(guò)全部的module_b服務(wù)。

總結(jié)

以上所述是小編給大家介紹的python負(fù)載均衡的簡(jiǎn)單實(shí)現(xiàn)方法，希望對(duì)大家有所幫助，如果大家有任何疑問(wèn)歡迎給我留言，小編會(huì)及時(shí)回復(fù)大家的！