0x2620/openmedialibrary
1
0
Fork 0
Code Issues 131 Pull requests Releases Wiki Activity

accept peering requests based on prefs

Browse source
This commit is contained in:
j 2016-01-17 18:42:56 +05:30
commit 954491759e
4 changed files with 27 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

31
oml/nodes.py
View file

@ -61,9 +61,8 @@ class Node(Thread):
action = self._q.get()
if not self._running:
break
if action == 'go_online':
if not self.online:
self._go_online()
if action == 'send_response':
self._send_response()
elif action == 'ping':
self.online = self.can_connect()
elif action == 'pull':
@ -88,9 +87,6 @@ class Node(Thread):
if state.online:
self._q.put('ping')
def go_online(self):
self._q.put('go_online')
@property
def url(self):
url = None
@ -234,11 +230,14 @@ class Node(Thread):
def is_online(self):
return self.online or self.get_local() != None
def _go_online(self):
def send_response(self):
self._q.put('send_response')
def _send_response(self):
with db.session():
u = user.models.User.get_or_create(self.user_id)
if u.peered or u.queued:
logger.debug('go_online peered=%s queued=%s %s (%s)', u.peered, u.queued, u.id, u.nickname)
logger.debug('go online peered=%s queued=%s %s (%s)', u.peered, u.queued, u.id, u.nickname)
try:
self.online = self.can_connect()
if self.online:
@ -255,8 +254,6 @@ class Node(Thread):
except:
logger.debug('failed to connect to %s', self.user_id)
self.online = False
else:
self.online = False
def trigger_status(self):
if self.online is not None:
@ -299,8 +296,6 @@ class Node(Thread):
logger.debug('peering failed? %s %s', action, r)
if action in ('cancelPeering', 'rejectPeering', 'removePeering'):
self.online = False
else:
self.go_online()
trigger_event('peering.%s'%action.replace('Peering', ''), u.json())
return True
@ -430,7 +425,7 @@ class Nodes(Thread):
self.queue('add', u.id)
for u in user.models.User.query.filter_by(queued=True):
logger.debug('adding queued node... %s', u.id)
self.queue('add', u.id)
self.queue('add', u.id, True)
self._local = LocalNodes()
self._cleanup = PeriodicCallback(lambda: self.queue('cleanup'), 120000)
self._cleanup.start()
@ -459,12 +454,14 @@ class Nodes(Thread):
elif target == 'online':
nodes = [n for n in list(self._nodes.values()) if n.online]
else:
if not target in self._nodes:
self._add(target)
nodes = [self._nodes[target]]
for node in nodes:
r = getattr(node, action)(*args)
logger.debug('call node api %s->%s%s = %s', node.user_id, action, args, r)
def _add(self, user_id):
def _add(self, user_id, send_response=False):
if user_id not in self._nodes:
from user.models import User
with db.session():
@ -472,6 +469,8 @@ class Nodes(Thread):
else:
if not self._nodes[user_id].online:
self._nodes[user_id].ping()
if send_response:
self._nodes[user_id].send_response()
def run(self):
while self._running:
@ -480,7 +479,7 @@ class Nodes(Thread):
if args[0] == 'cleanup':
self.cleanup()
elif args[0] == 'add':
self._add(args[1])
self._add(*args[1:])
else:
self._call(*args)
@ -523,4 +522,4 @@ def check_nodes():
for u in user.models.User.query.filter_by(queued=True):
if not state.nodes.is_online(u.id):
logger.debug('queued peering message for %s trying to connect...', u.id)
state.nodes.queue('add', u.id)
state.nodes.queue('add', u.id, True)