cyberes
/
proxy-loadbalancer
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 17 Wiki Activity

add option to bypass smartproxy via the Smartproxy-Bypass header

This commit is contained in:
Cyberes 2024-03-20 17:17:56 -06:00
parent 8c0e80efa8
commit c776c805e7
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
proxy-skeleton/app/plugins/load_balancer.py
View File

@ -65,7 +65,7 @@ class ProxyLoadBalancer(TcpUpstreamConnectionHandler, HttpProxyBasePlugin):
pass pass
# Select the proxy to use. # Select the proxy to use.
self._endpoint = self._select_proxy(request.host.decode()) self._endpoint = self._select_proxy(request.host.decode(), request.has_header(b'smartproxy-bypass'))
# If chosen proxy is the local instance, bypass upstream proxies # If chosen proxy is the local instance, bypass upstream proxies
assert self._endpoint.port and self._endpoint.hostname assert self._endpoint.port and self._endpoint.hostname
@ -202,13 +202,13 @@ class ProxyLoadBalancer(TcpUpstreamConnectionHandler, HttpProxyBasePlugin):
log_attrs[attr] = value.decode('utf-8') log_attrs[attr] = value.decode('utf-8')
logger.info(access_log_format.format_map(log_attrs)) logger.info(access_log_format.format_map(log_attrs))
def _select_proxy(self, request_host: str = None) -> Url: def _select_proxy(self, request_host: str = None, smartproxy_bypass: bool = True) -> Url:
online = int(self.redis.get('balancer_online')) online = int(self.redis.get('balancer_online'))
if not online: if not online:
logger.error('Server is not online!') logger.error('Server is not online!')
return Url() return Url()
if request_host in BYPASS_SMARTPROXY_DOMAINS: if request_host in BYPASS_SMARTPROXY_DOMAINS or smartproxy_bypass:
valid_backends = redis_cycle('our_proxy_backends') valid_backends = redis_cycle('our_proxy_backends')
else: else:
valid_backends = redis_cycle('all_proxy_backends') valid_backends = redis_cycle('all_proxy_backends')