mirror of https://github.com/helm/helm
parent
28d87e1780
commit
e909258fe8
@ -1,158 +0,0 @@
|
|||||||
/*
|
|
||||||
Copyright 2017 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 main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"fmt"
|
|
||||||
"net"
|
|
||||||
|
|
||||||
"golang.org/x/net/context"
|
|
||||||
"google.golang.org/grpc"
|
|
||||||
"google.golang.org/grpc/grpclog"
|
|
||||||
"k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset"
|
|
||||||
|
|
||||||
"github.com/spf13/pflag"
|
|
||||||
"k8s.io/helm/pkg/kube"
|
|
||||||
rudderAPI "k8s.io/helm/pkg/proto/hapi/rudder"
|
|
||||||
"k8s.io/helm/pkg/tiller"
|
|
||||||
"k8s.io/helm/pkg/version"
|
|
||||||
)
|
|
||||||
|
|
||||||
var kubeClient *kube.Client
|
|
||||||
var clientset internalclientset.Interface
|
|
||||||
|
|
||||||
type options struct {
|
|
||||||
listen string
|
|
||||||
}
|
|
||||||
|
|
||||||
func (opts *options) registerFlags() {
|
|
||||||
pflag.StringVarP(&opts.listen, "listen", "l", "127.0.0.1:10001",
|
|
||||||
"Socket for rudder grpc server (default: 127.0.0.1:10001).")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (opts *options) parseFlags() {
|
|
||||||
pflag.Parse()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (opts *options) regAndParseFlags() {
|
|
||||||
opts.registerFlags()
|
|
||||||
opts.parseFlags()
|
|
||||||
}
|
|
||||||
|
|
||||||
func main() {
|
|
||||||
opts := new(options)
|
|
||||||
opts.regAndParseFlags()
|
|
||||||
var err error
|
|
||||||
kubeClient = kube.New(nil)
|
|
||||||
clientset, err = kubeClient.ClientSet()
|
|
||||||
if err != nil {
|
|
||||||
grpclog.Fatalf("Cannot initialize Kubernetes connection: %s", err)
|
|
||||||
}
|
|
||||||
grpclog.Printf("Creating tcp socket on %s\n", opts.listen)
|
|
||||||
lis, err := net.Listen("tcp", opts.listen)
|
|
||||||
if err != nil {
|
|
||||||
grpclog.Fatalf("failed to listen: %v", err)
|
|
||||||
}
|
|
||||||
grpcServer := grpc.NewServer()
|
|
||||||
rudderAPI.RegisterReleaseModuleServiceServer(grpcServer, &ReleaseModuleServiceServer{})
|
|
||||||
|
|
||||||
grpclog.Printf("Starting server on %s\n", opts.listen)
|
|
||||||
grpcServer.Serve(lis)
|
|
||||||
}
|
|
||||||
|
|
||||||
// ReleaseModuleServiceServer provides implementation for rudderAPI.ReleaseModuleServiceServer
|
|
||||||
type ReleaseModuleServiceServer struct{}
|
|
||||||
|
|
||||||
// Version returns Rudder version based on helm version
|
|
||||||
func (r *ReleaseModuleServiceServer) Version(ctx context.Context, in *rudderAPI.VersionReleaseRequest) (*rudderAPI.VersionReleaseResponse, error) {
|
|
||||||
grpclog.Print("version")
|
|
||||||
return &rudderAPI.VersionReleaseResponse{
|
|
||||||
Name: "helm-rudder-native",
|
|
||||||
Version: version.Version,
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// InstallRelease creates a release using kubeClient.Create
|
|
||||||
func (r *ReleaseModuleServiceServer) InstallRelease(ctx context.Context, in *rudderAPI.InstallReleaseRequest) (*rudderAPI.InstallReleaseResponse, error) {
|
|
||||||
grpclog.Print("install")
|
|
||||||
b := bytes.NewBufferString(in.Release.Manifest)
|
|
||||||
err := kubeClient.Create(in.Release.Namespace, b, 500, false)
|
|
||||||
if err != nil {
|
|
||||||
grpclog.Printf("error when creating release: %v", err)
|
|
||||||
}
|
|
||||||
return &rudderAPI.InstallReleaseResponse{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// DeleteRelease deletes a provided release
|
|
||||||
func (r *ReleaseModuleServiceServer) DeleteRelease(ctx context.Context, in *rudderAPI.DeleteReleaseRequest) (*rudderAPI.DeleteReleaseResponse, error) {
|
|
||||||
grpclog.Print("delete")
|
|
||||||
|
|
||||||
resp := &rudderAPI.DeleteReleaseResponse{}
|
|
||||||
rel := in.Release
|
|
||||||
vs, err := tiller.GetVersionSet(clientset.Discovery())
|
|
||||||
if err != nil {
|
|
||||||
return resp, fmt.Errorf("Could not get apiVersions from Kubernetes: %v", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
kept, errs := tiller.DeleteRelease(rel, vs, kubeClient)
|
|
||||||
rel.Manifest = kept
|
|
||||||
|
|
||||||
allErrors := ""
|
|
||||||
for _, e := range errs {
|
|
||||||
allErrors = allErrors + "\n" + e.Error()
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(allErrors) > 0 {
|
|
||||||
err = fmt.Errorf(allErrors)
|
|
||||||
}
|
|
||||||
|
|
||||||
return &rudderAPI.DeleteReleaseResponse{
|
|
||||||
Release: rel,
|
|
||||||
}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// RollbackRelease rolls back the release
|
|
||||||
func (r *ReleaseModuleServiceServer) RollbackRelease(ctx context.Context, in *rudderAPI.RollbackReleaseRequest) (*rudderAPI.RollbackReleaseResponse, error) {
|
|
||||||
grpclog.Print("rollback")
|
|
||||||
c := bytes.NewBufferString(in.Current.Manifest)
|
|
||||||
t := bytes.NewBufferString(in.Target.Manifest)
|
|
||||||
err := kubeClient.Update(in.Target.Namespace, c, t, in.Force, in.Recreate, in.Timeout, in.Wait)
|
|
||||||
return &rudderAPI.RollbackReleaseResponse{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// UpgradeRelease upgrades manifests using kubernetes client
|
|
||||||
func (r *ReleaseModuleServiceServer) UpgradeRelease(ctx context.Context, in *rudderAPI.UpgradeReleaseRequest) (*rudderAPI.UpgradeReleaseResponse, error) {
|
|
||||||
grpclog.Print("upgrade")
|
|
||||||
c := bytes.NewBufferString(in.Current.Manifest)
|
|
||||||
t := bytes.NewBufferString(in.Target.Manifest)
|
|
||||||
err := kubeClient.Update(in.Target.Namespace, c, t, in.Force, in.Recreate, in.Timeout, in.Wait)
|
|
||||||
// upgrade response object should be changed to include status
|
|
||||||
return &rudderAPI.UpgradeReleaseResponse{}, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// ReleaseStatus retrieves release status
|
|
||||||
func (r *ReleaseModuleServiceServer) ReleaseStatus(ctx context.Context, in *rudderAPI.ReleaseStatusRequest) (*rudderAPI.ReleaseStatusResponse, error) {
|
|
||||||
grpclog.Print("status")
|
|
||||||
|
|
||||||
resp, err := kubeClient.Get(in.Release.Namespace, bytes.NewBufferString(in.Release.Manifest))
|
|
||||||
in.Release.Info.Status.Resources = resp
|
|
||||||
return &rudderAPI.ReleaseStatusResponse{
|
|
||||||
Release: in.Release,
|
|
||||||
Info: in.Release.Info,
|
|
||||||
}, err
|
|
||||||
}
|
|
@ -1,26 +0,0 @@
|
|||||||
# Copyright 2017 The Kubernetes Authors.
|
|
||||||
#
|
|
||||||
# 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.
|
|
||||||
|
|
||||||
FROM alpine:3.3
|
|
||||||
|
|
||||||
RUN apk update && apk add ca-certificates && rm -rf /var/cache/apk/*
|
|
||||||
|
|
||||||
ENV HOME /tmp
|
|
||||||
|
|
||||||
COPY tiller /tiller
|
|
||||||
|
|
||||||
EXPOSE 44134
|
|
||||||
|
|
||||||
CMD ["/tiller", "--experimental-release"]
|
|
||||||
|
|
@ -1,25 +0,0 @@
|
|||||||
# Copyright 2017 The Kubernetes Authors.
|
|
||||||
#
|
|
||||||
# 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.
|
|
||||||
|
|
||||||
FROM alpine:3.3
|
|
||||||
|
|
||||||
RUN apk update && apk add ca-certificates && rm -rf /var/cache/apk/*
|
|
||||||
|
|
||||||
ENV HOME /tmp
|
|
||||||
|
|
||||||
COPY rudder /rudder
|
|
||||||
|
|
||||||
EXPOSE 10001
|
|
||||||
|
|
||||||
CMD ["/rudder"]
|
|
Loading…
Reference in new issue