X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/aea8c62eaec4d170842b01e0fd1ddb5e256152b8..57b96147e575d7630ddf81cbb2c554d3adc9c7e2:/services/keepstore/keepstore.go diff --git a/services/keepstore/keepstore.go b/services/keepstore/keepstore.go index 3c6d634498..1e8c3d1e0f 100644 --- a/services/keepstore/keepstore.go +++ b/services/keepstore/keepstore.go @@ -13,6 +13,8 @@ import ( "strings" "syscall" "time" + "git.curoverse.com/arvados.git/sdk/go/arvadosclient" + "git.curoverse.com/arvados.git/sdk/go/keepclient" ) // ====================== @@ -263,10 +265,10 @@ func main() { KeepVM = MakeRRVolumeManager(goodvols) // Tell the built-in HTTP server to direct all requests to the REST router. - loggingRouter := MakeLoggingRESTRouter() - http.HandleFunc("/", func(resp http.ResponseWriter, req *http.Request) { - loggingRouter.ServeHTTP(resp, req) - }) + loggingRouter := MakeLoggingRESTRouter() + http.HandleFunc("/", func(resp http.ResponseWriter, req *http.Request) { + loggingRouter.ServeHTTP(resp, req) + }) // Set up a TCP listener. listener, err := net.Listen("tcp", listen) @@ -274,6 +276,20 @@ func main() { log.Fatal(err) } + // Initialize Pull queue and worker + arv, err := arvadosclient.MakeArvadosClient() + if err != nil { + log.Fatalf("Error setting up arvados client %s", err.Error()) + } + + keepClient, err := keepclient.MakeKeepClient(&arv) + if err != nil { + log.Fatalf("Error setting up keep client %s", err.Error()) + } + + pullq = NewWorkQueue() + go RunPullWorker(pullq, keepClient) + // Shut down the server gracefully (by closing the listener) // if SIGTERM is received. term := make(chan os.Signal, 1)