github.com/jgbaldwinbrown/perf@v0.1.1/storage/localperfdata/app.go (about)

     1  // Copyright 2016 The Go Authors.  All rights reserved.
     2  // Use of this source code is governed by a BSD-style
     3  // license that can be found in the LICENSE file.
     4  
     5  //go:build cgo
     6  // +build cgo
     7  
     8  // Localperfdata runs an HTTP server for benchmark storage.
     9  //
    10  // Usage:
    11  //
    12  //	localperfdata [-addr address] [-view_url_base url] [-base_dir ../appengine] [-dsn file.db]
    13  package main
    14  
    15  import (
    16  	"flag"
    17  	"log"
    18  	"net/http"
    19  
    20  	"golang.org/x/perf/internal/basedir"
    21  	"golang.org/x/perf/storage/app"
    22  	"golang.org/x/perf/storage/db"
    23  	_ "golang.org/x/perf/storage/db/sqlite3"
    24  	"golang.org/x/perf/storage/fs"
    25  	"golang.org/x/perf/storage/fs/local"
    26  )
    27  
    28  var (
    29  	addr        = flag.String("addr", ":8080", "serve HTTP on `address`")
    30  	viewURLBase = flag.String("view_url_base", "", "/upload response with `URL` for viewing")
    31  	dsn         = flag.String("dsn", ":memory:", "sqlite `dsn`")
    32  	data        = flag.String("data", "", "data `directory` (in-memory if empty)")
    33  	baseDir     = flag.String("base_dir", basedir.Find("golang.org/x/perf/storage/appengine"), "base `directory` for static files")
    34  )
    35  
    36  func main() {
    37  	flag.Parse()
    38  
    39  	if *baseDir == "" {
    40  		log.Print("base_dir is required and could not be automatically found")
    41  		flag.Usage()
    42  	}
    43  
    44  	db, err := db.OpenSQL("sqlite3", *dsn)
    45  	if err != nil {
    46  		log.Fatalf("open database: %v", err)
    47  	}
    48  	var fs fs.FS = fs.NewMemFS()
    49  
    50  	if *data != "" {
    51  		fs = local.NewFS(*data)
    52  	}
    53  
    54  	app := &app.App{
    55  		DB:          db,
    56  		FS:          fs,
    57  		ViewURLBase: *viewURLBase,
    58  		Auth:        func(http.ResponseWriter, *http.Request) (string, error) { return "", nil },
    59  		BaseDir:     *baseDir,
    60  	}
    61  	app.RegisterOnMux(http.DefaultServeMux)
    62  
    63  	log.Printf("Listening on %s", *addr)
    64  
    65  	log.Fatal(http.ListenAndServe(*addr, nil))
    66  }