Просмотр исходного кода

Support using a custom file name

Lucas Stadler лет назад: 9
Родитель
Сommit
5230573e28
1 измененных файлов с 6 добавлено и 2 удалено
  1. 6 2
      go/blog/blog.go

+ 6 - 2
go/blog/blog.go

26
var flags struct {
26
var flags struct {
27
	writeBack bool
27
	writeBack bool
28
}
28
}
29
var dataPath string = "blog.yaml"
29
30
30
func init() {
31
func init() {
31
	flag.BoolVar(&flags.writeBack, "write-back", false, "Rewrite the YAML file with the generated ids")
32
	flag.BoolVar(&flags.writeBack, "write-back", false, "Rewrite the YAML file with the generated ids")
34
func main() {
35
func main() {
35
	flag.Parse()
36
	flag.Parse()
36
37
37
	f, err := os.Open("blog.yaml")
38
	if flag.NArg() > 0 {
39
		dataPath = flag.Arg(0)
40
	}
41
	f, err := os.Open(dataPath)
38
	if err != nil {
42
	if err != nil {
39
		exit(err)
43
		exit(err)
40
	}
44
	}
89
		if err != nil {
93
		if err != nil {
90
			exit(err)
94
			exit(err)
91
		}
95
		}
92
		ioutil.WriteFile("blog.yaml", out, 0664)
96
		ioutil.WriteFile(dataPath, out, 0664)
93
	}
97
	}
94
}
98
}
95
99