@@ -3,7 +3,6 @@ package identities
33import (
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 :
0 commit comments