dnrops
/
Schedule
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use nested type `BagKey.Gen`

This commit is contained in:
Quentin Jin 2019-04-18 10:11:41 +08:00
parent 913a699a6e
commit 66e6c54aa4
2 changed files with 16 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
Sources/Schedule/Bag.swift
View File

@ -3,29 +3,20 @@ import Foundation
/// A unique key for removing an element from a bag.
struct BagKey: Equatable {
fileprivate let i: UInt64
private let i: UInt64
fileprivate init(underlying: UInt64) {
self.i = underlying
}
}
/// A generator that can generate a sequence of unique `BagKey`.
///
/// let k1 = gen.next()
/// let k2 = gen.next()
/// ...
struct BagKeyGenerator: Sequence, IteratorProtocol {
typealias Element = BagKey
private var k = BagKey(underlying: 0)
/// Advances to the next key and returns it, or nil if no next key exists.
mutating func next() -> BagKey? {
if k.i == UInt64.max { return nil }
defer { k = BagKey(underlying: k.i + 1) }
return k
/// A generator that can generate a sequence of unique `BagKey`.
///
/// let k1 = gen.next()
/// let k2 = gen.next()
/// ...
struct Gen {
private var key = BagKey(i: 0)
init() { }
mutating func next() -> BagKey {
defer { key = BagKey(i: key.i + 1) }
return key
}
}
}
@ -44,13 +35,13 @@ struct Bag<Element> {
private typealias Entry = (key: BagKey, val: Element)
private var keyGen = BagKeyGenerator()
private var keyGen = BagKey.Gen()
private var entries: [Entry] = []
/// Appends a new element at the end of this bag.
@discardableResult
mutating func append(_ new: Element) -> BagKey {
let key = keyGen.next()!
let key = keyGen.next()
let entry = (key: key, val: new)
entries.append(entry)

2
Tests/ScheduleTests/BagTests.swift
View File

@ -6,7 +6,7 @@ final class BagTests: XCTestCase {
typealias Fn = () -> Int
func testBagKey() {
var g = BagKeyGenerator()
var g = BagKey.Gen()
let k1 = g.next()
let k2 = g.next()
XCTAssertNotNil(k1)