Use ChanKeep system for joining channels with joinSingle
This commit is contained in:
parent
9780a2dfc8
commit
7415ca5556
|
@ -138,8 +138,9 @@ def emptyChanAllocate(net, flist):
|
||||||
allocated = {}
|
allocated = {}
|
||||||
|
|
||||||
newlist = list(flist)
|
newlist = list(flist)
|
||||||
sum_free = sum(chanfree.values()) # 250
|
chan_slots_used = getTotalChans(net)
|
||||||
trunc_list = newlist[:sum_free]
|
max_chans = getSumChanlimit(net) - chan_slots_used
|
||||||
|
trunc_list = newlist[:max_chans]
|
||||||
debug(f"emptyChanAllocate() {net}: newlist:{len(newlist)} trunc_list:{len(trunc_list)}")
|
debug(f"emptyChanAllocate() {net}: newlist:{len(newlist)} trunc_list:{len(trunc_list)}")
|
||||||
|
|
||||||
for i in chanfree.keys():
|
for i in chanfree.keys():
|
||||||
|
@ -162,6 +163,7 @@ def populateChans(net, clist):
|
||||||
:param new: list of newly provisioned relays to account for"""
|
:param new: list of newly provisioned relays to account for"""
|
||||||
# divided = array_split(clist, relay)
|
# divided = array_split(clist, relay)
|
||||||
allocated = emptyChanAllocate(net, clist)
|
allocated = emptyChanAllocate(net, clist)
|
||||||
|
trace(f"populateChans() allocated:{allocated}")
|
||||||
if not allocated:
|
if not allocated:
|
||||||
return
|
return
|
||||||
for i in allocated.keys():
|
for i in allocated.keys():
|
||||||
|
@ -169,6 +171,7 @@ def populateChans(net, clist):
|
||||||
main.TempChan[net][i] = allocated[i]
|
main.TempChan[net][i] = allocated[i]
|
||||||
else:
|
else:
|
||||||
main.TempChan[net] = {i: allocated[i]}
|
main.TempChan[net] = {i: allocated[i]}
|
||||||
|
trace(f"populateChans() TempChan {net}{i}: {allocated[i]}")
|
||||||
|
|
||||||
|
|
||||||
def notifyJoin(net):
|
def notifyJoin(net):
|
||||||
|
@ -180,10 +183,11 @@ def notifyJoin(net):
|
||||||
for i in getActiveRelays(net):
|
for i in getActiveRelays(net):
|
||||||
name = net + str(i)
|
name = net + str(i)
|
||||||
if name in main.IRCPool.keys():
|
if name in main.IRCPool.keys():
|
||||||
|
trace(f"notifyJoin() {name}")
|
||||||
main.IRCPool[name].checkChannels()
|
main.IRCPool[name].checkChannels()
|
||||||
|
|
||||||
|
|
||||||
def minifyChans(net, listinfo):
|
def minifyChans(net, listinfo, as_list=False):
|
||||||
"""
|
"""
|
||||||
Remove channels from listinfo that are already covered by a relay.
|
Remove channels from listinfo that are already covered by a relay.
|
||||||
:param net: network
|
:param net: network
|
||||||
|
@ -192,19 +196,32 @@ def minifyChans(net, listinfo):
|
||||||
:return: list of channels with joined channels removed
|
:return: list of channels with joined channels removed
|
||||||
:rtype: list of [channel, num_users]
|
:rtype: list of [channel, num_users]
|
||||||
"""
|
"""
|
||||||
|
# We want to make this reusable for joining a bunch of channels.
|
||||||
|
if as_list:
|
||||||
|
channel_list = listinfo
|
||||||
|
|
||||||
if not allRelaysActive(net):
|
if not allRelaysActive(net):
|
||||||
error("All relays for %s are not active, cannot minify list" % net)
|
error("All relays for %s are not active, cannot minify list" % net)
|
||||||
return False
|
return False
|
||||||
for i in getActiveRelays(net):
|
for i in getActiveRelays(net):
|
||||||
name = net + str(i)
|
name = net + str(i)
|
||||||
for x in main.IRCPool[name].channels:
|
for x in main.IRCPool[name].channels:
|
||||||
for y in listinfo:
|
if as_list:
|
||||||
if y[0] == x:
|
for y in channel_list:
|
||||||
listinfo.remove(y)
|
if y == x:
|
||||||
if not listinfo:
|
channel_list.remove(y)
|
||||||
log("We're on all the channels we want to be on, dropping LIST")
|
else:
|
||||||
return False
|
for y in listinfo:
|
||||||
return listinfo
|
if y[0] == x:
|
||||||
|
listinfo.remove(y)
|
||||||
|
if not as_list:
|
||||||
|
if not listinfo:
|
||||||
|
log("We're on all the channels we want to be on, dropping LIST")
|
||||||
|
return False
|
||||||
|
if as_list:
|
||||||
|
return channel_list
|
||||||
|
else:
|
||||||
|
return listinfo
|
||||||
|
|
||||||
|
|
||||||
def keepChannels(net, listinfo, mean, sigrelay, relay):
|
def keepChannels(net, listinfo, mean, sigrelay, relay):
|
||||||
|
@ -278,19 +295,13 @@ def joinSingle(net, channel):
|
||||||
"""
|
"""
|
||||||
if "," in channel:
|
if "," in channel:
|
||||||
channels = channel.split(",")
|
channels = channel.split(",")
|
||||||
eca = emptyChanAllocate(net, channels)
|
channels = minifyChans(net, channels, as_list=True)
|
||||||
|
|
||||||
else:
|
else:
|
||||||
eca = emptyChanAllocate(net, [channel])
|
channels = [channel]
|
||||||
if not eca:
|
populateChans(net, channels)
|
||||||
return False
|
notifyJoin(net)
|
||||||
if not len(eca.keys()) == 1:
|
return True
|
||||||
return False
|
|
||||||
num = list(eca.keys())[0]
|
|
||||||
name = f"{net}{num}"
|
|
||||||
if name not in main.IRCPool:
|
|
||||||
return False
|
|
||||||
main.IRCPool[name].join(channel)
|
|
||||||
return num
|
|
||||||
|
|
||||||
|
|
||||||
def partSingle(net, channel):
|
def partSingle(net, channel):
|
||||||
|
|
Loading…
Reference in New Issue