[RUST/COMMON] Iterator 트레잇 : 소수를 구하는 반복자 만들기

■ Iterator 트레잇을 구현해 소수를 구하는 반복자를 만드는 방법을 보여준다.

▶ 예제 코드 (RS)


struct PrimeIterator
{
    n : u8
}

impl PrimeIterator
{
    fn new() -> Self
    {
        return PrimeIterator { n : 1 };
    }

    fn is_prime(&self) -> bool
    {
        for i in 2..self.n
        {
            if self.n % i == 0
            {
                return false;
            }
        }

        return true;
    }
}

impl Iterator for PrimeIterator
{
    type Item = u8;

    fn next(&mut self) -> Option<Self::Item>
    {
        loop
        {
            self.n += 1;

            if std::u8::MAX == self.n
            {
                return None
            }

            if self.is_prime()
            {
                return Some(self.n);
            }
        }
    }
}

fn main()
{
    let prime_iterator : PrimeIterator = PrimeIterator::new();

    for n in prime_iterator
    {
        print!("{},", n);
    }
}

struct PrimeIterator

{

n : u8

}

impl PrimeIterator

{

fn new() -> Self

{

return PrimeIterator { n : 1 };

}

fn is_prime(&self) -> bool

{

for i in 2..self.n

{

if self.n % i == 0

{

return false;

}

return true;

}

impl Iterator for PrimeIterator

{

type Item = u8;

fn next(&mut self) -> Option<Self::Item>

{

loop

{

self.n += 1;

if std::u8::MAX == self.n

{

return None

}

if self.is_prime()

{

return Some(self.n);

}

fn main()

{

let prime_iterator : PrimeIterator = PrimeIterator::new();

for n in prime_iterator

{

print!("{},", n);

}