func init() { glog.SetToStderr(true) cfg, err := configFrom("cayley_appengine.cfg") if err != nil { glog.Fatalln("Error loading config:", err) } handle, err := db.Open(cfg) if err != nil { glog.Fatalln("Error opening database:", err) } http.SetupRoutes(handle, cfg) }
func prepare(t testing.TB) { cfg.DatabaseType = *backend switch *backend { case "memstore": cfg.DatabasePath = "../data/30kmoviedata.nq.gz" case "leveldb", "bolt": cfg.DatabasePath = "/tmp/cayley_test_" + *backend cfg.DatabaseOptions = map[string]interface{}{ "nosync": true, // It's a test. If we need to load, do it fast. } case "mongo": cfg.DatabasePath = "localhost:27017" cfg.DatabaseOptions = map[string]interface{}{ "database_name": "cayley_test", // provide a default test database } default: t.Fatalf("Untestable backend store %s", *backend) } var err error create.Do(func() { needsLoad := true if graph.IsPersistent(cfg.DatabaseType) { if _, err := os.Stat(cfg.DatabasePath); os.IsNotExist(err) { err = db.Init(cfg) if err != nil { t.Fatalf("Could not initialize database: %v", err) } } else { needsLoad = false } } handle, err = db.Open(cfg) if err != nil { t.Fatalf("Failed to open %q: %v", cfg.DatabasePath, err) } if needsLoad { err = internal.Load(handle.QuadWriter, cfg, "../data/30kmoviedata.nq.gz", "cquad") if err != nil { t.Fatalf("Failed to load %q: %v", cfg.DatabasePath, err) } } }) }
func main() { // No command? It's time for usage. if len(os.Args) == 1 { fmt.Fprintln(os.Stderr, "Cayley is a graph store and graph query layer.") usage() os.Exit(1) } cmd := os.Args[1] os.Args = append(os.Args[:1], os.Args[2:]...) flag.Parse() var buildString string if Version != "" { buildString = fmt.Sprint("Cayley ", Version, " built ", BuildDate) glog.Infoln(buildString) } cfg := configFrom(*configFile) if os.Getenv("GOMAXPROCS") == "" { runtime.GOMAXPROCS(runtime.NumCPU()) glog.Infoln("Setting GOMAXPROCS to", runtime.NumCPU()) } else { glog.Infoln("GOMAXPROCS currently", os.Getenv("GOMAXPROCS"), " -- not adjusting") } var ( handle *graph.Handle err error ) switch cmd { case "version": if Version != "" { fmt.Println(buildString) } else { fmt.Println("Cayley snapshot") } os.Exit(0) case "init": err = db.Init(cfg) if err != nil { break } if *quadFile != "" { handle, err = db.Open(cfg) if err != nil { break } err = internal.Load(handle.QuadWriter, cfg, *quadFile, *quadType) if err != nil { break } handle.Close() } case "load": handle, err = db.Open(cfg) if err != nil { break } err = internal.Load(handle.QuadWriter, cfg, *quadFile, *quadType) if err != nil { break } handle.Close() case "repl": handle, err = db.Open(cfg) if err != nil { break } if !graph.IsPersistent(cfg.DatabaseType) { err = internal.Load(handle.QuadWriter, cfg, "", *quadType) if err != nil { break } } err = db.Repl(handle, *queryLanguage, cfg) handle.Close() case "http": handle, err = db.Open(cfg) if err != nil { break } if !graph.IsPersistent(cfg.DatabaseType) { err = internal.Load(handle.QuadWriter, cfg, "", *quadType) if err != nil { break } } http.Serve(handle, cfg) handle.Close() default: fmt.Println("No command", cmd) usage() } if err != nil { glog.Errorln(err) } }