From 95d261b4d706520b77632c9ed2054fc33a52a63f Mon Sep 17 00:00:00 2001 From: Nia Weiss Date: Wed, 2 Sep 2020 12:53:16 -0400 Subject: [PATCH] configure slurp --- cmd/slurp/slurp.go | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/cmd/slurp/slurp.go b/cmd/slurp/slurp.go index 3c97da63f..1412fb057 100644 --- a/cmd/slurp/slurp.go +++ b/cmd/slurp/slurp.go @@ -19,6 +19,7 @@ import ( "bytes" "compress/gzip" "context" + "flag" "time" //"fmt" @@ -178,13 +179,20 @@ func UploadTar(srcFile string, client *http.InternalClient) error { } func main() { + var host string + var tarSrcPath string + flag.StringVar(&host, "host", "127.0.0.1:10101", "host to import into") + flag.StringVar(&tarSrcPath, "src", "q2.tar.gz", "data to import") + flag.Parse() + + uri, err := pilosa.NewURIFromAddress(host) + panicOn(err) + globURI = uri - host := "127.0.0.1:10101" h := &gohttp.Client{} c, err := http.NewInternalClient(host, h) panicOn(err) - tarSrcPath := os.Args[1] //"q2.tar.gz" t0 := time.Now() println("uploading", tarSrcPath) panicOn(UploadTar(tarSrcPath, c)) @@ -193,12 +201,6 @@ func main() { var globURI *pilosa.URI -func init() { - var err error - globURI, err = pilosa.NewURIFromHostPort("127.0.0.1", 10101) - panicOn(err) -} - // get correct node to go to. func GetImportRoaringURI(index string, shard uint64) *pilosa.URI { return globURI