1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83
|
package main
import (
"encoding/json"
"flag"
"fmt"
"io/ioutil"
"log"
"os"
"runtime"
"runtime/pprof"
"github.com/ovn-org/libovsdb/ovsdb"
)
func usage() {
fmt.Fprintf(os.Stderr, "Print schema information:\n")
fmt.Fprintf(os.Stderr, "\tprint_schema [flags] OVS_SCHEMA\n")
fmt.Fprintf(os.Stderr, "Flag:\n")
flag.PrintDefaults()
}
var cpuprofile = flag.String("cpuprofile", "", "write cpu profile to this file")
var memprofile = flag.String("memoryprofile", "", "write memory profile to this file")
var ntimes = flag.Int("ntimes", 1, "Parse the schema N times. Useful for profiling")
var schemas []ovsdb.DatabaseSchema
func main() {
log.SetFlags(0)
flag.Usage = usage
flag.Parse()
if *cpuprofile != "" {
f, err := os.Create(*cpuprofile)
if err != nil {
log.Fatal(err)
}
if err := pprof.StartCPUProfile(f); err != nil {
log.Fatal(err)
}
defer pprof.StopCPUProfile()
}
if len(flag.Args()) != 1 {
flag.Usage()
os.Exit(2)
}
schemaFile, err := os.Open(flag.Args()[0])
if err != nil {
log.Fatal(err)
}
defer schemaFile.Close()
schemaBytes, err := ioutil.ReadAll(schemaFile)
if err != nil {
log.Fatal(err)
}
schemas = make([]ovsdb.DatabaseSchema, *ntimes)
for i := 0; i < *ntimes; i++ {
if err := json.Unmarshal(schemaBytes, &schemas[i]); err != nil {
log.Fatal(err)
}
}
if *memprofile != "" {
f, err := os.Create(*memprofile)
if err != nil {
log.Fatal(err)
}
defer f.Close()
runtime.GC()
if err := pprof.WriteHeapProfile(f); err != nil {
log.Fatal("could not write memory profile: ", err)
}
}
// It only really makes sense to print 1 time
if *ntimes > 0 {
schemas[0].Print(os.Stdout)
}
}
|