blob: 94a96bfb6355eee66750f0773b0962b001d3ca93 [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"
"fmt"
"net/http"
"os"
"cloud.google.com/go/compute/metadata"
"cloud.google.com/go/datastore"
)
var log = newStdLogger()
func main() {
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 {
log.Printf("App (project ID: %q) is NOT caching results", pid)
}
s.log = log
return nil
})
if err != nil {
log.Fatalf("Error creating server: %v", err)
}
if len(os.Args) > 1 && os.Args[1] == "test" {
s.test()
return
}
port := os.Getenv("PORT")
if port == "" {
port = "8080"
}
log.Printf("Listening on :%v ...", port)
log.Fatalf("Error listening on :%v: %v", port, http.ListenAndServe(":"+port, s))
}
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
}