diff --git a/triangle-procedure.go b/triangle-procedure.go index 663485e..b607198 100644 --- a/triangle-procedure.go +++ b/triangle-procedure.go @@ -7,6 +7,10 @@ import ( ) func main() { + if validateArgs() == false { + fmt.Println("Required 3 arguments") + return + } a, _ := strconv.Atoi(os.Args[1]) b, _ := strconv.Atoi(os.Args[2]) c, _ := strconv.Atoi(os.Args[3]) @@ -32,3 +36,10 @@ func main() { } } } + +func validateArgs() bool { + if len(os.Args) != 3 { + return false + } + return true +}