msb-public
/
helm
mirror of https://github.com/helm/helm
1
0
Code Issues Projects Releases Wiki Activity

Merge pull request #3283 from nicolerenee/concurrency

Browse Source 
feat(pkg/helm): allow concurrent requests on a shared client
pull/3016/merge
Matthew Fisher 7 years ago committed by GitHub
parent 3b3dae5f75 6ade38fb72
commit 077be881c4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 98 additions and 60 deletions
Show Stats Download Patch File Download Diff File

115
pkg/helm/client.go
Unescape View File

@ -51,14 +51,15 @@ func (h *Client) Option(opts ...Option) *Client {
// ListReleases lists the current releases. // ListReleases lists the current releases.
func (h *Client) ListReleases(opts ...ReleaseListOption) (*rls.ListReleasesResponse, error) { func (h *Client) ListReleases(opts ...ReleaseListOption) (*rls.ListReleasesResponse, error) {
reqOpts := h.opts
for _, opt := range opts { for _, opt := range opts {
opt(&h.opts) opt(&reqOpts)
} }
req := &h.opts.listReq req := &reqOpts.listReq
ctx := NewContext() ctx := NewContext()
if h.opts.before != nil { if reqOpts.before != nil {
if err := h.opts.before(ctx, req); err != nil { if err := reqOpts.before(ctx, req); err != nil {
return nil, err return nil, err
} }
} }
@ -79,19 +80,20 @@ func (h *Client) InstallRelease(chstr, ns string, opts ...InstallOption) (*rls.I
// InstallReleaseFromChart installs a new chart and returns the release response. // InstallReleaseFromChart installs a new chart and returns the release response.
func (h *Client) InstallReleaseFromChart(chart *chart.Chart, ns string, opts ...InstallOption) (*rls.InstallReleaseResponse, error) { func (h *Client) InstallReleaseFromChart(chart *chart.Chart, ns string, opts ...InstallOption) (*rls.InstallReleaseResponse, error) {
// apply the install options // apply the install options
reqOpts := h.opts
for _, opt := range opts { for _, opt := range opts {
opt(&h.opts) opt(&reqOpts)
} }
req := &h.opts.instReq req := &reqOpts.instReq
req.Chart = chart req.Chart = chart
req.Namespace = ns req.Namespace = ns
req.DryRun = h.opts.dryRun req.DryRun = reqOpts.dryRun
req.DisableHooks = h.opts.disableHooks req.DisableHooks = reqOpts.disableHooks
req.ReuseName = h.opts.reuseName req.ReuseName = reqOpts.reuseName
ctx := NewContext() ctx := NewContext()
if h.opts.before != nil { if reqOpts.before != nil {
if err := h.opts.before(ctx, req); err != nil { if err := reqOpts.before(ctx, req); err != nil {
return nil, err return nil, err
} }
} }
@ -110,11 +112,12 @@ func (h *Client) InstallReleaseFromChart(chart *chart.Chart, ns string, opts ...
// DeleteRelease uninstalls a named release and returns the response. // DeleteRelease uninstalls a named release and returns the response.
func (h *Client) DeleteRelease(rlsName string, opts ...DeleteOption) (*rls.UninstallReleaseResponse, error) { func (h *Client) DeleteRelease(rlsName string, opts ...DeleteOption) (*rls.UninstallReleaseResponse, error) {
// apply the uninstall options // apply the uninstall options
reqOpts := h.opts
for _, opt := range opts { for _, opt := range opts {
opt(&h.opts) opt(&reqOpts)
} }
if h.opts.dryRun { if reqOpts.dryRun {
// In the dry run case, just see if the release exists // In the dry run case, just see if the release exists
r, err := h.ReleaseContent(rlsName) r, err := h.ReleaseContent(rlsName)
if err != nil { if err != nil {
@ -123,13 +126,13 @@ func (h *Client) DeleteRelease(rlsName string, opts ...DeleteOption) (*rls.Unins
return &rls.UninstallReleaseResponse{Release: r.Release}, nil return &rls.UninstallReleaseResponse{Release: r.Release}, nil
} }
req := &h.opts.uninstallReq req := &reqOpts.uninstallReq
req.Name = rlsName req.Name = rlsName
req.DisableHooks = h.opts.disableHooks req.DisableHooks = reqOpts.disableHooks
ctx := NewContext() ctx := NewContext()
if h.opts.before != nil { if reqOpts.before != nil {
if err := h.opts.before(ctx, req); err != nil { if err := reqOpts.before(ctx, req); err != nil {
return nil, err return nil, err
} }
} }
@ -149,24 +152,24 @@ func (h *Client) UpdateRelease(rlsName string, chstr string, opts ...UpdateOptio
// UpdateReleaseFromChart updates a release to a new/different chart. // UpdateReleaseFromChart updates a release to a new/different chart.
func (h *Client) UpdateReleaseFromChart(rlsName string, chart *chart.Chart, opts ...UpdateOption) (*rls.UpdateReleaseResponse, error) { func (h *Client) UpdateReleaseFromChart(rlsName string, chart *chart.Chart, opts ...UpdateOption) (*rls.UpdateReleaseResponse, error) {
// apply the update options // apply the update options
reqOpts := h.opts
for _, opt := range opts { for _, opt := range opts {
opt(&h.opts) opt(&reqOpts)
} }
req := &h.opts.updateReq req := &reqOpts.updateReq
req.Chart = chart req.Chart = chart
req.DryRun = h.opts.dryRun req.DryRun = reqOpts.dryRun
req.Name = rlsName req.Name = rlsName
req.DisableHooks = h.opts.disableHooks req.DisableHooks = reqOpts.disableHooks
req.Recreate = h.opts.recreate req.Recreate = reqOpts.recreate
req.Force = h.opts.force req.Force = reqOpts.force
req.ResetValues = h.opts.resetValues req.ResetValues = reqOpts.resetValues
req.ReuseValues = h.opts.reuseValues req.ReuseValues = reqOpts.reuseValues
ctx := NewContext() ctx := NewContext()
if h.opts.before != nil { if reqOpts.before != nil {
if err := h.opts.before(ctx, req); err != nil { if err := reqOpts.before(ctx, req); err != nil {
return nil, err return nil, err
} }
} }
@ -184,14 +187,15 @@ func (h *Client) UpdateReleaseFromChart(rlsName string, chart *chart.Chart, opts
// GetVersion returns the server version. // GetVersion returns the server version.
func (h *Client) GetVersion(opts ...VersionOption) (*rls.GetVersionResponse, error) { func (h *Client) GetVersion(opts ...VersionOption) (*rls.GetVersionResponse, error) {
reqOpts := h.opts
for _, opt := range opts { for _, opt := range opts {
opt(&h.opts) opt(&reqOpts)
} }
req := &rls.GetVersionRequest{} req := &rls.GetVersionRequest{}
ctx := NewContext() ctx := NewContext()
if h.opts.before != nil { if reqOpts.before != nil {
if err := h.opts.before(ctx, req); err != nil { if err := reqOpts.before(ctx, req); err != nil {
return nil, err return nil, err
} }
} }
@ -200,19 +204,20 @@ func (h *Client) GetVersion(opts ...VersionOption) (*rls.GetVersionResponse, err
// RollbackRelease rolls back a release to the previous version. // RollbackRelease rolls back a release to the previous version.
func (h *Client) RollbackRelease(rlsName string, opts ...RollbackOption) (*rls.RollbackReleaseResponse, error) { func (h *Client) RollbackRelease(rlsName string, opts ...RollbackOption) (*rls.RollbackReleaseResponse, error) {
reqOpts := h.opts
for _, opt := range opts { for _, opt := range opts {
opt(&h.opts) opt(&reqOpts)
} }
req := &h.opts.rollbackReq req := &reqOpts.rollbackReq
req.Recreate = h.opts.recreate req.Recreate = reqOpts.recreate
req.Force = h.opts.force req.Force = reqOpts.force
req.DisableHooks = h.opts.disableHooks req.DisableHooks = reqOpts.disableHooks
req.DryRun = h.opts.dryRun req.DryRun = reqOpts.dryRun
req.Name = rlsName req.Name = rlsName
ctx := NewContext() ctx := NewContext()
if h.opts.before != nil { if reqOpts.before != nil {
if err := h.opts.before(ctx, req); err != nil { if err := reqOpts.before(ctx, req); err != nil {
return nil, err return nil, err
} }
} }
@ -221,15 +226,16 @@ func (h *Client) RollbackRelease(rlsName string, opts ...RollbackOption) (*rls.R
// ReleaseStatus returns the given release's status. // ReleaseStatus returns the given release's status.
func (h *Client) ReleaseStatus(rlsName string, opts ...StatusOption) (*rls.GetReleaseStatusResponse, error) { func (h *Client) ReleaseStatus(rlsName string, opts ...StatusOption) (*rls.GetReleaseStatusResponse, error) {
reqOpts := h.opts
for _, opt := range opts { for _, opt := range opts {
opt(&h.opts) opt(&reqOpts)
} }
req := &h.opts.statusReq req := &reqOpts.statusReq
req.Name = rlsName req.Name = rlsName
ctx := NewContext() ctx := NewContext()
if h.opts.before != nil { if reqOpts.before != nil {
if err := h.opts.before(ctx, req); err != nil { if err := reqOpts.before(ctx, req); err != nil {
return nil, err return nil, err
} }
} }
@ -238,15 +244,16 @@ func (h *Client) ReleaseStatus(rlsName string, opts ...StatusOption) (*rls.GetRe
// ReleaseContent returns the configuration for a given release. // ReleaseContent returns the configuration for a given release.
func (h *Client) ReleaseContent(rlsName string, opts ...ContentOption) (*rls.GetReleaseContentResponse, error) { func (h *Client) ReleaseContent(rlsName string, opts ...ContentOption) (*rls.GetReleaseContentResponse, error) {
reqOpts := h.opts
for _, opt := range opts { for _, opt := range opts {
opt(&h.opts) opt(&reqOpts)
} }
req := &h.opts.contentReq req := &reqOpts.contentReq
req.Name = rlsName req.Name = rlsName
ctx := NewContext() ctx := NewContext()
if h.opts.before != nil { if reqOpts.before != nil {
if err := h.opts.before(ctx, req); err != nil { if err := reqOpts.before(ctx, req); err != nil {
return nil, err return nil, err
} }
} }
@ -255,16 +262,17 @@ func (h *Client) ReleaseContent(rlsName string, opts ...ContentOption) (*rls.Get
// ReleaseHistory returns a release's revision history. // ReleaseHistory returns a release's revision history.
func (h *Client) ReleaseHistory(rlsName string, opts ...HistoryOption) (*rls.GetHistoryResponse, error) { func (h *Client) ReleaseHistory(rlsName string, opts ...HistoryOption) (*rls.GetHistoryResponse, error) {
reqOpts := h.opts
for _, opt := range opts { for _, opt := range opts {
opt(&h.opts) opt(&reqOpts)
} }
req := &h.opts.histReq req := &reqOpts.histReq
req.Name = rlsName req.Name = rlsName
ctx := NewContext() ctx := NewContext()
if h.opts.before != nil { if reqOpts.before != nil {
if err := h.opts.before(ctx, req); err != nil { if err := reqOpts.before(ctx, req); err != nil {
return nil, err return nil, err
} }
} }
@ -273,11 +281,12 @@ func (h *Client) ReleaseHistory(rlsName string, opts ...HistoryOption) (*rls.Get
// RunReleaseTest executes a pre-defined test on a release. // RunReleaseTest executes a pre-defined test on a release.
func (h *Client) RunReleaseTest(rlsName string, opts ...ReleaseTestOption) (<-chan *rls.TestReleaseResponse, <-chan error) { func (h *Client) RunReleaseTest(rlsName string, opts ...ReleaseTestOption) (<-chan *rls.TestReleaseResponse, <-chan error) {
reqOpts := h.opts
for _, opt := range opts { for _, opt := range opts {
opt(&h.opts) opt(&reqOpts)
} }
req := &h.opts.testReq req := &reqOpts.testReq
req.Name = rlsName req.Name = rlsName
ctx := NewContext() ctx := NewContext()

43
pkg/helm/helm_test.go
Unescape View File

@ -87,9 +87,14 @@ func TestListReleases_VerifyOptions(t *testing.T) {
return errSkip return errSkip
}) })
if _, err := NewClient(b4c).ListReleases(ops...); err != errSkip { client := NewClient(b4c)
if _, err := client.ListReleases(ops...); err != errSkip {
t.Fatalf("did not expect error but got (%v)\n``", err) t.Fatalf("did not expect error but got (%v)\n``", err)
} }
// ensure options for call are not saved to client
assert(t, "", client.opts.listReq.Filter)
} }
// Verify each InstallOption is applied to an InstallReleaseRequest correctly. // Verify each InstallOption is applied to an InstallReleaseRequest correctly.
@ -136,9 +141,13 @@ func TestInstallRelease_VerifyOptions(t *testing.T) {
return errSkip return errSkip
}) })
if _, err := NewClient(b4c).InstallRelease(chartPath, namespace, ops...); err != errSkip { client := NewClient(b4c)
if _, err := client.InstallRelease(chartPath, namespace, ops...); err != errSkip {
t.Fatalf("did not expect error but got (%v)\n``", err) t.Fatalf("did not expect error but got (%v)\n``", err)
} }
// ensure options for call are not saved to client
assert(t, "", client.opts.instReq.Name)
} }
// Verify each DeleteOptions is applied to an UninstallReleaseRequest correctly. // Verify each DeleteOptions is applied to an UninstallReleaseRequest correctly.
@ -173,9 +182,13 @@ func TestDeleteRelease_VerifyOptions(t *testing.T) {
return errSkip return errSkip
}) })
if _, err := NewClient(b4c).DeleteRelease(releaseName, ops...); err != errSkip { client := NewClient(b4c)
if _, err := client.DeleteRelease(releaseName, ops...); err != errSkip {
t.Fatalf("did not expect error but got (%v)\n``", err) t.Fatalf("did not expect error but got (%v)\n``", err)
} }
// ensure options for call are not saved to client
assert(t, "", client.opts.uninstallReq.Name)
} }
// Verify each UpdateOption is applied to an UpdateReleaseRequest correctly. // Verify each UpdateOption is applied to an UpdateReleaseRequest correctly.
@ -216,9 +229,13 @@ func TestUpdateRelease_VerifyOptions(t *testing.T) {
return errSkip return errSkip
}) })
if _, err := NewClient(b4c).UpdateRelease(releaseName, chartPath, ops...); err != errSkip { client := NewClient(b4c)
if _, err := client.UpdateRelease(releaseName, chartPath, ops...); err != errSkip {
t.Fatalf("did not expect error but got (%v)\n``", err) t.Fatalf("did not expect error but got (%v)\n``", err)
} }
// ensure options for call are not saved to client
assert(t, "", client.opts.updateReq.Name)
} }
// Verify each RollbackOption is applied to a RollbackReleaseRequest correctly. // Verify each RollbackOption is applied to a RollbackReleaseRequest correctly.
@ -256,9 +273,13 @@ func TestRollbackRelease_VerifyOptions(t *testing.T) {
return errSkip return errSkip
}) })
if _, err := NewClient(b4c).RollbackRelease(releaseName, ops...); err != errSkip { client := NewClient(b4c)
if _, err := client.RollbackRelease(releaseName, ops...); err != errSkip {
t.Fatalf("did not expect error but got (%v)\n``", err) t.Fatalf("did not expect error but got (%v)\n``", err)
} }
// ensure options for call are not saved to client
assert(t, "", client.opts.rollbackReq.Name)
} }
// Verify each StatusOption is applied to a GetReleaseStatusRequest correctly. // Verify each StatusOption is applied to a GetReleaseStatusRequest correctly.
@ -285,9 +306,13 @@ func TestReleaseStatus_VerifyOptions(t *testing.T) {
return errSkip return errSkip
}) })
if _, err := NewClient(b4c).ReleaseStatus(releaseName, StatusReleaseVersion(revision)); err != errSkip { client := NewClient(b4c)
if _, err := client.ReleaseStatus(releaseName, StatusReleaseVersion(revision)); err != errSkip {
t.Fatalf("did not expect error but got (%v)\n``", err) t.Fatalf("did not expect error but got (%v)\n``", err)
} }
// ensure options for call are not saved to client
assert(t, "", client.opts.statusReq.Name)
} }
// Verify each ContentOption is applied to a GetReleaseContentRequest correctly. // Verify each ContentOption is applied to a GetReleaseContentRequest correctly.
@ -314,9 +339,13 @@ func TestReleaseContent_VerifyOptions(t *testing.T) {
return errSkip return errSkip
}) })
if _, err := NewClient(b4c).ReleaseContent(releaseName, ContentReleaseVersion(revision)); err != errSkip { client := NewClient(b4c)
if _, err := client.ReleaseContent(releaseName, ContentReleaseVersion(revision)); err != errSkip {
t.Fatalf("did not expect error but got (%v)\n``", err) t.Fatalf("did not expect error but got (%v)\n``", err)
} }
// ensure options for call are not saved to client
assert(t, "", client.opts.contentReq.Name)
} }
func assert(t *testing.T, expect, actual interface{}) { func assert(t *testing.T, expect, actual interface{}) {

Write Preview
Loading…
Cancel
Save