greet/greet.go

43 lines
669 B
Go
Raw Normal View History

2021-08-29 15:36:09 +01:00
package main
import (
"fmt"
"io"
"log"
"os"
2021-08-29 15:36:09 +01:00
)
type numberOfArgumentError struct {
length int
}
func (e numberOfArgumentError) Error() string {
return fmt.Sprintf("unexpected number of arguments passed to the greeter: want 0 or 1 argument, got %d", e.length)
}
2021-08-29 15:36:09 +01:00
func main() {
osArgs := os.Args[1:]
if err := greet(os.Stdout, osArgs); err != nil {
log.Fatalf("ERROR: %s.\n", err)
2021-08-29 15:36:09 +01:00
}
}
func greet(writer io.Writer, args []string) error {
2021-08-29 15:36:09 +01:00
var name string
if len(args) > 1 {
return numberOfArgumentError{length: len(args)}
}
if len(args) == 0 {
2021-08-29 15:36:09 +01:00
name = "World"
} else {
name = args[0]
2021-08-29 15:36:09 +01:00
}
fmt.Fprintf(writer, "Hello, %s!\n", name)
2021-08-29 15:36:09 +01:00
return nil
}