mirror of https://github.com/helm/helm
Merge pull request #2477 from fibonacci1729/master
ref(tiller): refactor tiller APIs into logical filespull/2484/head
Brian
8 years ago
committed by
GitHub
commit
fcf48f430b
@ -0,0 +1,216 @@
|
|||||||
|
/*
|
||||||
|
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 tiller
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
ctx "golang.org/x/net/context"
|
||||||
|
"k8s.io/helm/pkg/chartutil"
|
||||||
|
"k8s.io/helm/pkg/hooks"
|
||||||
|
"k8s.io/helm/pkg/proto/hapi/release"
|
||||||
|
"k8s.io/helm/pkg/proto/hapi/services"
|
||||||
|
relutil "k8s.io/helm/pkg/releaseutil"
|
||||||
|
"k8s.io/helm/pkg/timeconv"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
// InstallRelease installs a release and stores the release record.
|
||||||
|
func (s *ReleaseServer) InstallRelease(c ctx.Context, req *services.InstallReleaseRequest) (*services.InstallReleaseResponse, error) {
|
||||||
|
rel, err := s.prepareRelease(req)
|
||||||
|
if err != nil {
|
||||||
|
s.Log("Failed install prepare step: %s", err)
|
||||||
|
res := &services.InstallReleaseResponse{Release: rel}
|
||||||
|
|
||||||
|
// On dry run, append the manifest contents to a failed release. This is
|
||||||
|
// a stop-gap until we can revisit an error backchannel post-2.0.
|
||||||
|
if req.DryRun && strings.HasPrefix(err.Error(), "YAML parse error") {
|
||||||
|
err = fmt.Errorf("%s\n%s", err, rel.Manifest)
|
||||||
|
}
|
||||||
|
return res, err
|
||||||
|
}
|
||||||
|
|
||||||
|
res, err := s.performRelease(rel, req)
|
||||||
|
if err != nil {
|
||||||
|
s.Log("Failed install perform step: %s", err)
|
||||||
|
}
|
||||||
|
return res, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// prepareRelease builds a release for an install operation.
|
||||||
|
func (s *ReleaseServer) prepareRelease(req *services.InstallReleaseRequest) (*release.Release, error) {
|
||||||
|
if req.Chart == nil {
|
||||||
|
return nil, errMissingChart
|
||||||
|
}
|
||||||
|
|
||||||
|
name, err := s.uniqName(req.Name, req.ReuseName)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
caps, err := capabilities(s.clientset.Discovery())
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
revision := 1
|
||||||
|
ts := timeconv.Now()
|
||||||
|
options := chartutil.ReleaseOptions{
|
||||||
|
Name: name,
|
||||||
|
Time: ts,
|
||||||
|
Namespace: req.Namespace,
|
||||||
|
Revision: revision,
|
||||||
|
IsInstall: true,
|
||||||
|
}
|
||||||
|
valuesToRender, err := chartutil.ToRenderValuesCaps(req.Chart, req.Values, options, caps)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
hooks, manifestDoc, notesTxt, err := s.renderResources(req.Chart, valuesToRender, caps.APIVersions)
|
||||||
|
if err != nil {
|
||||||
|
// Return a release with partial data so that client can show debugging
|
||||||
|
// information.
|
||||||
|
rel := &release.Release{
|
||||||
|
Name: name,
|
||||||
|
Namespace: req.Namespace,
|
||||||
|
Chart: req.Chart,
|
||||||
|
Config: req.Values,
|
||||||
|
Info: &release.Info{
|
||||||
|
FirstDeployed: ts,
|
||||||
|
LastDeployed: ts,
|
||||||
|
Status: &release.Status{Code: release.Status_UNKNOWN},
|
||||||
|
Description: fmt.Sprintf("Install failed: %s", err),
|
||||||
|
},
|
||||||
|
Version: 0,
|
||||||
|
}
|
||||||
|
if manifestDoc != nil {
|
||||||
|
rel.Manifest = manifestDoc.String()
|
||||||
|
}
|
||||||
|
return rel, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Store a release.
|
||||||
|
rel := &release.Release{
|
||||||
|
Name: name,
|
||||||
|
Namespace: req.Namespace,
|
||||||
|
Chart: req.Chart,
|
||||||
|
Config: req.Values,
|
||||||
|
Info: &release.Info{
|
||||||
|
FirstDeployed: ts,
|
||||||
|
LastDeployed: ts,
|
||||||
|
Status: &release.Status{Code: release.Status_UNKNOWN},
|
||||||
|
Description: "Initial install underway", // Will be overwritten.
|
||||||
|
},
|
||||||
|
Manifest: manifestDoc.String(),
|
||||||
|
Hooks: hooks,
|
||||||
|
Version: int32(revision),
|
||||||
|
}
|
||||||
|
if len(notesTxt) > 0 {
|
||||||
|
rel.Info.Status.Notes = notesTxt
|
||||||
|
}
|
||||||
|
|
||||||
|
err = validateManifest(s.env.KubeClient, req.Namespace, manifestDoc.Bytes())
|
||||||
|
return rel, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// performRelease runs a release.
|
||||||
|
func (s *ReleaseServer) performRelease(r *release.Release, req *services.InstallReleaseRequest) (*services.InstallReleaseResponse, error) {
|
||||||
|
res := &services.InstallReleaseResponse{Release: r}
|
||||||
|
|
||||||
|
if req.DryRun {
|
||||||
|
s.Log("Dry run for %s", r.Name)
|
||||||
|
res.Release.Info.Description = "Dry run complete"
|
||||||
|
return res, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// pre-install hooks
|
||||||
|
if !req.DisableHooks {
|
||||||
|
if err := s.execHook(r.Hooks, r.Name, r.Namespace, hooks.PreInstall, req.Timeout); err != nil {
|
||||||
|
return res, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
switch h, err := s.env.Releases.History(req.Name); {
|
||||||
|
// if this is a replace operation, append to the release history
|
||||||
|
case req.ReuseName && err == nil && len(h) >= 1:
|
||||||
|
// get latest release revision
|
||||||
|
relutil.Reverse(h, relutil.SortByRevision)
|
||||||
|
|
||||||
|
// old release
|
||||||
|
old := h[0]
|
||||||
|
|
||||||
|
// update old release status
|
||||||
|
old.Info.Status.Code = release.Status_SUPERSEDED
|
||||||
|
s.recordRelease(old, true)
|
||||||
|
|
||||||
|
// update new release with next revision number
|
||||||
|
// so as to append to the old release's history
|
||||||
|
r.Version = old.Version + 1
|
||||||
|
updateReq := &services.UpdateReleaseRequest{
|
||||||
|
Wait: req.Wait,
|
||||||
|
Recreate: false,
|
||||||
|
Timeout: req.Timeout,
|
||||||
|
}
|
||||||
|
if err := s.ReleaseModule.Update(old, r, updateReq, s.env); err != nil {
|
||||||
|
msg := fmt.Sprintf("Release replace %q failed: %s", r.Name, err)
|
||||||
|
s.Log("warning: %s", msg)
|
||||||
|
old.Info.Status.Code = release.Status_SUPERSEDED
|
||||||
|
r.Info.Status.Code = release.Status_FAILED
|
||||||
|
r.Info.Description = msg
|
||||||
|
s.recordRelease(old, true)
|
||||||
|
s.recordRelease(r, false)
|
||||||
|
return res, err
|
||||||
|
}
|
||||||
|
|
||||||
|
default:
|
||||||
|
// nothing to replace, create as normal
|
||||||
|
// regular manifests
|
||||||
|
if err := s.ReleaseModule.Create(r, req, s.env); err != nil {
|
||||||
|
msg := fmt.Sprintf("Release %q failed: %s", r.Name, err)
|
||||||
|
s.Log("warning: %s", msg)
|
||||||
|
r.Info.Status.Code = release.Status_FAILED
|
||||||
|
r.Info.Description = msg
|
||||||
|
s.recordRelease(r, false)
|
||||||
|
return res, fmt.Errorf("release %s failed: %s", r.Name, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// post-install hooks
|
||||||
|
if !req.DisableHooks {
|
||||||
|
if err := s.execHook(r.Hooks, r.Name, r.Namespace, hooks.PostInstall, req.Timeout); err != nil {
|
||||||
|
msg := fmt.Sprintf("Release %q failed post-install: %s", r.Name, err)
|
||||||
|
s.Log("warning: %s", msg)
|
||||||
|
r.Info.Status.Code = release.Status_FAILED
|
||||||
|
r.Info.Description = msg
|
||||||
|
s.recordRelease(r, false)
|
||||||
|
return res, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
r.Info.Status.Code = release.Status_DEPLOYED
|
||||||
|
r.Info.Description = "Install complete"
|
||||||
|
// This is a tricky case. The release has been created, but the result
|
||||||
|
// cannot be recorded. The truest thing to tell the user is that the
|
||||||
|
// release was created. However, the user will not be able to do anything
|
||||||
|
// further with this release.
|
||||||
|
//
|
||||||
|
// One possible strategy would be to do a timed retry to see if we can get
|
||||||
|
// this stored in the future.
|
||||||
|
s.recordRelease(r, false)
|
||||||
|
|
||||||
|
return res, nil
|
||||||
|
}
|
||||||
@ -0,0 +1,168 @@
|
|||||||
|
/*
|
||||||
|
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 tiller
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
ctx "golang.org/x/net/context"
|
||||||
|
"k8s.io/helm/pkg/chartutil"
|
||||||
|
"k8s.io/helm/pkg/hooks"
|
||||||
|
"k8s.io/helm/pkg/proto/hapi/release"
|
||||||
|
"k8s.io/helm/pkg/proto/hapi/services"
|
||||||
|
"k8s.io/helm/pkg/timeconv"
|
||||||
|
)
|
||||||
|
|
||||||
|
// UpdateRelease takes an existing release and new information, and upgrades the release.
|
||||||
|
func (s *ReleaseServer) UpdateRelease(c ctx.Context, req *services.UpdateReleaseRequest) (*services.UpdateReleaseResponse, error) {
|
||||||
|
err := s.env.Releases.LockRelease(req.Name)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer s.env.Releases.UnlockRelease(req.Name)
|
||||||
|
|
||||||
|
currentRelease, updatedRelease, err := s.prepareUpdate(req)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
res, err := s.performUpdate(currentRelease, updatedRelease, req)
|
||||||
|
if err != nil {
|
||||||
|
return res, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if !req.DryRun {
|
||||||
|
if err := s.env.Releases.Create(updatedRelease); err != nil {
|
||||||
|
return res, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return res, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// prepareUpdate builds an updated release for an update operation.
|
||||||
|
func (s *ReleaseServer) prepareUpdate(req *services.UpdateReleaseRequest) (*release.Release, *release.Release, error) {
|
||||||
|
if !ValidName.MatchString(req.Name) {
|
||||||
|
return nil, nil, errMissingRelease
|
||||||
|
}
|
||||||
|
|
||||||
|
if req.Chart == nil {
|
||||||
|
return nil, nil, errMissingChart
|
||||||
|
}
|
||||||
|
|
||||||
|
// finds the non-deleted release with the given name
|
||||||
|
currentRelease, err := s.env.Releases.Last(req.Name)
|
||||||
|
if err != nil {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// If new values were not supplied in the upgrade, re-use the existing values.
|
||||||
|
if err := s.reuseValues(req, currentRelease); err != nil {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Increment revision count. This is passed to templates, and also stored on
|
||||||
|
// the release object.
|
||||||
|
revision := currentRelease.Version + 1
|
||||||
|
|
||||||
|
ts := timeconv.Now()
|
||||||
|
options := chartutil.ReleaseOptions{
|
||||||
|
Name: req.Name,
|
||||||
|
Time: ts,
|
||||||
|
Namespace: currentRelease.Namespace,
|
||||||
|
IsUpgrade: true,
|
||||||
|
Revision: int(revision),
|
||||||
|
}
|
||||||
|
|
||||||
|
caps, err := capabilities(s.clientset.Discovery())
|
||||||
|
if err != nil {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
valuesToRender, err := chartutil.ToRenderValuesCaps(req.Chart, req.Values, options, caps)
|
||||||
|
if err != nil {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
hooks, manifestDoc, notesTxt, err := s.renderResources(req.Chart, valuesToRender, caps.APIVersions)
|
||||||
|
if err != nil {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Store an updated release.
|
||||||
|
updatedRelease := &release.Release{
|
||||||
|
Name: req.Name,
|
||||||
|
Namespace: currentRelease.Namespace,
|
||||||
|
Chart: req.Chart,
|
||||||
|
Config: req.Values,
|
||||||
|
Info: &release.Info{
|
||||||
|
FirstDeployed: currentRelease.Info.FirstDeployed,
|
||||||
|
LastDeployed: ts,
|
||||||
|
Status: &release.Status{Code: release.Status_UNKNOWN},
|
||||||
|
Description: "Preparing upgrade", // This should be overwritten later.
|
||||||
|
},
|
||||||
|
Version: revision,
|
||||||
|
Manifest: manifestDoc.String(),
|
||||||
|
Hooks: hooks,
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(notesTxt) > 0 {
|
||||||
|
updatedRelease.Info.Status.Notes = notesTxt
|
||||||
|
}
|
||||||
|
err = validateManifest(s.env.KubeClient, currentRelease.Namespace, manifestDoc.Bytes())
|
||||||
|
return currentRelease, updatedRelease, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *ReleaseServer) performUpdate(originalRelease, updatedRelease *release.Release, req *services.UpdateReleaseRequest) (*services.UpdateReleaseResponse, error) {
|
||||||
|
res := &services.UpdateReleaseResponse{Release: updatedRelease}
|
||||||
|
|
||||||
|
if req.DryRun {
|
||||||
|
s.Log("Dry run for %s", updatedRelease.Name)
|
||||||
|
res.Release.Info.Description = "Dry run complete"
|
||||||
|
return res, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// pre-upgrade hooks
|
||||||
|
if !req.DisableHooks {
|
||||||
|
if err := s.execHook(updatedRelease.Hooks, updatedRelease.Name, updatedRelease.Namespace, hooks.PreUpgrade, req.Timeout); err != nil {
|
||||||
|
return res, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if err := s.ReleaseModule.Update(originalRelease, updatedRelease, req, s.env); err != nil {
|
||||||
|
msg := fmt.Sprintf("Upgrade %q failed: %s", updatedRelease.Name, err)
|
||||||
|
s.Log("warning: %s", msg)
|
||||||
|
originalRelease.Info.Status.Code = release.Status_SUPERSEDED
|
||||||
|
updatedRelease.Info.Status.Code = release.Status_FAILED
|
||||||
|
updatedRelease.Info.Description = msg
|
||||||
|
s.recordRelease(originalRelease, true)
|
||||||
|
s.recordRelease(updatedRelease, false)
|
||||||
|
return res, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// post-upgrade hooks
|
||||||
|
if !req.DisableHooks {
|
||||||
|
if err := s.execHook(updatedRelease.Hooks, updatedRelease.Name, updatedRelease.Namespace, hooks.PostUpgrade, req.Timeout); err != nil {
|
||||||
|
return res, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
originalRelease.Info.Status.Code = release.Status_SUPERSEDED
|
||||||
|
s.recordRelease(originalRelease, true)
|
||||||
|
|
||||||
|
updatedRelease.Info.Status.Code = release.Status_DEPLOYED
|
||||||
|
updatedRelease.Info.Description = "Upgrade complete"
|
||||||
|
|
||||||
|
return res, nil
|
||||||
|
}
|
||||||
Loading…
Reference in new issue