From 1e85d14a48d17eedc4afe699878142be955f7211 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=A0=B4=E5=A8=83=E9=85=B1?= <mmgac001@gmail.com>
Date: Sun, 30 Apr 2017 21:26:36 +0800
Subject: [PATCH] optimize

---
 db_transfer.py          | 2 +-
 shadowsocks/udprelay.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/db_transfer.py b/db_transfer.py
index 4bb7ba6..0f3a7c9 100644
--- a/db_transfer.py
+++ b/db_transfer.py
@@ -88,6 +88,7 @@ class TransferBase(object):
 		new_servers = {}
 		allow_users = {}
 		mu_servers  = {}
+		config = shell.get_config(False)
 		for row in rows:
 			try:
 				allow = switchrule.isTurnOn(row) and row['enable'] == 1 and row['u'] + row['d'] < row['transfer_enable']
@@ -152,7 +153,6 @@ class TransferBase(object):
 				else:
 					self.new_server(port, passwd, cfg)
 			else:
-				config = shell.get_config(False)
 				if ServerPool.get_instance().server_is_run(port) > 0:
 					if config['additional_ports_only'] or not allow:
 						logging.info('db stop server at port [%s]' % (port,))
diff --git a/shadowsocks/udprelay.py b/shadowsocks/udprelay.py
index fa9f404..f117781 100644
--- a/shadowsocks/udprelay.py
+++ b/shadowsocks/udprelay.py
@@ -506,11 +506,11 @@ class UDPRelay(object):
         try:
             #logging.info('UDP handle_server sendto %s:%d %d bytes' % (common.to_str(server_addr), server_port, len(data)))
             client.sendto(data, (server_addr, server_port))
+            self.add_transfer_u(client_uid, len(data))
             if client_pair is None: # new request
                 addr, port = client.getsockname()[:2]
                 common.connect_log('UDP data to %s:%d from %s:%d by UID %d' %
                         (common.to_str(server_addr), server_port, addr, port, user_id))
-            self.add_transfer_u(client_uid, len(data))
         except IOError as e:
             err = eventloop.errno_from_exception(e)
             if err in (errno.EINPROGRESS, errno.EAGAIN):