lib/controller/railsproxy/railsproxy.go

   1 // Copyright (C) The Arvados Authors. All rights reserved.
   2 //
   3 // SPDX-License-Identifier: AGPL-3.0
   4
   5 // Package railsproxy implements Arvados APIs by proxying to the
   6 // RailsAPI server on the local machine.
   7 package railsproxy
   8
   9 import (
  10         "fmt"
  11         "net/url"
  12         "strings"
  13
  14         "git.arvados.org/arvados.git/lib/controller/rpc"
  15         "git.arvados.org/arvados.git/sdk/go/arvados"
  16 )
  17
  18 // For now, FindRailsAPI always uses the rails API running on this
  19 // node.
  20 func FindRailsAPI(cluster *arvados.Cluster) (*url.URL, bool, error) {
  21         var best *url.URL
  22         for target := range cluster.Services.RailsAPI.InternalURLs {
  23                 target := url.URL(target)
  24                 best = &target
  25                 if strings.HasPrefix(target.Host, "localhost:") || strings.HasPrefix(target.Host, "127.0.0.1:") || strings.HasPrefix(target.Host, "[::1]:") {
  26                         break
  27                 }
  28         }
  29         if best == nil {
  30                 return nil, false, fmt.Errorf("Services.RailsAPI.InternalURLs is empty")
  31         }
  32         return best, cluster.TLS.Insecure, nil
  33 }
  34
  35 func NewConn(cluster *arvados.Cluster) *rpc.Conn {
  36         url, insecure, err := FindRailsAPI(cluster)
  37         if err != nil {
  38                 panic(err)
  39         }
  40         return rpc.NewConn(cluster.ClusterID, url, insecure, rpc.PassthroughTokenProvider)
  41 }