Stream API 를 활용한 로또 번호 생성기 코드 리팩터링

[corock]

다음 제시하는 코드 블록은 스트림을 이용해서 외부 반복을 내부 반복으로 바꾸는 활용 편입니다.

LotteryNumberGeneratorShell.java

public class LotteryNumberGeneratorShell {

    public static void main(String[] args) {
        LotteryNumberGenerator lottoNumber = new LotteryNumberGenerator();
        lottoNumber.draw();
        lottoNumber.print();

        // TODO: 스트림을 이용하여 1 ~ 45 사이의 로또 번호 6개를 중복 없이 오름차순으로 추출하세요.

    }

}

LotteryNumberGenerator.java

public class LotteryNumberGenerator {

    private static final int NUM_OF_LOTTO = 6;
    private static final int MAX_LOTTO = 45;

    int[] lottoNumbers = new int[NUM_OF_LOTTO];
    static int[] countArray = new int[MAX_LOTTO + 1];
    Random random = new Random();

    public void draw() {
        int count = 0;
        while (count < 6) {
            int drawnNumber = random.nextInt(45) + 1;
            if (hasDuplication(drawnNumber)) {
                count -= 1;
                continue;
            }
            lottoNumbers[count++] = drawnNumber;
            countArray[drawnNumber] += 1;
        }
        lottoNumbers = countingSort(lottoNumbers);
    }

    public boolean hasDuplication(int drawnNumber) {
        return countArray[drawnNumber] > 0;
    }

    public void print() {
        for (int lottoNumber : lottoNumbers) {
            System.out.print(lottoNumber + " ");
        }
    }

    public static int[] countingSort(int[] numbers) {
        int[] sortedNumbers = new int[numbers.length];
        int i = 0;
        for (int j = 0; j < countArray.length; j++) {
            if (countArray[j] > 0) {
                sortedNumbers[i++] = j;
            }
        }
        return sortedNumbers;
    }

}

[corock]

스트림을 사용해 본 샘플 코드입니다.

LotteryNumberGeneratorShell.java

public class LotteryNumberGeneratorShell {

    public static void main(String[] args) {
        // TODO: 스트림을 이용하여 1 ~ 45 사이의 로또 번호 6개를 중복 없이 오름차순으로 추출하세요.
        LotteryNumberGenerator.generate1().forEach(System.out::println);
        LotteryNumberGenerator.generate2().forEach(System.out::println);
    }

}

LotteryNumberGenerator.java

public final class LotteryNumberGenerator {

    public static List<Integer> generate1() {
        return Stream.generate(() -> new SecureRandom().nextInt(45) + 1)
                .distinct()
                .limit(6)
                .sorted()
                .collect(Collectors.toList());
    }

    public static List<Integer> generate2() {
        return new SecureRandom().ints(1, 46)
                .distinct()
                .limit(6)
                .sorted()
                .boxed()
                .collect(Collectors.toList());
    }

    public static List<Integer> generate3() {
        // STUB: Anything else?
        return null;
    }

}