142 lines
2.7 KiB
Go
142 lines
2.7 KiB
Go
package kubernetes
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/metrue/fx/deploy"
|
|
"github.com/metrue/fx/packer"
|
|
"github.com/metrue/fx/types"
|
|
"k8s.io/client-go/kubernetes"
|
|
"k8s.io/client-go/tools/clientcmd"
|
|
)
|
|
|
|
// K8S client
|
|
type K8S struct {
|
|
*kubernetes.Clientset
|
|
}
|
|
|
|
const namespace = "default"
|
|
|
|
// Create a k8s cluster client
|
|
func Create() (*K8S, error) {
|
|
config, err := clientcmd.BuildConfigFromKubeconfigGetter("", clientcmd.NewDefaultClientConfigLoadingRules().Load)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
clientset, err := kubernetes.NewForConfig(config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &K8S{clientset}, nil
|
|
}
|
|
|
|
// Deploy a image to be a service
|
|
func (k *K8S) Deploy(
|
|
ctx context.Context,
|
|
fn types.Func,
|
|
name string,
|
|
ports []types.PortBinding,
|
|
) error {
|
|
// put source code of function docker project into k8s config map
|
|
tree, err := packer.PackIntoK8SConfigMapFile(fn)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
data := map[string]string{}
|
|
data[ConfigMap.AppMetaEnvName] = tree
|
|
if _, err := k.CreateOrUpdateConfigMap(namespace, name, data); err != nil {
|
|
return err
|
|
}
|
|
|
|
selector := map[string]string{
|
|
"app": "fx-app-" + name,
|
|
}
|
|
|
|
const replicas = int32(3)
|
|
if _, err := k.GetDeployment(namespace, name); err != nil {
|
|
// TODO enable passing replica from fx CLI
|
|
if _, err := k.CreateDeploymentWithInitContainer(
|
|
namespace,
|
|
name,
|
|
ports,
|
|
replicas,
|
|
selector,
|
|
); err != nil {
|
|
return err
|
|
}
|
|
} else {
|
|
if _, err := k.UpdateDeployment(
|
|
namespace,
|
|
name,
|
|
name,
|
|
ports,
|
|
replicas,
|
|
selector,
|
|
); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
// TODO fx should be able to know what's the target Kubernetes service platform
|
|
// it's going to deploy to
|
|
const isOnPublicCloud = true
|
|
typ := "LoadBalancer"
|
|
if !isOnPublicCloud {
|
|
typ = "NodePort"
|
|
}
|
|
|
|
if _, err := k.GetService(namespace, name); err != nil {
|
|
if _, err := k.CreateService(
|
|
namespace,
|
|
name,
|
|
typ,
|
|
ports,
|
|
selector,
|
|
); err != nil {
|
|
return err
|
|
}
|
|
} else {
|
|
if _, err := k.UpdateService(
|
|
namespace,
|
|
name,
|
|
typ,
|
|
ports,
|
|
selector,
|
|
); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Update a service
|
|
func (k *K8S) Update(ctx context.Context, name string) error {
|
|
return nil
|
|
}
|
|
|
|
// Destroy a service
|
|
func (k *K8S) Destroy(ctx context.Context, name string) error {
|
|
if err := k.DeleteService(namespace, name); err != nil {
|
|
return err
|
|
}
|
|
if err := k.DeleteDeployment(namespace, name); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// GetStatus get status of a service
|
|
func (k *K8S) GetStatus(ctx context.Context, name string) error {
|
|
return nil
|
|
}
|
|
|
|
// List services
|
|
func (k *K8S) List(ctx context.Context, name string) ([]types.Service, error) {
|
|
return []types.Service{}, nil
|
|
}
|
|
|
|
var (
|
|
_ deploy.Deployer = &K8S{}
|
|
)
|