lib/config/deprecated.go

   1 // Copyright (C) The Arvados Authors. All rights reserved.
   2 //
   3 // SPDX-License-Identifier: AGPL-3.0
   4
   5 package config
   6
   7 import (
   8         "fmt"
   9         "io/ioutil"
  10         "os"
  11         "strings"
  12
  13         "git.curoverse.com/arvados.git/sdk/go/arvados"
  14         "github.com/ghodss/yaml"
  15 )
  16
  17 type deprRequestLimits struct {
  18         MaxItemsPerResponse            *int
  19         MultiClusterRequestConcurrency *int
  20 }
  21
  22 type deprCluster struct {
  23         RequestLimits deprRequestLimits
  24         NodeProfiles  map[string]nodeProfile
  25 }
  26
  27 type deprecatedConfig struct {
  28         Clusters map[string]deprCluster
  29 }
  30
  31 type nodeProfile struct {
  32         Controller    systemServiceInstance `json:"arvados-controller"`
  33         Health        systemServiceInstance `json:"arvados-health"`
  34         Keepbalance   systemServiceInstance `json:"keep-balance"`
  35         Keepproxy     systemServiceInstance `json:"keepproxy"`
  36         Keepstore     systemServiceInstance `json:"keepstore"`
  37         Keepweb       systemServiceInstance `json:"keep-web"`
  38         Nodemanager   systemServiceInstance `json:"arvados-node-manager"`
  39         DispatchCloud systemServiceInstance `json:"arvados-dispatch-cloud"`
  40         RailsAPI      systemServiceInstance `json:"arvados-api-server"`
  41         Websocket     systemServiceInstance `json:"arvados-ws"`
  42         Workbench1    systemServiceInstance `json:"arvados-workbench"`
  43 }
  44
  45 type systemServiceInstance struct {
  46         Listen   string
  47         TLS      bool
  48         Insecure bool
  49 }
  50
  51 func (ldr *Loader) applyDeprecatedConfig(cfg *arvados.Config) error {
  52         var dc deprecatedConfig
  53         err := yaml.Unmarshal(ldr.configdata, &dc)
  54         if err != nil {
  55                 return err
  56         }
  57         hostname, err := os.Hostname()
  58         if err != nil {
  59                 return err
  60         }
  61         for id, dcluster := range dc.Clusters {
  62                 cluster, ok := cfg.Clusters[id]
  63                 if !ok {
  64                         return fmt.Errorf("can't load legacy config %q that is not present in current config", id)
  65                 }
  66                 for name, np := range dcluster.NodeProfiles {
  67                         if name == "*" || name == os.Getenv("ARVADOS_NODE_PROFILE") || name == hostname {
  68                                 name = "localhost"
  69                         } else if ldr.Logger != nil {
  70                                 ldr.Logger.Warnf("overriding Clusters.%s.Services using Clusters.%s.NodeProfiles.%s (guessing %q is a hostname)", id, id, name, name)
  71                         }
  72                         applyDeprecatedNodeProfile(name, np.RailsAPI, &cluster.Services.RailsAPI)
  73                         applyDeprecatedNodeProfile(name, np.Controller, &cluster.Services.Controller)
  74                         applyDeprecatedNodeProfile(name, np.DispatchCloud, &cluster.Services.DispatchCloud)
  75                 }
  76                 if dst, n := &cluster.API.MaxItemsPerResponse, dcluster.RequestLimits.MaxItemsPerResponse; n != nil && *n != *dst {
  77                         *dst = *n
  78                 }
  79                 if dst, n := &cluster.API.MaxRequestAmplification, dcluster.RequestLimits.MultiClusterRequestConcurrency; n != nil && *n != *dst {
  80                         *dst = *n
  81                 }
  82                 cfg.Clusters[id] = cluster
  83         }
  84         return nil
  85 }
  86
  87 func applyDeprecatedNodeProfile(hostname string, ssi systemServiceInstance, svc *arvados.Service) {
  88         scheme := "https"
  89         if !ssi.TLS {
  90                 scheme = "http"
  91         }
  92         if svc.InternalURLs == nil {
  93                 svc.InternalURLs = map[arvados.URL]arvados.ServiceInstance{}
  94         }
  95         host := ssi.Listen
  96         if host == "" {
  97                 return
  98         }
  99         if strings.HasPrefix(host, ":") {
 100                 host = hostname + host
 101         }
 102         svc.InternalURLs[arvados.URL{Scheme: scheme, Host: host}] = arvados.ServiceInstance{}
 103 }
 104
 105 const defaultKeepstoreConfigPath = "/etc/arvados/keepstore/keepstore.yml"
 106
 107 type oldKeepstoreConfig struct {
 108         Debug *bool
 109 }
 110
 111 func (ldr *Loader) loadOldConfigHelper(component, path string, target interface{}) error {
 112         if path == "" {
 113                 return nil
 114         }
 115         buf, err := ioutil.ReadFile(path)
 116         if err != nil {
 117                 return err
 118         }
 119
 120         ldr.Logger.Warnf("you should remove the legacy %v config file (%s) after migrating all config keys to the cluster configuration file (%s)", component, path, ldr.Path)
 121
 122         err = yaml.Unmarshal(buf, target)
 123         if err != nil {
 124                 return fmt.Errorf("%s: %s", path, err)
 125         }
 126         return nil
 127 }
 128
 129 // update config using values from an old-style keepstore config file.
 130 func (ldr *Loader) loadOldKeepstoreConfig(cfg *arvados.Config) error {
 131         if ldr.KeepstorePath == "" {
 132                 return nil
 133         }
 134         var oc oldKeepstoreConfig
 135         err := ldr.loadOldConfigHelper("keepstore", ldr.KeepstorePath, &oc)
 136         if os.IsNotExist(err) && (ldr.KeepstorePath == defaultKeepstoreConfigPath) {
 137                 return nil
 138         } else if err != nil {
 139                 return err
 140         }
 141
 142         cluster, err := cfg.GetCluster("")
 143         if err != nil {
 144                 return err
 145         }
 146
 147         if v := oc.Debug; v == nil {
 148         } else if *v && cluster.SystemLogs.LogLevel != "debug" {
 149                 cluster.SystemLogs.LogLevel = "debug"
 150         } else if !*v && cluster.SystemLogs.LogLevel != "info" {
 151                 cluster.SystemLogs.LogLevel = "info"
 152         }
 153
 154         cfg.Clusters[cluster.ClusterID] = *cluster
 155         return nil
 156 }
 157
 158 type oldCrunchDispatchSlurmConfig struct {
 159         Client *arvados.Client
 160
 161         SbatchArguments *[]string
 162         PollPeriod      *arvados.Duration
 163         PrioritySpread  *int64
 164
 165         // crunch-run command to invoke. The container UUID will be
 166         // appended. If nil, []string{"crunch-run"} will be used.
 167         //
 168         // Example: []string{"crunch-run", "--cgroup-parent-subsystem=memory"}
 169         CrunchRunCommand *[]string
 170
 171         // Extra RAM to reserve (in Bytes) for SLURM job, in addition
 172         // to the amount specified in the container's RuntimeConstraints
 173         ReserveExtraRAM *int64
 174
 175         // Minimum time between two attempts to run the same container
 176         MinRetryPeriod *arvados.Duration
 177
 178         // Batch size for container queries
 179         BatchSize *int64
 180 }
 181
 182 const defaultCrunchDispatchSlurmConfigPath = "/etc/arvados/crunch-dispatch-slurm/crunch-dispatch-slurm.yml"
 183
 184 func loadOldClientConfig(cluster *arvados.Cluster, client *arvados.Client) {
 185         if client == nil {
 186                 return
 187         }
 188         if client.APIHost != "" {
 189                 cluster.Services.Controller.ExternalURL.Host = client.APIHost
 190         }
 191         if client.Scheme != "" {
 192                 cluster.Services.Controller.ExternalURL.Scheme = client.Scheme
 193         } else {
 194                 cluster.Services.Controller.ExternalURL.Scheme = "https"
 195         }
 196         if client.AuthToken != "" {
 197                 cluster.SystemRootToken = client.AuthToken
 198         }
 199         cluster.TLS.Insecure = client.Insecure
 200 }
 201
 202 // update config using values from an crunch-dispatch-slurm config file.
 203 func (ldr *Loader) loadOldCrunchDispatchSlurmConfig(cfg *arvados.Config) error {
 204         if ldr.CrunchDispatchSlurmPath == "" {
 205                 return nil
 206         }
 207         var oc oldCrunchDispatchSlurmConfig
 208         err := ldr.loadOldConfigHelper("crunch-dispatch-slurm", ldr.CrunchDispatchSlurmPath, &oc)
 209         if os.IsNotExist(err) && (ldr.CrunchDispatchSlurmPath == defaultCrunchDispatchSlurmConfigPath) {
 210                 return nil
 211         } else if err != nil {
 212                 return err
 213         }
 214
 215         cluster, err := cfg.GetCluster("")
 216         if err != nil {
 217                 return err
 218         }
 219
 220         loadOldClientConfig(cluster, oc.Client)
 221
 222         if oc.SbatchArguments != nil {
 223                 cluster.Containers.SLURM.SbatchArgumentsList = *oc.SbatchArguments
 224         }
 225         if oc.PollPeriod != nil {
 226                 cluster.Containers.CloudVMs.PollInterval = *oc.PollPeriod
 227         }
 228         if oc.PrioritySpread != nil {
 229                 cluster.Containers.SLURM.PrioritySpread = *oc.PrioritySpread
 230         }
 231         if oc.CrunchRunCommand != nil {
 232                 if len(*oc.CrunchRunCommand) >= 1 {
 233                         cluster.Containers.CrunchRunCommand = (*oc.CrunchRunCommand)[0]
 234                 }
 235                 if len(*oc.CrunchRunCommand) >= 2 {
 236                         cluster.Containers.CrunchRunArgumentsList = (*oc.CrunchRunCommand)[1:]
 237                 }
 238         }
 239         if oc.ReserveExtraRAM != nil {
 240                 cluster.Containers.ReserveExtraRAM = arvados.ByteSize(*oc.ReserveExtraRAM)
 241         }
 242         if oc.MinRetryPeriod != nil {
 243                 cluster.Containers.MinRetryPeriod = *oc.MinRetryPeriod
 244         }
 245         if oc.BatchSize != nil {
 246                 cluster.API.MaxItemsPerResponse = int(*oc.BatchSize)
 247         }
 248
 249         cfg.Clusters[cluster.ClusterID] = *cluster
 250         return nil
 251 }
 252
 253 type oldWsConfig struct {
 254         Client       *arvados.Client
 255         Postgres     *arvados.PostgreSQLConnection
 256         PostgresPool *int
 257         Listen       *string
 258         LogLevel     *string
 259         LogFormat    *string
 260
 261         PingTimeout      *arvados.Duration
 262         ClientEventQueue *int
 263         ServerEventQueue *int
 264
 265         ManagementToken *string
 266 }
 267
 268 const defaultWebsocketConfigPath = "/etc/arvados/ws/ws.yml"
 269
 270 // update config using values from an crunch-dispatch-slurm config file.
 271 func (ldr *Loader) loadOldWebsocketConfig(cfg *arvados.Config) error {
 272         if ldr.WebsocketPath == "" {
 273                 return nil
 274         }
 275         var oc oldWsConfig
 276         err := ldr.loadOldConfigHelper("arvados-ws", ldr.WebsocketPath, &oc)
 277         if os.IsNotExist(err) && ldr.WebsocketPath == defaultWebsocketConfigPath {
 278                 return nil
 279         } else if err != nil {
 280                 return err
 281         }
 282
 283         cluster, err := cfg.GetCluster("")
 284         if err != nil {
 285                 return err
 286         }
 287
 288         loadOldClientConfig(cluster, oc.Client)
 289
 290         if oc.Postgres != nil {
 291                 cluster.PostgreSQL.Connection = *oc.Postgres
 292         }
 293         if oc.PostgresPool != nil {
 294                 cluster.PostgreSQL.ConnectionPool = *oc.PostgresPool
 295         }
 296         if oc.Listen != nil {
 297                 cluster.Services.Websocket.InternalURLs[arvados.URL{Host: *oc.Listen}] = arvados.ServiceInstance{}
 298         }
 299         if oc.LogLevel != nil {
 300                 cluster.SystemLogs.LogLevel = *oc.LogLevel
 301         }
 302         if oc.LogFormat != nil {
 303                 cluster.SystemLogs.Format = *oc.LogFormat
 304         }
 305         if oc.PingTimeout != nil {
 306                 cluster.API.SendTimeout = *oc.PingTimeout
 307         }
 308         if oc.ClientEventQueue != nil {
 309                 cluster.API.WebsocketClientEventQueue = *oc.ClientEventQueue
 310         }
 311         if oc.ServerEventQueue != nil {
 312                 cluster.API.WebsocketServerEventQueue = *oc.ServerEventQueue
 313         }
 314         if oc.ManagementToken != nil {
 315                 cluster.ManagementToken = *oc.ManagementToken
 316         }
 317
 318         cfg.Clusters[cluster.ClusterID] = *cluster
 319         return nil
 320 }