diff options
| -rw-r--r-- | go/task_runner/main.go | 19 |
1 files changed, 17 insertions, 2 deletions
diff --git a/go/task_runner/main.go b/go/task_runner/main.go index 8f78fa22..659cfcfa 100644 --- a/go/task_runner/main.go +++ b/go/task_runner/main.go @@ -2,6 +2,7 @@ package main import ( "encoding/json" + "errors" "fmt" "log" "os" @@ -15,10 +16,11 @@ type Rpc struct { Args []string `json:"args"` } -func main() { +func get_redis_client() (*redis.Client, error) { host, ok := os.LookupEnv("REDIS_HOST") if !ok { - log.Fatal("Please set REDIS_HOST environment variable") + err := errors.New("Please set REDIS_HOST environment variable") + return nil, err } var network = "tcp" if strings.Contains(host, string(os.PathSeparator)) { @@ -32,6 +34,16 @@ func main() { Password: "", // no password set DB: 0, // use default DB }) + return client, nil +} + +func main() { + log.SetOutput(os.Stdout) + client, err := get_redis_client() + if err != nil { + log.Println(err) + } + for { if result, err := client.BLPop(0, "tasks").Result(); err != nil { log.Println(err) @@ -41,10 +53,13 @@ func main() { if err != nil { log.Println(err) } + log.Printf("Running %s with %v\n", call.Fun, call.Args) if err := tasks[call.Fun](call.Args); err != nil { log.Println(err) + log.Printf("'%s' did not complete\n", call.Fun) continue } else { + log.Printf("'%s' completed\n") if call.Fun == "build_portfolios" { call.Fun = "build_scenarios" } else if call.Fun == "build_scenarios" { |