| OLD | NEW |
| 1 /* | 1 /* |
| 2 Common initialization for worker scripts. | 2 Common initialization for worker scripts. |
| 3 */ | 3 */ |
| 4 | 4 |
| 5 package worker_common | 5 package worker_common |
| 6 | 6 |
| 7 import ( | 7 import ( |
| 8 "flag" | 8 "flag" |
| 9 "os" |
| 10 |
| 11 "github.com/skia-dev/glog" |
| 9 | 12 |
| 10 "go.skia.org/infra/ct/go/util" | 13 "go.skia.org/infra/ct/go/util" |
| 11 "go.skia.org/infra/go/common" | 14 "go.skia.org/infra/go/common" |
| 15 "go.skia.org/infra/go/exec" |
| 16 skutil "go.skia.org/infra/go/util" |
| 12 ) | 17 ) |
| 13 | 18 |
| 14 var ( | 19 var ( |
| 15 Local = flag.Bool("local", false, "Running locally if true. As opposed t
o in production.") | 20 Local = flag.Bool("local", false, "Running locally if true. As opposed t
o in production.") |
| 16 ) | 21 ) |
| 17 | 22 |
| 18 func Init() { | 23 func Init() { |
| 19 common.Init() | 24 common.Init() |
| 20 if *Local { | 25 if *Local { |
| 21 util.SetVarsForLocal() | 26 util.SetVarsForLocal() |
| 27 } else { |
| 28 // Add depot_tools to the PATH. |
| 29 skutil.LogErr(os.Setenv("PATH", os.Getenv("PATH")+":"+util.Depot
ToolsDir)) |
| 30 // Bring up Xvfb on workers (for GCE instances). |
| 31 if _, _, err := exec.RunIndefinitely(&exec.Command{ |
| 32 Name: "sudo", |
| 33 Args: []string{"Xvfb", ":0", "-screen", "0", "128
0x1024x24"}, |
| 34 Env: []string{}, |
| 35 InheritPath: true, |
| 36 Timeout: util.XVFB_TIMEOUT, |
| 37 LogStdout: true, |
| 38 Stdout: nil, |
| 39 LogStderr: true, |
| 40 Stderr: nil, |
| 41 }); err != nil { |
| 42 // CT's baremetal machines will already have an active d
isplay 0. |
| 43 glog.Infof("Could not run Xvfb on Display 0: %s", err) |
| 44 } |
| 22 } | 45 } |
| 23 } | 46 } |
| OLD | NEW |