함수

반복되는 코드를 묶어서 실행하는 코드 블럭의 단위입니다.
func 키워드를 사용하여 함수를 정의합니다.
파라미터명 뒤에 파라미터의 자료형을 정의 합니다.
return type은 함수명 뒤에 적습니다.

소수 판별 프로그램

소수 판별 함수

func prime(number int){
	p, err := isPrime(number)
	if err != nil {
		fmt.Println(err)
		os.Exit(0)
	}

	if p {
		fmt.Println(number ,"는(은) 소수입니다!")
	}else {
		fmt.Println(number ,"는(은) 소수가 아닙니다~")
	}
}

func isPrime(n int) (bool, error) {
	if n < 2 {
		return false, fmt.Errorf("%d는(은) 소수가 아닙니다~", n)
	}

	for i := 2; i < n; i++ {
		if n%i == 0 {
			return false,nil
		}
	}
	return true, nil // true 리턴이면 소수, false 소수 X
}

구간별 소수 판별 함수

func primeRange(a int, b int){
	if a>b {
		temp := a
		a =b
		b =temp
	}
	for i := a; i<=b; i++{
		p, err := isPrime(i)
		if err != nil {
			fmt.Println(err)
			os.Exit(0)
		}
		if p{
			fmt.Print(i, " ")
		}else {
			fmt.Println( "는(은) 소수가 아닙니다~")
		}
	}
}

main 함수

func main() {
	var menu int;

	for true{
		fmt.Print("MENU : 1) 소수판정 2) 구간 소수판정 : ")
		_, err := fmt.Scanln(&menu)

		if  err != nil{
			log.Fatal(err)
		}

		switch menu{
			case 1:
				var in int;
				fmt.Print("정수입력 : ")
				_, err := fmt.Scanln(&in)

				if  err != nil{
					log.Fatal(err)
				}
				prime(in)
			case 2:
				var n1, n2 int;
				fmt.Print("정수 2개 입력 : ")
				_, err := fmt.Scanln(&n1, &n2)

				if err != nil{
					log.Fatal(err)
				}
				primeRange(n1, n2)
			default:
				fmt.Print("프로그램을 종료합니다")
				os.Exit(0)
		}

	}
}