INNER CODE UNIT · Go

main

CrunchyData/crunchy-containers · badger/badgerserver.go:36

func main() {
	http.HandleFunc("/api/badgergenerate", BadgerGenerate)
	http.Handle("/static/", http.StripPrefix("/static", http.FileServer(http.Dir("/report"))))
	http.HandleFunc("/", RootPathRedirect)
	port := os.Getenv(PG_BADGER_SERVICE_PORT)
	if port == "" {
		log.Printf("PGBadger port not found. Using default %s\n", DEFAULT_PGBADGER_PORT)
		port = DEFAULT_PGBADGER_PORT
	}
	log.Fatal(http.ListenAndServe(":" + port, nil))
}

// BadgerGenerate perform a pgbadger to create the HTML output file
func BadgerGenerate(w http.ResponseWriter, r *http.Request) {
	log.Println("Generating report..")

	var cmd *exec.Cmd
	cmd = exec.Command("badger-generate.sh")

View source record →