blob: ed59695635ae26e6fc2963c7bca38bdaacc9d2e5 [file] [log] [blame]
// 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
}