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.tempdir, "bin", "arvados-server")
  34         err := super.RunProgram(ctx, super.tempdir, 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 }