mirror of https://github.com/helm/helm
Adam Reese
7 years ago
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