Skip to content

Commit 7be2602

Browse files
committed
Satisfied some OCDs
1 parent 8d901b9 commit 7be2602

File tree

2 files changed

+19
-24
lines changed

2 files changed

+19
-24
lines changed

identities/generator.go

Lines changed: 13 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@ package identities
33
import (
44
"encoding/json"
55
"errors"
6-
"flag"
76
"fmt"
87
"io"
98
"math/rand"
@@ -19,30 +18,28 @@ func MainModule(args map[string]interface{}, out io.Writer) {
1918
fmt.Fprintln(out, "Error occurred: ", r)
2019
}
2120
}()
21+
dt_fmt := args["dt_fmt"].(string)
2222
minage := args["minage"].(int)
2323
maxage := args["maxage"].(int)
24-
n := args["n"].(int)
25-
dateformat := args["dateformat"].(string)
24+
number := args["number"].(int)
2625
format := args["format"].(string)
27-
filter := args["filter"].(string)
26+
fields := args["fields"].(string)
2827

29-
if minage >= maxage || n <= 0 {
30-
flag.PrintDefaults()
28+
if minage >= maxage || number <= 0 {
29+
panic("'minage' should be less than or equal to 'maxage'")
3130
}
32-
if filter != "all" {
33-
tmp := strings.Split(filter, ",")
31+
if fields != "all" {
32+
tmp := strings.Split(fields, ",")
3433
err := SetFilter(tmp)
3534
if err != nil {
36-
fmt.Println(err)
37-
flag.PrintDefaults()
35+
panic(err)
3836
}
3937
}
4038

41-
LocalizDate = NewDateFormat(dateformat)
42-
people, err := RandomPeople(minage, maxage, n)
39+
LocalizDate = NewDateFormat(dt_fmt)
40+
people, err := RandomPeople(minage, maxage, number)
4341
if err != nil {
44-
fmt.Println(err)
45-
return
42+
panic(err)
4643
}
4744
tmpArray := strings.Split(format, ",")
4845

@@ -60,16 +57,14 @@ func MainModule(args map[string]interface{}, out io.Writer) {
6057
case "json":
6158
b, err := json.MarshalIndent(&people, "", "\t")
6259
if err != nil {
63-
fmt.Println("error:", err)
64-
return
60+
panic(err)
6561
}
6662
_, _ = out.Write(b)
6763
fmt.Fprintln(out)
6864
case "csv":
6965
err := MarshalCSV(people, out)
7066
if err != nil {
71-
fmt.Println("error:", err)
72-
return
67+
panic(err)
7368
}
7469
fmt.Fprintln(out)
7570
default:

main.go

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -9,20 +9,20 @@ import (
99

1010
var minage = flag.Int("minage", 25, "The minimum age for random people generation. Must be positive and less than maxage.")
1111
var maxage = flag.Int("maxage", 55, "The maximum age for random people generation. Must be positive and more than minage.")
12-
var n = flag.Int("n", 1, "The amount of random people to generate. Must be positive.")
13-
var dateformat = flag.String("dateformat", "eu", "The format of the dates. Supports: 'eu','us','ja'")
12+
var number = flag.Int("number", 1, "The amount of random people to generate. Must be positive.")
13+
var dt_fmt = flag.String("dt_fmt", "eu", "The format of the dates. Supports: 'eu','us','ja'")
1414
var format = flag.String("format", "human", "The comma separated list of formats for the output. Supports: 'json', 'csv', 'human'")
15-
var filter = flag.String("fields", "all", "The comma separated list of fields to print. Use 'all' to print all of them")
15+
var fields = flag.String("fields", "all", "The comma separated list of fields to print. Use 'all' to print all of them")
1616

1717
func main() {
1818
flag.Parse()
1919
args := make(map[string]interface{})
20+
args["dt_fmt"] = *dt_fmt
2021
args["minage"] = *minage
2122
args["maxage"] = *maxage
22-
args["n"] = *n
23-
args["dateformat"] = *dateformat
23+
args["number"] = *number
2424
args["format"] = *format
25-
args["filter"] = *filter
25+
args["fields"] = *fields
2626

2727
identities.MainModule(args, os.Stdout)
2828
}

0 commit comments

Comments
 (0)