Added HTTP endpoints for Correctness counts

golden/go/skiacorrectness/main.go

Index: golden/go/skiacorrectness/main.go
diff --git a/golden/go/skiacorrectness/main.go b/golden/go/skiacorrectness/main.go
index 3b23b3161c47f6e0b40804b5695b035521126f64..83953a81fb94fc5b0ca4d00aa6fe5f1742d501c1 100644
--- a/golden/go/skiacorrectness/main.go
+++ b/golden/go/skiacorrectness/main.go
@@ -1,32 +1,118 @@
 package main
 
 import (
+	"encoding/json"
 	"flag"
 	"net/http"
-)
+	"time"
 
-import (
 	"github.com/golang/glog"
+	"github.com/gorilla/mux"
+
+	"skia.googlesource.com/buildbot.git/go/database"
+	"skia.googlesource.com/buildbot.git/golden/go/analysis"
+	"skia.googlesource.com/buildbot.git/golden/go/db"
+	"skia.googlesource.com/buildbot.git/golden/go/expstorage"
+	"skia.googlesource.com/buildbot.git/golden/go/filediffstore"
+	"skia.googlesource.com/buildbot.git/perf/go/filetilestore"
 )
 
 // flags
 var (
-	port      = flag.String("port", ":9000", "HTTP service address (e.g., ':9000')")
-	staticDir = flag.String("static", "./app", "Directory with static content to serve")
-
-// TODO (stephana): Just ideas to be sorted out later
-// tempDir   = flag.String("temp", "./.cache",
-// "Directory to store temporary file and application cache")
+	port         = flag.String("port", ":9000", "HTTP service address (e.g., ':9000')")
+	local        = flag.Bool("local", false, "Running locally if true. As opposed to in production.")
+	staticDir    = flag.String("static_dir", "./app", "Directory with static content to serve")
+	tileStoreDir = flag.String("tile_store_dir", "/tmp/tileStore", "What directory to look for tiles in.")
+	imageDiffDir = flag.String("image_diff_dir", "/tmp/imagediffdir", "What directory to store diff images in.")
+	gsBucketName = flag.String("gs_bucket", "chromium-skia-gm", "Name of the google storage bucket that holds uploaded images.")
+	mysqlConnStr = flag.String("mysql_conn", "", "MySQL connection string for backend database. If 'local' is false the password in this string will be substituted via the metadata server.")
+	sqlitePath   = flag.String("sqlite_path", "./golden.db", "Filepath of the embedded SQLite database. Requires 'local' to be set to true and 'mysql_conn' to be empty to take effect.")
 )
 
+// Response envelope. All responses follow this format. Some fields might

[jcgregorio, 2014/10/18 00:09:03]

Comments begin with the name of the thing being described.

// ResponseEnvelope is...

Here and all the functions below.

[stephana, 2014/10/20 12:56:08]

Done.

+// be empty depending on context.
+type ResponseEnvelope struct {
+	Data   *interface{} `json:"data"`
+	Err    *string      `json:"err"`
+	Status int          `json:"status"`
+}
+
+var analyzer *analysis.Analyzer = nil
+
+// Get the aggregated counts
+func tileCountsHandler(w http.ResponseWriter, r *http.Request) {
+	result, err := analyzer.GetTileCounts()
+	if err != nil {
+		sendErrorResponse(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	sendResponse(w, result, http.StatusOK)
+}
+
+// Process a diff request send via HTTP.

[jcgregorio, 2014/10/18 00:09:03]

sent

[stephana, 2014/10/20 12:56:08]

Done.

+func testCountsHandler(w http.ResponseWriter, r *http.Request) {
+	testName := mux.Vars(r)["testname"]
+	result, err := analyzer.GetTestCounts(testName)
+	if err != nil {
+		sendErrorResponse(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	sendResponse(w, result, http.StatusOK)
+}
+
+// Send an error response with the given error message.
+func sendErrorResponse(w http.ResponseWriter, errorMsg string, status int) {
+	resp := ResponseEnvelope{nil, &errorMsg, status}
+	sendJson(w, &resp)
+}
+
+// Send a non-error response with the given data.
+func sendResponse(w http.ResponseWriter, data interface{}, status int) {
+	resp := ResponseEnvelope{&data, nil, status}
+	sendJson(w, &resp)
+}
+
+// Sends the response envelope rendered as JSON.
+func sendJson(w http.ResponseWriter, resp *ResponseEnvelope) {
+	jsonBytes, err := json.Marshal(resp)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	w.Header().Set("Content-Type", "application/json")
+	w.Write(jsonBytes)
+}
+
 func main() {
 	// parse the arguments
 	flag.Parse()
 
-	// // Static file handling
-	http.Handle("/", http.FileServer(http.Dir(*staticDir)))
+	// Get the expecations storage, the filediff storage and the tilestore.
+	diffStore := filediffstore.NewFileDiffStore(nil, *imageDiffDir, *gsBucketName)
+	vdb := database.NewVersionedDB(db.GetDatabaseConfig(*mysqlConnStr, *sqlitePath, *local))
+	expStore := expstorage.NewSQLExpectationStore(vdb)
+	tileStore := filetilestore.NewFileTileStore(*tileStoreDir, "golden", -1)
+
+	// Initialize the Analyzer
+	analyzer = analysis.NewAnalyzer(expStore, tileStore, diffStore, 5*time.Minute)
+
+	router := mux.NewRouter()
+
+	// Wire up the resources. We use the 'rest' prefix to avoid any name
+	// clashes witht the static files being served.
+	// TODO (stephana): Wrap the handlers in autogzip unless we defer that to
+	// the front-end proxy.
+	router.HandleFunc("/rest/tilecounts", tileCountsHandler)
+	router.HandleFunc("/rest/tilecounts/{testname}", testCountsHandler)
+
+	// Everything else is served out of the static directory.
+	router.PathPrefix("/").Handler(http.FileServer(http.Dir(*staticDir)))
 
-	// Wire up the resources
+	// Send all requests to the router
+	http.Handle("/", router)
 
 	// Start the server
 	glog.Infoln("Serving on http://127.0.0.1" + *port)