2019-11-02 15:25:10 +02:00
|
|
|
package client
|
|
|
|
|
|
|
|
import (
|
2019-11-15 15:41:40 +02:00
|
|
|
"bytes"
|
2019-11-02 15:25:10 +02:00
|
|
|
"crypto/tls"
|
|
|
|
"errors"
|
2019-12-17 18:09:51 +02:00
|
|
|
"io"
|
2019-11-02 15:25:10 +02:00
|
|
|
"io/ioutil"
|
|
|
|
"net/http"
|
|
|
|
"os"
|
|
|
|
"path"
|
2019-12-21 01:16:05 +02:00
|
|
|
"strconv"
|
2019-11-02 15:25:10 +02:00
|
|
|
|
2019-12-24 19:42:22 +02:00
|
|
|
"github.com/micro/go-micro/util/kubernetes/api"
|
2019-11-02 15:25:10 +02:00
|
|
|
"github.com/micro/go-micro/util/log"
|
|
|
|
)
|
|
|
|
|
|
|
|
var (
|
2019-11-15 15:41:40 +02:00
|
|
|
// path to kubernetes service account token
|
2019-11-02 15:25:10 +02:00
|
|
|
serviceAccountPath = "/var/run/secrets/kubernetes.io/serviceaccount"
|
|
|
|
// ErrReadNamespace is returned when the names could not be read from service account
|
|
|
|
ErrReadNamespace = errors.New("Could not read namespace from service account secret")
|
|
|
|
)
|
|
|
|
|
|
|
|
// Client ...
|
|
|
|
type client struct {
|
|
|
|
opts *api.Options
|
|
|
|
}
|
|
|
|
|
2019-12-21 01:16:05 +02:00
|
|
|
// NewLocalDevClient returns a client that can be used with `kubectl proxy` on an optional port
|
|
|
|
func NewLocalDevClient(port ...int) *client {
|
|
|
|
var p int
|
|
|
|
if len(port) > 1 {
|
|
|
|
log.Fatal("Expected 0 or 1 port parameters")
|
|
|
|
}
|
|
|
|
if len(port) == 0 {
|
|
|
|
p = 8001
|
|
|
|
} else {
|
|
|
|
p = port[0]
|
|
|
|
}
|
|
|
|
return &client{
|
|
|
|
opts: &api.Options{
|
|
|
|
Client: http.DefaultClient,
|
|
|
|
Host: "http://localhost:" + strconv.Itoa(p),
|
|
|
|
Namespace: "default",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-11-15 15:41:40 +02:00
|
|
|
// NewClientInCluster creates a Kubernetes client for use from within a k8s pod.
|
2019-11-02 15:25:10 +02:00
|
|
|
func NewClientInCluster() *client {
|
|
|
|
host := "https://" + os.Getenv("KUBERNETES_SERVICE_HOST") + ":" + os.Getenv("KUBERNETES_SERVICE_PORT")
|
|
|
|
|
|
|
|
s, err := os.Stat(serviceAccountPath)
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
|
|
|
if s == nil || !s.IsDir() {
|
2019-11-15 15:41:40 +02:00
|
|
|
log.Fatal(errors.New("service account not found"))
|
2019-11-02 15:25:10 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
token, err := ioutil.ReadFile(path.Join(serviceAccountPath, "token"))
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
|
|
|
t := string(token)
|
|
|
|
|
|
|
|
ns, err := detectNamespace()
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
crt, err := CertPoolFromFile(path.Join(serviceAccountPath, "ca.crt"))
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
c := &http.Client{
|
|
|
|
Transport: &http.Transport{
|
|
|
|
TLSClientConfig: &tls.Config{
|
|
|
|
RootCAs: crt,
|
|
|
|
},
|
|
|
|
DisableCompression: true,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
return &client{
|
|
|
|
opts: &api.Options{
|
|
|
|
Client: c,
|
|
|
|
Host: host,
|
|
|
|
Namespace: ns,
|
|
|
|
BearerToken: &t,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func detectNamespace() (string, error) {
|
|
|
|
nsPath := path.Join(serviceAccountPath, "namespace")
|
|
|
|
|
|
|
|
// Make sure it's a file and we can read it
|
|
|
|
if s, e := os.Stat(nsPath); e != nil {
|
|
|
|
return "", e
|
|
|
|
} else if s.IsDir() {
|
|
|
|
return "", ErrReadNamespace
|
|
|
|
}
|
|
|
|
|
|
|
|
// Read the file, and cast to a string
|
|
|
|
if ns, e := ioutil.ReadFile(nsPath); e != nil {
|
|
|
|
return string(ns), e
|
|
|
|
} else {
|
|
|
|
return string(ns), nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-11-15 15:41:40 +02:00
|
|
|
// Create creates new API object
|
|
|
|
func (c *client) Create(r *Resource) error {
|
|
|
|
b := new(bytes.Buffer)
|
|
|
|
if err := renderTemplate(r.Kind, b, r.Value); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2019-11-02 15:25:10 +02:00
|
|
|
return api.NewRequest(c.opts).
|
2019-11-15 15:41:40 +02:00
|
|
|
Post().
|
|
|
|
SetHeader("Content-Type", "application/yaml").
|
|
|
|
Resource(r.Kind).
|
|
|
|
Body(b).
|
2019-11-02 15:25:10 +02:00
|
|
|
Do().
|
|
|
|
Error()
|
|
|
|
}
|
2019-11-07 09:44:57 +02:00
|
|
|
|
2019-11-15 15:41:40 +02:00
|
|
|
// Get queries API objects and stores the result in r
|
|
|
|
func (c *client) Get(r *Resource, labels map[string]string) error {
|
|
|
|
return api.NewRequest(c.opts).
|
2019-11-07 09:44:57 +02:00
|
|
|
Get().
|
2019-11-15 15:41:40 +02:00
|
|
|
Resource(r.Kind).
|
2019-11-07 09:44:57 +02:00
|
|
|
Params(&api.Params{LabelSelector: labels}).
|
|
|
|
Do().
|
2019-11-15 15:41:40 +02:00
|
|
|
Into(r.Value)
|
|
|
|
}
|
2019-11-07 09:44:57 +02:00
|
|
|
|
2019-12-24 19:33:05 +02:00
|
|
|
// Log returns logs for a pod
|
|
|
|
func (c *client) Log(r *Resource, opts ...LogOption) (io.ReadCloser, error) {
|
|
|
|
var options LogOptions
|
|
|
|
for _, o := range opts {
|
|
|
|
o(&options)
|
2019-12-21 01:16:05 +02:00
|
|
|
}
|
2019-12-24 19:33:05 +02:00
|
|
|
|
2019-12-17 18:09:51 +02:00
|
|
|
req := api.NewRequest(c.opts).
|
|
|
|
Get().
|
2019-12-24 19:33:05 +02:00
|
|
|
Resource(r.Kind).
|
2019-12-17 18:09:51 +02:00
|
|
|
SubResource("log").
|
2019-12-24 19:33:05 +02:00
|
|
|
Name(r.Name)
|
2019-12-17 18:09:51 +02:00
|
|
|
|
2019-12-24 19:33:05 +02:00
|
|
|
if options.Params != nil {
|
|
|
|
req.Params(&api.Params{Additional: options.Params})
|
2019-12-21 01:16:05 +02:00
|
|
|
}
|
|
|
|
|
2019-12-17 18:09:51 +02:00
|
|
|
resp, err := req.Raw()
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2019-12-21 01:16:05 +02:00
|
|
|
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
|
|
|
|
resp.Body.Close()
|
|
|
|
return nil, errors.New(resp.Request.URL.String() + ": " + resp.Status)
|
|
|
|
}
|
2019-12-17 18:09:51 +02:00
|
|
|
return resp.Body, nil
|
|
|
|
}
|
|
|
|
|
2019-11-15 15:41:40 +02:00
|
|
|
// Update updates API object
|
|
|
|
func (c *client) Update(r *Resource) error {
|
|
|
|
req := api.NewRequest(c.opts).
|
|
|
|
Patch().
|
|
|
|
SetHeader("Content-Type", "application/strategic-merge-patch+json").
|
|
|
|
Resource(r.Kind).
|
|
|
|
Name(r.Name)
|
|
|
|
|
|
|
|
switch r.Kind {
|
|
|
|
case "service":
|
2019-11-27 00:28:08 +02:00
|
|
|
req.Body(r.Value.(*Service))
|
2019-11-15 15:41:40 +02:00
|
|
|
case "deployment":
|
2019-11-27 00:28:08 +02:00
|
|
|
req.Body(r.Value.(*Deployment))
|
2019-11-15 15:41:40 +02:00
|
|
|
default:
|
|
|
|
return errors.New("unsupported resource")
|
|
|
|
}
|
|
|
|
|
|
|
|
return req.Do().Error()
|
|
|
|
}
|
|
|
|
|
|
|
|
// Delete removes API object
|
|
|
|
func (c *client) Delete(r *Resource) error {
|
|
|
|
return api.NewRequest(c.opts).
|
|
|
|
Delete().
|
|
|
|
Resource(r.Kind).
|
|
|
|
Name(r.Name).
|
|
|
|
Do().
|
|
|
|
Error()
|
|
|
|
}
|
|
|
|
|
|
|
|
// List lists API objects and stores the result in r
|
|
|
|
func (c *client) List(r *Resource) error {
|
|
|
|
labels := map[string]string{
|
|
|
|
"micro": "service",
|
|
|
|
}
|
2019-11-22 19:10:00 +02:00
|
|
|
return c.Get(r, labels)
|
2019-11-07 09:44:57 +02:00
|
|
|
}
|