X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/f51138289b7625bde6cc0bb9840ccf681944941a..d81ed360051af98d7c0fba3afb07a4b5a815ee41:/services/keepstore/keepstore.go diff --git a/services/keepstore/keepstore.go b/services/keepstore/keepstore.go index eafcc2395e..a363bac255 100644 --- a/services/keepstore/keepstore.go +++ b/services/keepstore/keepstore.go @@ -4,6 +4,7 @@ import ( "bytes" "flag" "fmt" + "git.curoverse.com/arvados.git/sdk/go/keepclient" "io/ioutil" "log" "net" @@ -263,10 +264,10 @@ func main() { KeepVM = MakeRRVolumeManager(goodvols) // Tell the built-in HTTP server to direct all requests to the REST router. - routerWrapper := RESTRouterWrapper{MakeRESTRouter()} - http.HandleFunc("/", func(resp http.ResponseWriter, req *http.Request) { - routerWrapper.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 +275,22 @@ func main() { log.Fatal(err) } + // Initialize Pull queue and worker + keepClient := keepclient.KeepClient{ + Arvados: nil, + Want_replicas: 1, + Using_proxy: true, + Client: &http.Client{}, + } + + // Initialize the pullq and worker + pullq = NewWorkQueue() + go RunPullWorker(pullq, keepClient) + + // Initialize the trashq and worker + trashq = NewWorkQueue() + go RunTrashWorker(trashq) + // Shut down the server gracefully (by closing the listener) // if SIGTERM is received. term := make(chan os.Signal, 1)