16306: Show logprefix like 'passenger@railsapi'
[arvados.git] / lib / boot / service.go
1 // Copyright (C) The Arvados Authors. All rights reserved.
2 //
3 // SPDX-License-Identifier: AGPL-3.0
4
5 package boot
6
7 import (
8         "context"
9         "errors"
10         "path/filepath"
11
12         "git.arvados.org/arvados.git/sdk/go/arvados"
13 )
14
15 // Run a service using the arvados-server binary.
16 //
17 // In future this will bring up the service in the current process,
18 // but for now (at least until the subcommand handlers get a shutdown
19 // mechanism) it starts a child process using the arvados-server
20 // binary, which the supervisor is assumed to have installed in
21 // {super.tempdir}/bin/.
22 type runServiceCommand struct {
23         name    string           // arvados-server subcommand, e.g., "controller"
24         svc     arvados.Service  // cluster.Services.* entry with the desired InternalURLs
25         depends []supervisedTask // wait for these tasks before starting
26 }
27
28 func (runner runServiceCommand) String() string {
29         return runner.name
30 }
31
32 func (runner runServiceCommand) Run(ctx context.Context, fail func(error), super *Supervisor) error {
33         binfile := filepath.Join(super.bindir, "arvados-server")
34         err := super.RunProgram(ctx, super.bindir, nil, nil, binfile, "-version")
35         if err != nil {
36                 return err
37         }
38         super.wait(ctx, runner.depends...)
39         for u := range runner.svc.InternalURLs {
40                 u := u
41                 if islocal, err := addrIsLocal(u.Host); err != nil {
42                         return err
43                 } else if !islocal {
44                         continue
45                 }
46                 super.waitShutdown.Add(1)
47                 go func() {
48                         defer super.waitShutdown.Done()
49                         fail(super.RunProgram(ctx, super.tempdir, nil, []string{"ARVADOS_SERVICE_INTERNAL_URL=" + u.String()}, binfile, runner.name, "-config", super.configfile))
50                 }()
51         }
52         return nil
53 }
54
55 // Run a Go service that isn't bundled in arvados-server.
56 type runGoProgram struct {
57         src     string           // source dir, e.g., "services/keepproxy"
58         svc     arvados.Service  // cluster.Services.* entry with the desired InternalURLs
59         depends []supervisedTask // wait for these tasks before starting
60 }
61
62 func (runner runGoProgram) String() string {
63         _, basename := filepath.Split(runner.src)
64         return basename
65 }
66
67 func (runner runGoProgram) Run(ctx context.Context, fail func(error), super *Supervisor) error {
68         if len(runner.svc.InternalURLs) == 0 {
69                 return errors.New("bug: runGoProgram needs non-empty svc.InternalURLs")
70         }
71
72         binfile, err := super.installGoProgram(ctx, runner.src)
73         if err != nil {
74                 return err
75         }
76         if ctx.Err() != nil {
77                 return ctx.Err()
78         }
79
80         err = super.RunProgram(ctx, super.tempdir, nil, nil, binfile, "-version")
81         if err != nil {
82                 return err
83         }
84
85         super.wait(ctx, runner.depends...)
86         for u := range runner.svc.InternalURLs {
87                 u := u
88                 if islocal, err := addrIsLocal(u.Host); err != nil {
89                         return err
90                 } else if !islocal {
91                         continue
92                 }
93                 super.waitShutdown.Add(1)
94                 go func() {
95                         defer super.waitShutdown.Done()
96                         fail(super.RunProgram(ctx, super.tempdir, nil, []string{"ARVADOS_SERVICE_INTERNAL_URL=" + u.String()}, binfile))
97                 }()
98         }
99         return nil
100 }