command line arguments 를 받는 코드 추가

Author: gitdog01

main.go

@@ -2,13 +2,32 @@ package main
 
 import (
 	"fmt"
+	"os"
 	"module/monolib"
 )
 
 func main(){
-	fmt.Println("Hello World")
-	err := monolib.TestListen()
-	if err != nil {
-		fmt.Println(err)
+
+	// args를 가져옵니다.
+	args := os.Args
+	var listenerFlag bool = true
+
+	// args가 2개보다 크면 에러를 출력하고 종료합니다.
+	if len(args) > 2 {
+		fmt.Println("Usage: go run main.go -c")
+		return
+	}
+
+	if args[1] == "-c" {
+		listenerFlag = false
+	}	
+
+	if listenerFlag {
+		// 리스너 모드
+		err := monolib.RunListen()
+		if err != nil {
+			fmt.Println(err)
+		}
+		return
 	}
 }

monolib/listen.go

@@ -6,7 +6,7 @@ import (
 	"fmt"
 )
 
-func TestListen() error {
+func RunListen() error {
 	// listener 와 err 를 리턴 받습니다.
 	// listener 는 net.Listener 인터페이스를 구현한 객체입니다.
 	// err 는 error 인터페이스를 구현한 객체입니다.