burmatovan/openuds
1
0
Fork 0
forked from shaba/openuds
Code Issues Pull Requests Projects Releases Wiki Activity

Fixed assignement of new services if pool is at 100% usage

Browse Source
This commit is contained in:
Adolfo Gómez García 2021-11-30 12:18:04 +01:00
parent bde63f7b4f
commit 250ade6aee
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
server/src/uds/core/managers/user_service.py
View File

@ -939,14 +939,21 @@ class UserServiceManager(metaclass=singleton.Singleton):
] # Just shuffle them
# Sort pools related to policy now, and xtract only pools, not sort keys
# split resuult in two lists, 100% full and not 100% full
# Remove "full" pools (100%) from result and pools in maintenance mode, not ready pools, etc...
sortedPools = sorted(sortPools, key=lambda x: x[0])
pools: typing.List[ServicePool] = [
p[1]
for p in sorted(sortPools, key=lambda x: x[0])
for p in sortedPools
if p[1].usage() < 100 and p[1].isUsable()
]
poolsFull: typing.List[ServicePool] = [
p[1]
for p in sortedPools
if p[1].usage() == 100 and p[1].isUsable()
]
logger.debug('Pools: %s', pools)
logger.debug('Pools: %s/%s', pools, poolsFull)
usable: typing.Optional[typing.Tuple[ServicePool, Transport]] = None
# Now, Lets find first if there is one assigned in ANY pool
@ -975,8 +982,9 @@ class UserServiceManager(metaclass=singleton.Singleton):
return found
try:
# Already assigned should look for in all usable pools, not only "non-full" ones
alreadyAssigned: UserService = UserService.objects.filter(
deployed_service__in=pools,
deployed_service__in=pools+poolsFull,
state__in=State.VALID_STATES,
user=user,
cache_level=0,