swift Swift 2 - FizzBu​​zz Generator

Posted 2021-05-19

struct FizzBuzzGenerator: SequenceType {
    private let bounds: Range<Int>

    func generate() -> AnyGenerator<String> {
        var current = bounds.startIndex

        return anyGenerator {
            guard current < self.bounds.endIndex else {
                return nil
            }

            let output: String
            switch (current % 3, current % 5) {
            case (0, 0):
                output = "FizzBuzz"
            case (0, _):
                output = "Fizz"
            case (_, 0):
                output = "Buzz"
            default:
                output = String(current)
            }

            current++
            return output
        }
    }
}

let fizzBuzzSequence = FizzBuzzGenerator(bounds: 1...100).lazy
for x in fizzBuzzSequence {
    print(x)
}