dorkbox/Network
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Changed session ID -> ID for RandomIdAllocator.kt

Browse Source
This commit is contained in:
Robinson 2021-04-28 11:18:34 +02:00
parent 31f559accc
commit 32cd2822ee
1 changed files with 15 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
src/dorkbox/network/handshake/RandomIdAllocator.kt
View File

@ -20,10 +20,10 @@ import org.agrona.collections.IntHashSet
import java.security.SecureRandom import java.security.SecureRandom
/** /**
* An allocator for session IDs. * An allocator for random IDs.
* *
* The allocator randomly selects values from the given range `[min, max]` and will not return a previously-returned value `x` * The allocator randomly selects values from the given range `[min, max]` and will not return a previously-returned value `x`
* until `x` has been freed with `{ SessionAllocator#free(int)}. * until `x` has been freed with [free].
* *
* This implementation uses storage proportional to the number of currently-allocated * This implementation uses storage proportional to the number of currently-allocated
* values. Allocation time is bounded by { max - min}, will be { O(1)} * values. Allocation time is bounded by { max - min}, will be { O(1)}
@ -32,10 +32,10 @@ import java.security.SecureRandom
* *
* NOTE: THIS IS NOT THREAD SAFE! * NOTE: THIS IS NOT THREAD SAFE!
* *
* @param min The minimum session ID (inclusive) * @param min The minimum ID (inclusive)
* @param max The maximum session ID (exclusive) * @param max The maximum ID (exclusive)
*/ */
class RandomIdAllocator(private val min: Int, max: Int) { class RandomIdAllocator(private val min: Int = Integer.MIN_VALUE, max: Int = Integer.MAX_VALUE) {
private val used = IntHashSet() private val used = IntHashSet()
private val random = SecureRandom() private val random = SecureRandom()
private val maxAssignments: Int private val maxAssignments: Int
@ -50,15 +50,15 @@ class RandomIdAllocator(private val min: Int, max: Int) {
} }
/** /**
* Allocate a new session. Will never allocate session ID '0' * Allocate an unused ID. Will never allocate ID '0'
* *
* @return A new session ID * @return A new, unused ID
* *
* @throws AllocationException If there are no non-allocated sessions left * @throws AllocationException If there are no non-allocated IDs left
*/ */
fun allocate(): Int { fun allocate(): Int {
if (used.size == maxAssignments) { if (used.size == maxAssignments) {
throw AllocationException("No session IDs left to allocate") throw AllocationException("No IDs left to allocate")
} }
for (index in 0 until maxAssignments) { for (index in 0 until maxAssignments) {
@ -69,22 +69,21 @@ class RandomIdAllocator(private val min: Int, max: Int) {
} }
} }
throw AllocationException("Unable to allocate a session ID after $maxAssignments attempts (${used.size} values in use") throw AllocationException("Unable to allocate a ID after $maxAssignments attempts (${used.size} values in use")
} }
/** /**
* Free a session. After this method returns, `session` becomes eligible * Free an ID for use later. After this method returns, the ID becomes eligible for allocation by future calls to [allocate].
* for allocation by future calls to [.allocate].
* *
* @param session The session to free * @param id The ID to free
*/ */
fun free(session: Int) { fun free(id: Int) {
used.remove(session) used.remove(id)
} }
/** /**
* Removes all used sessions from the internal data structures * Removes all used IDs from the internal data structures
*/ */
fun clear() { fun clear() {
used.clear() used.clear()