| // Copyright 2013 The Go Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style |
| // license that can be found in the LICENSE file. |
| |
| package main |
| |
| import ( |
| "context" |
| "flag" |
| "fmt" |
| "net/http" |
| "os" |
| |
| "cloud.google.com/go/compute/metadata" |
| "cloud.google.com/go/datastore" |
| "golang.org/x/playground/internal/metrics" |
| ) |
| |
| var log = newStdLogger() |
| |
| var ( |
| runtests = flag.Bool("runtests", false, "Run integration tests instead of Playground server.") |
| backendURL = flag.String("backend-url", "", "URL for sandbox backend that runs Go binaries.") |
| ) |
| |
| func main() { |
| flag.Parse() |
| s, err := newServer(func(s *server) error { |
| pid := projectID() |
| if pid == "" { |
| s.db = &inMemStore{} |
| } else { |
| c, err := datastore.NewClient(context.Background(), pid) |
| if err != nil { |
| return fmt.Errorf("could not create cloud datastore client: %v", err) |
| } |
| s.db = cloudDatastore{client: c} |
| } |
| if caddr := os.Getenv("MEMCACHED_ADDR"); caddr != "" { |
| s.cache = newGobCache(caddr) |
| log.Printf("App (project ID: %q) is caching results", pid) |
| } else { |
| s.cache = (*gobCache)(nil) // Use a no-op cache implementation. |
| log.Printf("App (project ID: %q) is NOT caching results", pid) |
| } |
| s.log = log |
| return nil |
| }, enableMetrics) |
| if err != nil { |
| log.Fatalf("Error creating server: %v", err) |
| } |
| |
| if *runtests { |
| s.test() |
| return |
| } |
| if *backendURL != "" { |
| // TODO(golang.org/issue/25224) - Remove environment variable and use a flag. |
| os.Setenv("SANDBOX_BACKEND_URL", *backendURL) |
| } |
| |
| port := os.Getenv("PORT") |
| if port == "" { |
| port = "8080" |
| } |
| |
| // Get the backend dialer warmed up. This starts |
| // RegionInstanceGroupDialer queries and health checks. |
| go sandboxBackendClient() |
| |
| log.Printf("Listening on :%v ...", port) |
| log.Fatalf("Error listening on :%v: %v", port, http.ListenAndServe(":"+port, s)) |
| } |
| |
| func enableMetrics(s *server) error { |
| gr, err := metrics.GAEResource(context.Background()) |
| if err != nil { |
| s.log.Printf("metrics.GAEResource() = _, %q", err) |
| } |
| ms, err := metrics.NewService(gr, views) |
| if err != nil { |
| s.log.Printf("Failed to initialize metrics: metrics.NewService() = _, %q. (not on GCP?)", err) |
| } |
| if ms != nil && !metadata.OnGCE() { |
| s.mux.Handle("/metrics", ms) |
| } |
| return nil |
| } |
| |
| func projectID() string { |
| id, err := metadata.ProjectID() |
| if err != nil && os.Getenv("GAE_INSTANCE") != "" { |
| log.Fatalf("Could not determine the project ID: %v", err) |
| } |
| return id |
| } |