"crypto/rand"
"errors"
"fmt"
- "git.curoverse.com/arvados.git/sdk/go/arvadosclient"
"git.curoverse.com/arvados.git/sdk/go/keepclient"
"io"
"io/ioutil"
"log"
- "os"
"time"
)
-var arv arvadosclient.ArvadosClient
-var keepClient keepclient.KeepClient
-
/*
Keepstore initiates pull worker channel goroutine.
The channel will process pull list.
Skip the rest of the servers if no errors
Repeat
*/
-func RunPullWorker(nextItem <-chan interface{}) {
- var err error
- 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())
- }
-
+func RunPullWorker(pullq *WorkQueue, keepClient *keepclient.KeepClient) {
+ nextItem := pullq.NextItem
for item := range nextItem {
- Pull(item.(PullRequest))
+ pullRequest := item.(PullRequest)
+ err := PullItemAndProcess(item.(PullRequest), GenerateRandomApiToken(), keepClient)
+ if err == nil {
+ log.Printf("Pull %s success", pullRequest)
+ } else {
+ log.Printf("Pull %s error: %s", pullRequest, err)
+ }
}
}
Using this token & signature, retrieve the given block.
Write to storage
*/
-func Pull(pullRequest PullRequest) (err error) {
- defer func() {
- if err == nil {
- log.Printf("Pull %s success", pullRequest)
- } else {
- log.Printf("Pull %s error: %s", pullRequest, err)
- }
- }()
+func PullItemAndProcess(pullRequest PullRequest, token string, keepClient *keepclient.KeepClient) (err error) {
+ keepClient.Arvados.ApiToken = token
service_roots := make(map[string]string)
for _, addr := range pullRequest.Servers {
service_roots[addr] = addr
}
- keepClient.SetServiceRoots(service_roots)
+ keepClient.SetServiceRoots(service_roots, nil, nil)
// Generate signature with a random token
- PermissionSecret = []byte(os.Getenv("ARVADOS_API_TOKEN"))
expires_at := time.Now().Add(60 * time.Second)
- signedLocator := SignLocator(pullRequest.Locator, GenerateRandomApiToken(), expires_at)
-
- reader, contentLen, _, err := GetContent(signedLocator)
+ signedLocator := SignLocator(pullRequest.Locator, token, expires_at)
+ reader, contentLen, _, err := GetContent(signedLocator, keepClient)
if err != nil {
return
}
}
// Fetch the content for the given locator using keepclient.
-var GetContent = func(signedLocator string) (reader io.ReadCloser, contentLength int64, url string, err error) {
+var GetContent = func(signedLocator string, keepClient *keepclient.KeepClient) (
+ reader io.ReadCloser, contentLength int64, url string, err error) {
reader, blocklen, url, err := keepClient.Get(signedLocator)
return reader, blocklen, url, err
}