mirror of https://github.com/helm/helm
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
440 lines
11 KiB
440 lines
11 KiB
/*
|
|
Copyright 2016 The Kubernetes Authors All rights reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package helm // import "k8s.io/helm/pkg/helm"
|
|
|
|
import (
|
|
"io"
|
|
"time"
|
|
|
|
"golang.org/x/net/context"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/keepalive"
|
|
|
|
"k8s.io/helm/pkg/chartutil"
|
|
"k8s.io/helm/pkg/proto/hapi/chart"
|
|
"k8s.io/helm/pkg/proto/hapi/release"
|
|
rls "k8s.io/helm/pkg/proto/hapi/services"
|
|
"k8s.io/helm/pkg/storage"
|
|
"k8s.io/helm/pkg/storage/driver"
|
|
)
|
|
|
|
// maxMsgSize use 20MB as the default message size limit.
|
|
// grpc library default is 4MB
|
|
const maxMsgSize = 1024 * 1024 * 20
|
|
|
|
// Client manages client side of the Helm-Tiller protocol.
|
|
type Client struct {
|
|
opts options
|
|
store *storage.Storage
|
|
}
|
|
|
|
// NewClient creates a new client.
|
|
func NewClient(opts ...Option) *Client {
|
|
var c Client
|
|
c.store = storage.Init(driver.NewMemory())
|
|
// set some sane defaults
|
|
c.Option(ConnectTimeout(5))
|
|
return c.Option(opts...)
|
|
}
|
|
|
|
// Option configures the Helm client with the provided options.
|
|
func (h *Client) Option(opts ...Option) *Client {
|
|
for _, opt := range opts {
|
|
opt(&h.opts)
|
|
}
|
|
return h
|
|
}
|
|
|
|
// ListReleases lists the current releases.
|
|
func (h *Client) ListReleases(opts ...ReleaseListOption) (*rls.ListReleasesResponse, error) {
|
|
reqOpts := h.opts
|
|
for _, opt := range opts {
|
|
opt(&reqOpts)
|
|
}
|
|
req := &reqOpts.listReq
|
|
ctx := NewContext()
|
|
|
|
if reqOpts.before != nil {
|
|
if err := reqOpts.before(ctx, req); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return h.list(ctx, req)
|
|
}
|
|
|
|
// InstallRelease loads a chart from chstr, installs it, and returns the release response.
|
|
func (h *Client) InstallRelease(chstr, ns string, opts ...InstallOption) (*rls.InstallReleaseResponse, error) {
|
|
// load the chart to install
|
|
chart, err := chartutil.Load(chstr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return h.InstallReleaseFromChart(chart, ns, opts...)
|
|
}
|
|
|
|
// InstallReleaseFromChart installs a new chart and returns the release response.
|
|
func (h *Client) InstallReleaseFromChart(chart *chart.Chart, ns string, opts ...InstallOption) (*rls.InstallReleaseResponse, error) {
|
|
// apply the install options
|
|
reqOpts := h.opts
|
|
for _, opt := range opts {
|
|
opt(&reqOpts)
|
|
}
|
|
req := &reqOpts.instReq
|
|
req.Chart = chart
|
|
req.Namespace = ns
|
|
req.DryRun = reqOpts.dryRun
|
|
req.DisableHooks = reqOpts.disableHooks
|
|
req.ReuseName = reqOpts.reuseName
|
|
ctx := NewContext()
|
|
|
|
if reqOpts.before != nil {
|
|
if err := reqOpts.before(ctx, req); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
err := chartutil.ProcessRequirementsEnabled(req.Chart, req.Values)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
err = chartutil.ProcessRequirementsImportValues(req.Chart)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return h.install(ctx, req)
|
|
}
|
|
|
|
// DeleteRelease uninstalls a named release and returns the response.
|
|
func (h *Client) DeleteRelease(rlsName string, opts ...DeleteOption) (*rls.UninstallReleaseResponse, error) {
|
|
// apply the uninstall options
|
|
reqOpts := h.opts
|
|
for _, opt := range opts {
|
|
opt(&reqOpts)
|
|
}
|
|
|
|
if reqOpts.dryRun {
|
|
// In the dry run case, just see if the release exists
|
|
r, err := h.ReleaseContent(rlsName, 0)
|
|
if err != nil {
|
|
return &rls.UninstallReleaseResponse{}, err
|
|
}
|
|
return &rls.UninstallReleaseResponse{Release: r}, nil
|
|
}
|
|
|
|
req := &reqOpts.uninstallReq
|
|
req.Name = rlsName
|
|
req.DisableHooks = reqOpts.disableHooks
|
|
ctx := NewContext()
|
|
|
|
if reqOpts.before != nil {
|
|
if err := reqOpts.before(ctx, req); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return h.delete(ctx, req)
|
|
}
|
|
|
|
// UpdateRelease loads a chart from chstr and updates a release to a new/different chart.
|
|
func (h *Client) UpdateRelease(rlsName string, chstr string, opts ...UpdateOption) (*rls.UpdateReleaseResponse, error) {
|
|
// load the chart to update
|
|
chart, err := chartutil.Load(chstr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return h.UpdateReleaseFromChart(rlsName, chart, opts...)
|
|
}
|
|
|
|
// UpdateReleaseFromChart updates a release to a new/different chart.
|
|
func (h *Client) UpdateReleaseFromChart(rlsName string, chart *chart.Chart, opts ...UpdateOption) (*rls.UpdateReleaseResponse, error) {
|
|
// apply the update options
|
|
reqOpts := h.opts
|
|
for _, opt := range opts {
|
|
opt(&reqOpts)
|
|
}
|
|
req := &reqOpts.updateReq
|
|
req.Chart = chart
|
|
req.DryRun = reqOpts.dryRun
|
|
req.Name = rlsName
|
|
req.DisableHooks = reqOpts.disableHooks
|
|
req.Recreate = reqOpts.recreate
|
|
req.Force = reqOpts.force
|
|
req.ResetValues = reqOpts.resetValues
|
|
req.ReuseValues = reqOpts.reuseValues
|
|
ctx := NewContext()
|
|
|
|
if reqOpts.before != nil {
|
|
if err := reqOpts.before(ctx, req); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
err := chartutil.ProcessRequirementsEnabled(req.Chart, req.Values)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
err = chartutil.ProcessRequirementsImportValues(req.Chart)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return h.update(ctx, req)
|
|
}
|
|
|
|
// RollbackRelease rolls back a release to the previous version.
|
|
func (h *Client) RollbackRelease(rlsName string, opts ...RollbackOption) (*rls.RollbackReleaseResponse, error) {
|
|
reqOpts := h.opts
|
|
for _, opt := range opts {
|
|
opt(&reqOpts)
|
|
}
|
|
req := &reqOpts.rollbackReq
|
|
req.Recreate = reqOpts.recreate
|
|
req.Force = reqOpts.force
|
|
req.DisableHooks = reqOpts.disableHooks
|
|
req.DryRun = reqOpts.dryRun
|
|
req.Name = rlsName
|
|
ctx := NewContext()
|
|
|
|
if reqOpts.before != nil {
|
|
if err := reqOpts.before(ctx, req); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return h.rollback(ctx, req)
|
|
}
|
|
|
|
// ReleaseStatus returns the given release's status.
|
|
func (h *Client) ReleaseStatus(rlsName string, opts ...StatusOption) (*rls.GetReleaseStatusResponse, error) {
|
|
reqOpts := h.opts
|
|
for _, opt := range opts {
|
|
opt(&reqOpts)
|
|
}
|
|
req := &reqOpts.statusReq
|
|
req.Name = rlsName
|
|
ctx := NewContext()
|
|
|
|
if reqOpts.before != nil {
|
|
if err := reqOpts.before(ctx, req); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return h.status(ctx, req)
|
|
}
|
|
|
|
// ReleaseContent returns the configuration for a given release.
|
|
func (c *Client) ReleaseContent(name string, version int32) (*release.Release, error) {
|
|
if version <= 0 {
|
|
return c.store.Last(name)
|
|
}
|
|
return c.store.Get(name, version)
|
|
}
|
|
|
|
// ReleaseHistory returns a release's revision history.
|
|
func (h *Client) ReleaseHistory(rlsName string, opts ...HistoryOption) (*rls.GetHistoryResponse, error) {
|
|
reqOpts := h.opts
|
|
for _, opt := range opts {
|
|
opt(&reqOpts)
|
|
}
|
|
|
|
req := &reqOpts.histReq
|
|
req.Name = rlsName
|
|
ctx := NewContext()
|
|
|
|
if reqOpts.before != nil {
|
|
if err := reqOpts.before(ctx, req); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return h.history(ctx, req)
|
|
}
|
|
|
|
// RunReleaseTest executes a pre-defined test on a release.
|
|
func (h *Client) RunReleaseTest(rlsName string, opts ...ReleaseTestOption) (<-chan *rls.TestReleaseResponse, <-chan error) {
|
|
reqOpts := h.opts
|
|
for _, opt := range opts {
|
|
opt(&reqOpts)
|
|
}
|
|
|
|
req := &reqOpts.testReq
|
|
req.Name = rlsName
|
|
ctx := NewContext()
|
|
|
|
return h.test(ctx, req)
|
|
}
|
|
|
|
// connect returns a gRPC connection to Tiller or error. The gRPC dial options
|
|
// are constructed here.
|
|
func (h *Client) connect(ctx context.Context) (conn *grpc.ClientConn, err error) {
|
|
opts := []grpc.DialOption{
|
|
grpc.WithBlock(),
|
|
grpc.WithKeepaliveParams(keepalive.ClientParameters{
|
|
// Send keepalive every 30 seconds to prevent the connection from
|
|
// getting closed by upstreams
|
|
Time: time.Duration(30) * time.Second,
|
|
}),
|
|
grpc.WithDefaultCallOptions(grpc.MaxCallRecvMsgSize(maxMsgSize)),
|
|
}
|
|
opts = append(opts, grpc.WithInsecure())
|
|
ctx, cancel := context.WithTimeout(ctx, h.opts.connectTimeout)
|
|
defer cancel()
|
|
if conn, err = grpc.DialContext(ctx, h.opts.host, opts...); err != nil {
|
|
return nil, err
|
|
}
|
|
return conn, nil
|
|
}
|
|
|
|
// Executes tiller.ListReleases RPC.
|
|
func (h *Client) list(ctx context.Context, req *rls.ListReleasesRequest) (*rls.ListReleasesResponse, error) {
|
|
c, err := h.connect(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer c.Close()
|
|
|
|
rlc := rls.NewReleaseServiceClient(c)
|
|
s, err := rlc.ListReleases(ctx, req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var resp *rls.ListReleasesResponse
|
|
for {
|
|
r, err := s.Recv()
|
|
if err == io.EOF {
|
|
break
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if resp == nil {
|
|
resp = r
|
|
continue
|
|
}
|
|
resp.Releases = append(resp.Releases, r.GetReleases()[0])
|
|
}
|
|
return resp, nil
|
|
}
|
|
|
|
// Executes tiller.InstallRelease RPC.
|
|
func (h *Client) install(ctx context.Context, req *rls.InstallReleaseRequest) (*rls.InstallReleaseResponse, error) {
|
|
c, err := h.connect(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer c.Close()
|
|
|
|
rlc := rls.NewReleaseServiceClient(c)
|
|
return rlc.InstallRelease(ctx, req)
|
|
}
|
|
|
|
// Executes tiller.UninstallRelease RPC.
|
|
func (h *Client) delete(ctx context.Context, req *rls.UninstallReleaseRequest) (*rls.UninstallReleaseResponse, error) {
|
|
c, err := h.connect(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer c.Close()
|
|
|
|
rlc := rls.NewReleaseServiceClient(c)
|
|
return rlc.UninstallRelease(ctx, req)
|
|
}
|
|
|
|
// Executes tiller.UpdateRelease RPC.
|
|
func (h *Client) update(ctx context.Context, req *rls.UpdateReleaseRequest) (*rls.UpdateReleaseResponse, error) {
|
|
c, err := h.connect(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer c.Close()
|
|
|
|
rlc := rls.NewReleaseServiceClient(c)
|
|
return rlc.UpdateRelease(ctx, req)
|
|
}
|
|
|
|
// Executes tiller.RollbackRelease RPC.
|
|
func (h *Client) rollback(ctx context.Context, req *rls.RollbackReleaseRequest) (*rls.RollbackReleaseResponse, error) {
|
|
c, err := h.connect(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer c.Close()
|
|
|
|
rlc := rls.NewReleaseServiceClient(c)
|
|
return rlc.RollbackRelease(ctx, req)
|
|
}
|
|
|
|
// Executes tiller.GetReleaseStatus RPC.
|
|
func (h *Client) status(ctx context.Context, req *rls.GetReleaseStatusRequest) (*rls.GetReleaseStatusResponse, error) {
|
|
c, err := h.connect(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer c.Close()
|
|
|
|
rlc := rls.NewReleaseServiceClient(c)
|
|
return rlc.GetReleaseStatus(ctx, req)
|
|
}
|
|
|
|
// Executes tiller.GetHistory RPC.
|
|
func (h *Client) history(ctx context.Context, req *rls.GetHistoryRequest) (*rls.GetHistoryResponse, error) {
|
|
c, err := h.connect(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer c.Close()
|
|
|
|
rlc := rls.NewReleaseServiceClient(c)
|
|
return rlc.GetHistory(ctx, req)
|
|
}
|
|
|
|
// Executes tiller.TestRelease RPC.
|
|
func (h *Client) test(ctx context.Context, req *rls.TestReleaseRequest) (<-chan *rls.TestReleaseResponse, <-chan error) {
|
|
errc := make(chan error, 1)
|
|
c, err := h.connect(ctx)
|
|
if err != nil {
|
|
errc <- err
|
|
return nil, errc
|
|
}
|
|
|
|
ch := make(chan *rls.TestReleaseResponse, 1)
|
|
go func() {
|
|
defer close(errc)
|
|
defer close(ch)
|
|
defer c.Close()
|
|
|
|
rlc := rls.NewReleaseServiceClient(c)
|
|
s, err := rlc.RunReleaseTest(ctx, req)
|
|
if err != nil {
|
|
errc <- err
|
|
return
|
|
}
|
|
|
|
for {
|
|
msg, err := s.Recv()
|
|
if err == io.EOF {
|
|
return
|
|
}
|
|
if err != nil {
|
|
errc <- err
|
|
return
|
|
}
|
|
ch <- msg
|
|
}
|
|
}()
|
|
|
|
return ch, errc
|
|
}
|