Example #1
0
func main() {
	filename := flag.String("outfile", "", "File to write generated source code to.")
	flag.Parse()
	if *filename == "" {
		log.Fatal("--outfile must be set")
	}
	file, err := os.OpenFile(*filename, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0644)
	if err != nil {
		log.Fatalf("Failed to open %q for writing: %v", *filename, err)
	}
	defer file.Close()
	if err = internal.GenerateFunctionsForRegisteredOps(file); err != nil {
		log.Fatal(err)
	}
}
Example #2
0
func main() {
	filename := flag.String("outfile", "", "File to write generated source code to.")
	flag.Parse()
	if *filename == "" {
		log.Fatal("-outfile must be set")
	}
	os.MkdirAll(filepath.Dir(*filename), 0755)

	var buf bytes.Buffer
	if err := internal.GenerateFunctionsForRegisteredOps(&buf); err != nil {
		log.Fatal(err)
	}
	formatted, err := format.Source(buf.Bytes())
	if err != nil {
		log.Fatalf("Failed to generate valid source? 'go fmt' failed: %v", err)
	}
	if err := ioutil.WriteFile(*filename, formatted, 0644); err != nil {
		log.Fatalf("Failed to write to %q: %v", *filename, err)
	}
}