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                 super.waitShutdown.Add(1)
  42                 go func() {
  43                         defer super.waitShutdown.Done()
  44                         fail(super.RunProgram(ctx, super.tempdir, nil, []string{"ARVADOS_SERVICE_INTERNAL_URL=" + u.String()}, binfile, runner.name, "-config", super.configfile))
  45                 }()
  46         }
  47         return nil
  48 }
  49
  50 // Run a Go service that isn't bundled in arvados-server.
  51 type runGoProgram struct {
  52         src     string           // source dir, e.g., "services/keepproxy"
  53         svc     arvados.Service  // cluster.Services.* entry with the desired InternalURLs
  54         depends []supervisedTask // wait for these tasks before starting
  55 }
  56
  57 func (runner runGoProgram) String() string {
  58         _, basename := filepath.Split(runner.src)
  59         return basename
  60 }
  61
  62 func (runner runGoProgram) Run(ctx context.Context, fail func(error), super *Supervisor) error {
  63         if len(runner.svc.InternalURLs) == 0 {
  64                 return errors.New("bug: runGoProgram needs non-empty svc.InternalURLs")
  65         }
  66
  67         binfile, err := super.installGoProgram(ctx, runner.src)
  68         if err != nil {
  69                 return err
  70         }
  71         if ctx.Err() != nil {
  72                 return ctx.Err()
  73         }
  74
  75         err = super.RunProgram(ctx, super.tempdir, nil, nil, binfile, "-version")
  76         if err != nil {
  77                 return err
  78         }
  79
  80         super.wait(ctx, runner.depends...)
  81         for u := range runner.svc.InternalURLs {
  82                 u := u
  83                 super.waitShutdown.Add(1)
  84                 go func() {
  85                         defer super.waitShutdown.Done()
  86                         fail(super.RunProgram(ctx, super.tempdir, nil, []string{"ARVADOS_SERVICE_INTERNAL_URL=" + u.String()}, binfile))
  87                 }()
  88         }
  89         return nil
  90 }