Some Refactoring

Author: devlights

cmd/trygolang/args.go

@@ -0,0 +1,22 @@
+package main
+
+import "flag"
+
+// Args は、プログラム引数の値を持つ構造体です１
+type Args struct {
+	OneTime   bool
+	ShowNames bool
+}
+
+// NewArgs は、Argsのコンストラクタ関数です
+func NewArgs() *Args {
+	return new(Args)
+}
+
+// Parse は、コマンドライン引数を解析しパッケージ変数に格納します
+func (a *Args) Parse() {
+	flag.BoolVar(&a.OneTime, "onetime", false, "run only one time")
+	flag.BoolVar(&a.ShowNames, "list", false, "show all example names")
+
+	flag.Parse()
+}

cmd/trygolang/main.go

@@ -2,7 +2,6 @@ package main
 
 import (
 	"bufio"
-	"flag"
 	"fmt"
 	"github.com/devlights/try-golang/lib"
 	"log"
@@ -11,9 +10,16 @@ import (
 	"strings"
 )
 
-var mapping = make(lib.SampleMapping)
+var (
+	args    *Args
+	mapping lib.SampleMapping
+)
 
 func init() {
+	args = NewArgs()
+	args.Parse()
+
+	mapping = lib.NewSampleMapping()
 	mapping.MakeMapping()
 }
 
@@ -34,15 +40,30 @@ func printAllExampleNames() {
 	}
 }
 
-func main() {
-	var (
-		onetime   = flag.Bool("onetime", false, "run only one time")
-		showNames = flag.Bool("list", false, "show all example names")
-	)
+func makeCandidates(userInput string) []string {
+	candidates := make([]string, 0, len(mapping))
+	for k := range mapping {
+		if strings.Contains(k, userInput) {
+			candidates = append(candidates, k)
+		}
+	}
+
+	return candidates
+}
+
+func exec(target string) error {
+	if v, ok := mapping[target]; ok {
+		fmt.Printf("[Name] %q\n", target)
+		if err := v(); err != nil {
+			return err
+		}
+	}
 
-	flag.Parse()
+	return nil
+}
 
-	if *showNames {
+func main() {
+	if args.ShowNames {
 		printAllExampleNames()
 		return
 	}
@@ -66,36 +87,24 @@ func main() {
 			break
 		}
 
-		candidates = make([]string, 0, len(mapping))
-		for k := range mapping {
-			if strings.Contains(k, userInput) {
-				candidates = append(candidates, k)
-			}
-		}
-
+		candidates = makeCandidates(userInput)
 		numberOfCandidate = len(candidates)
+
 		switch {
 		case numberOfCandidate == 0:
 			fmt.Printf("Not found...Try Again")
 			goto nextinput
 		case numberOfCandidate == 1:
 			userInput = candidates[0]
-			if v, ok := mapping[userInput]; ok {
-				fmt.Printf("[Name] %q\n", userInput)
-				if err := v(); err != nil {
-					log.Fatal(err)
-				}
+			if err := exec(userInput); err != nil {
+				log.Fatal(err)
 			}
 		case 1 < numberOfCandidate:
-			// 完全一致するものがあるか？
 			isPerfectMatchFound := false
 			for _, c := range candidates {
 				if c == userInput {
-					if v, ok := mapping[c]; ok {
-						fmt.Printf("[Name] %q\n", userInput)
-						if err := v(); err != nil {
-							log.Fatal(err)
-						}
+					if err := exec(userInput); err != nil {
+						log.Fatal(err)
 					}
 
 					isPerfectMatchFound = true
@@ -114,7 +123,7 @@ func main() {
 			}
 		}
 
-		if *onetime {
+		if args.OneTime {
 			break
 		}
 

lib/mapping.go

@@ -30,10 +30,15 @@ import (
 	"github.com/devlights/try-golang/basic/variables"
 )
 
-// サンプル名とサンプル呼び出し関数のマッピング定義の型
+// SampleMappingは、サンプル名とサンプル呼び出し関数のマッピング定義を持つ型です
 type SampleMapping map[string]func() error
 
-// マッピング生成
+// NewSampleMapping は、SampleMappingのコンストラクタ関数です
+func NewSampleMapping() SampleMapping {
+	return make(SampleMapping)
+}
+
+// MakeMapping は、マッピング生成します
 func (m SampleMapping) MakeMapping() {
 	m["error01"] = error_.Error01
 	m["helloworld"] = helloworld.HelloWorld