Runtime refactoring and NetworkPolicy support (#2016)
This commit is contained in:
@@ -34,7 +34,24 @@ func (k *kubernetes) Init(opts ...runtime.Option) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (k *kubernetes) Logs(s *runtime.Service, options ...runtime.LogsOption) (runtime.Logs, error) {
|
||||
func (k *kubernetes) Logs(resource runtime.Resource, options ...runtime.LogsOption) (runtime.Logs, error) {
|
||||
|
||||
// Handle the various different types of resources:
|
||||
switch resource.Type() {
|
||||
case runtime.TypeNamespace:
|
||||
// noop (Namespace is not supported by *kubernetes.Logs())
|
||||
return nil, nil
|
||||
case runtime.TypeNetworkPolicy:
|
||||
// noop (NetworkPolicy is not supported by *kubernetes.Logs()))
|
||||
return nil, nil
|
||||
case runtime.TypeService:
|
||||
|
||||
// Assert the resource back into a *runtime.Service
|
||||
s, ok := resource.(*runtime.Service)
|
||||
if !ok {
|
||||
return nil, runtime.ErrInvalidResource
|
||||
}
|
||||
|
||||
klo := newLog(k.client, s.Name, options...)
|
||||
|
||||
if !klo.options.Stream {
|
||||
@@ -60,6 +77,9 @@ func (k *kubernetes) Logs(s *runtime.Service, options ...runtime.LogsOption) (ru
|
||||
return nil, err
|
||||
}
|
||||
return stream, nil
|
||||
default:
|
||||
return nil, runtime.ErrInvalidResource
|
||||
}
|
||||
}
|
||||
|
||||
type kubeStream struct {
|
||||
@@ -92,11 +112,14 @@ func (k *kubeStream) Stop() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
// Creates a service
|
||||
func (k *kubernetes) Create(s *runtime.Service, opts ...runtime.CreateOption) error {
|
||||
// Create a resource
|
||||
func (k *kubernetes) Create(resource runtime.Resource, opts ...runtime.CreateOption) error {
|
||||
k.Lock()
|
||||
defer k.Unlock()
|
||||
return k.create(resource, opts...)
|
||||
}
|
||||
|
||||
func (k *kubernetes) create(resource runtime.Resource, opts ...runtime.CreateOption) error {
|
||||
// parse the options
|
||||
options := &runtime.CreateOptions{
|
||||
Type: k.options.Type,
|
||||
@@ -107,6 +130,30 @@ func (k *kubernetes) Create(s *runtime.Service, opts ...runtime.CreateOption) er
|
||||
o(options)
|
||||
}
|
||||
|
||||
// Handle the various different types of resources:
|
||||
switch resource.Type() {
|
||||
case runtime.TypeNamespace:
|
||||
// Assert the resource back into a *runtime.Namespace
|
||||
namespace, ok := resource.(*runtime.Namespace)
|
||||
if !ok {
|
||||
return runtime.ErrInvalidResource
|
||||
}
|
||||
return k.createNamespace(namespace)
|
||||
case runtime.TypeNetworkPolicy:
|
||||
// Assert the resource back into a *runtime.NetworkPolicy
|
||||
networkPolicy, ok := resource.(*runtime.NetworkPolicy)
|
||||
if !ok {
|
||||
return runtime.ErrInvalidResource
|
||||
}
|
||||
return k.createNetworkPolicy(networkPolicy)
|
||||
case runtime.TypeService:
|
||||
|
||||
// Assert the resource back into a *runtime.Service
|
||||
s, ok := resource.(*runtime.Service)
|
||||
if !ok {
|
||||
return runtime.ErrInvalidResource
|
||||
}
|
||||
|
||||
// default the service's source and version
|
||||
if len(s.Source) == 0 {
|
||||
s.Source = k.options.Source
|
||||
@@ -148,6 +195,9 @@ func (k *kubernetes) Create(s *runtime.Service, opts ...runtime.CreateOption) er
|
||||
}
|
||||
|
||||
return nil
|
||||
default:
|
||||
return runtime.ErrInvalidResource
|
||||
}
|
||||
}
|
||||
|
||||
// Read returns all instances of given service
|
||||
@@ -180,8 +230,8 @@ func (k *kubernetes) Read(opts ...runtime.ReadOption) ([]*runtime.Service, error
|
||||
return k.getServices(client.GetNamespace(options.Namespace), client.GetLabels(labels))
|
||||
}
|
||||
|
||||
// Update the service in place
|
||||
func (k *kubernetes) Update(s *runtime.Service, opts ...runtime.UpdateOption) error {
|
||||
// Update a resource in place
|
||||
func (k *kubernetes) Update(resource runtime.Resource, opts ...runtime.UpdateOption) error {
|
||||
k.Lock()
|
||||
defer k.Unlock()
|
||||
|
||||
@@ -193,6 +243,26 @@ func (k *kubernetes) Update(s *runtime.Service, opts ...runtime.UpdateOption) er
|
||||
o(&options)
|
||||
}
|
||||
|
||||
// Handle the various different types of resources:
|
||||
switch resource.Type() {
|
||||
case runtime.TypeNamespace:
|
||||
// noop (Namespace is not supported by *kubernetes.Update())
|
||||
return nil
|
||||
case runtime.TypeNetworkPolicy:
|
||||
// Assert the resource back into a *runtime.NetworkPolicy
|
||||
networkPolicy, ok := resource.(*runtime.NetworkPolicy)
|
||||
if !ok {
|
||||
return runtime.ErrInvalidResource
|
||||
}
|
||||
return k.updateNetworkPolicy(networkPolicy)
|
||||
case runtime.TypeService:
|
||||
|
||||
// Assert the resource back into a *runtime.Service
|
||||
s, ok := resource.(*runtime.Service)
|
||||
if !ok {
|
||||
return runtime.ErrInvalidResource
|
||||
}
|
||||
|
||||
// construct the query
|
||||
labels := map[string]string{}
|
||||
if len(s.Name) > 0 {
|
||||
@@ -248,14 +318,16 @@ func (k *kubernetes) Update(s *runtime.Service, opts ...runtime.UpdateOption) er
|
||||
}
|
||||
|
||||
return nil
|
||||
default:
|
||||
return runtime.ErrInvalidResource
|
||||
}
|
||||
}
|
||||
|
||||
// Delete removes a service
|
||||
func (k *kubernetes) Delete(s *runtime.Service, opts ...runtime.DeleteOption) error {
|
||||
// Delete removes a resource
|
||||
func (k *kubernetes) Delete(resource runtime.Resource, opts ...runtime.DeleteOption) error {
|
||||
k.Lock()
|
||||
defer k.Unlock()
|
||||
|
||||
// parse the options
|
||||
options := runtime.DeleteOptions{
|
||||
Namespace: client.DefaultNamespace,
|
||||
}
|
||||
@@ -263,6 +335,30 @@ func (k *kubernetes) Delete(s *runtime.Service, opts ...runtime.DeleteOption) er
|
||||
o(&options)
|
||||
}
|
||||
|
||||
// Handle the various different types of resources:
|
||||
switch resource.Type() {
|
||||
case runtime.TypeNamespace:
|
||||
// Assert the resource back into a *runtime.Namespace
|
||||
namespace, ok := resource.(*runtime.Namespace)
|
||||
if !ok {
|
||||
return runtime.ErrInvalidResource
|
||||
}
|
||||
return k.deleteNamespace(namespace)
|
||||
case runtime.TypeNetworkPolicy:
|
||||
// Assert the resource back into a *runtime.NetworkPolicy
|
||||
networkPolicy, ok := resource.(*runtime.NetworkPolicy)
|
||||
if !ok {
|
||||
return runtime.ErrInvalidResource
|
||||
}
|
||||
return k.deleteNetworkPolicy(networkPolicy)
|
||||
case runtime.TypeService:
|
||||
|
||||
// Assert the resource back into a *runtime.Service
|
||||
s, ok := resource.(*runtime.Service)
|
||||
if !ok {
|
||||
return runtime.ErrInvalidResource
|
||||
}
|
||||
|
||||
// delete the deployment
|
||||
dep := client.NewDeployment(s, &runtime.CreateOptions{
|
||||
Type: k.options.Type,
|
||||
@@ -308,6 +404,9 @@ func (k *kubernetes) Delete(s *runtime.Service, opts ...runtime.DeleteOption) er
|
||||
}
|
||||
|
||||
return nil
|
||||
default:
|
||||
return runtime.ErrInvalidResource
|
||||
}
|
||||
}
|
||||
|
||||
// Start starts the runtime
|
||||
|
@@ -11,6 +11,9 @@ import (
|
||||
"regexp"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/micro/go-micro/v3/runtime"
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
func setupClient(t *testing.T) {
|
||||
@@ -47,19 +50,45 @@ func setupClient(t *testing.T) {
|
||||
|
||||
func TestNamespaceCreateDelete(t *testing.T) {
|
||||
defer func() {
|
||||
exec.Command("kubectl", "-n", "foobar", "delete", "networkpolicy", "baz").Run()
|
||||
exec.Command("kubectl", "delete", "namespace", "foobar").Run()
|
||||
}()
|
||||
setupClient(t)
|
||||
r := NewRuntime()
|
||||
if err := r.CreateNamespace("foobar"); err != nil {
|
||||
t.Fatalf("Unexpected error creating namespace %s", err)
|
||||
|
||||
// Create a namespace
|
||||
testNamespace, err := runtime.NewNamespace("foobar")
|
||||
assert.NoError(t, err)
|
||||
if err := r.Create(testNamespace); err != nil {
|
||||
t.Fatalf("Unexpected error creating namespace: %v", err)
|
||||
}
|
||||
|
||||
// Check that the namespace exists
|
||||
if !namespaceExists(t, "foobar") {
|
||||
t.Fatalf("Namespace foobar not found")
|
||||
}
|
||||
if err := r.DeleteNamespace("foobar"); err != nil {
|
||||
t.Fatalf("Unexpected error deleting namespace %s", err)
|
||||
|
||||
// Create a networkpolicy:
|
||||
testNetworkPolicy, err := runtime.NewNetworkPolicy("baz", "foobar", nil)
|
||||
assert.NoError(t, err)
|
||||
if err := r.Create(testNetworkPolicy); err != nil {
|
||||
t.Fatalf("Unexpected error creating networkpolicy: %v", err)
|
||||
}
|
||||
|
||||
// Check that the networkpolicy exists:
|
||||
if !networkPolicyExists(t, "foobar", "baz") {
|
||||
t.Fatalf("NetworkPolicy foobar.baz not found")
|
||||
}
|
||||
|
||||
// Tidy up
|
||||
if err := r.Delete(testNetworkPolicy); err != nil {
|
||||
t.Fatalf("Unexpected error deleting networkpolicy: %v", err)
|
||||
}
|
||||
if networkPolicyExists(t, "foobar", "baz") {
|
||||
t.Fatalf("NetworkPolicy foobar.baz still exists")
|
||||
}
|
||||
if err := r.Delete(testNamespace); err != nil {
|
||||
t.Fatalf("Unexpected error deleting namespace: %v", err)
|
||||
}
|
||||
if namespaceExists(t, "foobar") {
|
||||
t.Fatalf("Namespace foobar still exists")
|
||||
@@ -77,5 +106,17 @@ func namespaceExists(t *testing.T, ns string) bool {
|
||||
t.Fatalf("Error listing namespaces %s", err)
|
||||
}
|
||||
return exists
|
||||
|
||||
}
|
||||
|
||||
func networkPolicyExists(t *testing.T, ns, np string) bool {
|
||||
cmd := exec.Command("kubectl", "-n", ns, "get", "networkpolicy")
|
||||
outp, err := cmd.Output()
|
||||
if err != nil {
|
||||
t.Fatalf("Unexpected error listing networkpolicies %s", err)
|
||||
}
|
||||
exists, err := regexp.Match(np, outp)
|
||||
if err != nil {
|
||||
t.Fatalf("Error listing networkpolicies %s", err)
|
||||
}
|
||||
return exists
|
||||
}
|
||||
|
@@ -4,6 +4,7 @@ import (
|
||||
"strings"
|
||||
|
||||
"github.com/micro/go-micro/v3/logger"
|
||||
"github.com/micro/go-micro/v3/runtime"
|
||||
"github.com/micro/go-micro/v3/util/kubernetes/client"
|
||||
)
|
||||
|
||||
@@ -24,7 +25,7 @@ func (k *kubernetes) ensureNamepaceExists(ns string) error {
|
||||
return err
|
||||
}
|
||||
|
||||
if err := k.createNamespace(namespace); err != nil {
|
||||
if err := k.autoCreateNamespace(namespace); err != nil {
|
||||
if logger.V(logger.WarnLevel, logger.DefaultLogger) {
|
||||
logger.Warnf("Error creating namespace %v: %v", namespace, err)
|
||||
}
|
||||
@@ -64,8 +65,8 @@ func (k *kubernetes) namespaceExists(name string) (bool, error) {
|
||||
return false, nil
|
||||
}
|
||||
|
||||
// createNamespace creates a new k8s namespace
|
||||
func (k *kubernetes) createNamespace(namespace string) error {
|
||||
// autoCreateNamespace creates a new k8s namespace
|
||||
func (k *kubernetes) autoCreateNamespace(namespace string) error {
|
||||
ns := client.Namespace{Metadata: &client.Metadata{Name: namespace}}
|
||||
err := k.client.Create(&client.Resource{Kind: "namespace", Value: ns})
|
||||
|
||||
@@ -75,38 +76,54 @@ func (k *kubernetes) createNamespace(namespace string) error {
|
||||
err = nil
|
||||
}
|
||||
|
||||
// add to cache
|
||||
// add to cache and create networkpolicy
|
||||
if err == nil && k.namespaces != nil {
|
||||
k.namespaces = append(k.namespaces, ns)
|
||||
|
||||
if networkPolicy, err := runtime.NewNetworkPolicy("ingress", namespace, map[string]string{"owner": "micro"}); err != nil {
|
||||
return err
|
||||
} else {
|
||||
return k.create(networkPolicy)
|
||||
}
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func (k *kubernetes) CreateNamespace(ns string) error {
|
||||
// createNamespace creates a namespace resource
|
||||
func (k *kubernetes) createNamespace(namespace *runtime.Namespace) error {
|
||||
err := k.client.Create(&client.Resource{
|
||||
Kind: "namespace",
|
||||
Name: namespace.Name,
|
||||
Value: client.Namespace{
|
||||
Metadata: &client.Metadata{
|
||||
Name: ns,
|
||||
Name: namespace.Name,
|
||||
},
|
||||
},
|
||||
})
|
||||
}, client.CreateNamespace(namespace.Name))
|
||||
if err != nil {
|
||||
if logger.V(logger.ErrorLevel, logger.DefaultLogger) {
|
||||
logger.Errorf("Error creating namespace %v: %v", ns, err)
|
||||
logger.Errorf("Error creating namespace %s: %v", namespace.String(), err)
|
||||
}
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
func (k *kubernetes) DeleteNamespace(ns string) error {
|
||||
// deleteNamespace deletes a namespace resource
|
||||
func (k *kubernetes) deleteNamespace(namespace *runtime.Namespace) error {
|
||||
err := k.client.Delete(&client.Resource{
|
||||
Kind: "namespace",
|
||||
Name: ns,
|
||||
})
|
||||
if err != nil && logger.V(logger.ErrorLevel, logger.DefaultLogger) {
|
||||
logger.Errorf("Error deleting namespace %v: %v", ns, err)
|
||||
Name: namespace.Name,
|
||||
Value: client.Namespace{
|
||||
Metadata: &client.Metadata{
|
||||
Name: namespace.Name,
|
||||
},
|
||||
},
|
||||
}, client.DeleteNamespace(namespace.Name))
|
||||
if err != nil {
|
||||
if logger.V(logger.ErrorLevel, logger.DefaultLogger) {
|
||||
logger.Errorf("Error deleting namespace %s: %v", namespace.String(), err)
|
||||
}
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
67
runtime/kubernetes/networkpolicy.go
Normal file
67
runtime/kubernetes/networkpolicy.go
Normal file
@@ -0,0 +1,67 @@
|
||||
package kubernetes
|
||||
|
||||
import (
|
||||
"github.com/micro/go-micro/v3/logger"
|
||||
"github.com/micro/go-micro/v3/runtime"
|
||||
"github.com/micro/go-micro/v3/util/kubernetes/client"
|
||||
)
|
||||
|
||||
// createNetworkPolicy creates a networkpolicy resource
|
||||
func (k *kubernetes) createNetworkPolicy(networkPolicy *runtime.NetworkPolicy) error {
|
||||
err := k.client.Create(&client.Resource{
|
||||
Kind: "networkpolicy",
|
||||
Value: client.NetworkPolicy{
|
||||
AllowedLabels: networkPolicy.AllowedLabels,
|
||||
Metadata: &client.Metadata{
|
||||
Name: networkPolicy.Name,
|
||||
Namespace: networkPolicy.Namespace,
|
||||
},
|
||||
},
|
||||
}, client.CreateNamespace(networkPolicy.Namespace))
|
||||
if err != nil {
|
||||
if logger.V(logger.ErrorLevel, logger.DefaultLogger) {
|
||||
logger.Errorf("Error creating resource %s: %v", networkPolicy.String(), err)
|
||||
}
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
// updateNetworkPolicy updates a networkpolicy resource in-place
|
||||
func (k *kubernetes) updateNetworkPolicy(networkPolicy *runtime.NetworkPolicy) error {
|
||||
err := k.client.Update(&client.Resource{
|
||||
Kind: "networkpolicy",
|
||||
Value: client.NetworkPolicy{
|
||||
AllowedLabels: networkPolicy.AllowedLabels,
|
||||
Metadata: &client.Metadata{
|
||||
Name: networkPolicy.Name,
|
||||
Namespace: networkPolicy.Namespace,
|
||||
},
|
||||
},
|
||||
}, client.UpdateNamespace(networkPolicy.Namespace))
|
||||
if err != nil {
|
||||
if logger.V(logger.ErrorLevel, logger.DefaultLogger) {
|
||||
logger.Errorf("Error updating resource %s: %v", networkPolicy.String(), err)
|
||||
}
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
// deleteNetworkPolicy deletes a networkpolicy resource
|
||||
func (k *kubernetes) deleteNetworkPolicy(networkPolicy *runtime.NetworkPolicy) error {
|
||||
err := k.client.Delete(&client.Resource{
|
||||
Kind: "networkpolicy",
|
||||
Value: client.NetworkPolicy{
|
||||
AllowedLabels: networkPolicy.AllowedLabels,
|
||||
Metadata: &client.Metadata{
|
||||
Name: networkPolicy.Name,
|
||||
Namespace: networkPolicy.Namespace,
|
||||
},
|
||||
},
|
||||
}, client.DeleteNamespace(networkPolicy.Namespace))
|
||||
if err != nil {
|
||||
if logger.V(logger.ErrorLevel, logger.DefaultLogger) {
|
||||
logger.Errorf("Error deleting resource %s: %v", networkPolicy.String(), err)
|
||||
}
|
||||
}
|
||||
return err
|
||||
}
|
@@ -45,6 +45,20 @@ rules:
|
||||
- get
|
||||
- watch
|
||||
- list
|
||||
- apiGroups:
|
||||
- "networking.k8s.io"
|
||||
resources:
|
||||
- networkpolicy
|
||||
- networkpolicies
|
||||
verbs:
|
||||
- get
|
||||
- create
|
||||
- update
|
||||
- delete
|
||||
- deletecollection
|
||||
- list
|
||||
- patch
|
||||
- watch
|
||||
---
|
||||
apiVersion: rbac.authorization.k8s.io/v1
|
||||
kind: ClusterRoleBinding
|
||||
|
@@ -81,7 +81,7 @@ func serviceKey(s *runtime.Service) string {
|
||||
}
|
||||
|
||||
// Create creates a new service which is then started by runtime
|
||||
func (r *localRuntime) Create(s *runtime.Service, opts ...runtime.CreateOption) error {
|
||||
func (r *localRuntime) Create(resource runtime.Resource, opts ...runtime.CreateOption) error {
|
||||
var options runtime.CreateOptions
|
||||
for _, o := range opts {
|
||||
o(&options)
|
||||
@@ -90,6 +90,22 @@ func (r *localRuntime) Create(s *runtime.Service, opts ...runtime.CreateOption)
|
||||
r.Lock()
|
||||
defer r.Unlock()
|
||||
|
||||
// Handle the various different types of resources:
|
||||
switch resource.Type() {
|
||||
case runtime.TypeNamespace:
|
||||
// noop (Namespace is not supported by local)
|
||||
return nil
|
||||
case runtime.TypeNetworkPolicy:
|
||||
// noop (NetworkPolicy is not supported by local)
|
||||
return nil
|
||||
case runtime.TypeService:
|
||||
|
||||
// Assert the resource back into a *runtime.Service
|
||||
s, ok := resource.(*runtime.Service)
|
||||
if !ok {
|
||||
return runtime.ErrInvalidResource
|
||||
}
|
||||
|
||||
if len(options.Namespace) == 0 {
|
||||
options.Namespace = defaultNamespace
|
||||
}
|
||||
@@ -139,6 +155,9 @@ func (r *localRuntime) Create(s *runtime.Service, opts ...runtime.CreateOption)
|
||||
r.namespaces[options.Namespace][serviceKey(s)] = service
|
||||
|
||||
return nil
|
||||
default:
|
||||
return runtime.ErrInvalidResource
|
||||
}
|
||||
}
|
||||
|
||||
// exists returns whether the given file or directory exists
|
||||
@@ -157,11 +176,28 @@ func exists(path string) (bool, error) {
|
||||
// The reason for this is because it's hard to calculate line offset
|
||||
// as opposed to character offset.
|
||||
// This logger streams by default and only supports the `StreamCount` option.
|
||||
func (r *localRuntime) Logs(s *runtime.Service, options ...runtime.LogsOption) (runtime.Logs, error) {
|
||||
func (r *localRuntime) Logs(resource runtime.Resource, options ...runtime.LogsOption) (runtime.Logs, error) {
|
||||
lopts := runtime.LogsOptions{}
|
||||
for _, o := range options {
|
||||
o(&lopts)
|
||||
}
|
||||
|
||||
// Handle the various different types of resources:
|
||||
switch resource.Type() {
|
||||
case runtime.TypeNamespace:
|
||||
// noop (Namespace is not supported by local)
|
||||
return nil, nil
|
||||
case runtime.TypeNetworkPolicy:
|
||||
// noop (NetworkPolicy is not supported by local)
|
||||
return nil, nil
|
||||
case runtime.TypeService:
|
||||
|
||||
// Assert the resource back into a *runtime.Service
|
||||
s, ok := resource.(*runtime.Service)
|
||||
if !ok {
|
||||
return nil, runtime.ErrInvalidResource
|
||||
}
|
||||
|
||||
ret := &logStream{
|
||||
service: s.Name,
|
||||
stream: make(chan runtime.Log),
|
||||
@@ -216,6 +252,9 @@ func (r *localRuntime) Logs(s *runtime.Service, options ...runtime.LogsOption) (
|
||||
|
||||
}()
|
||||
return ret, nil
|
||||
default:
|
||||
return nil, runtime.ErrInvalidResource
|
||||
}
|
||||
}
|
||||
|
||||
type logStream struct {
|
||||
@@ -298,18 +337,36 @@ func (r *localRuntime) Read(opts ...runtime.ReadOption) ([]*runtime.Service, err
|
||||
}
|
||||
|
||||
// Update attempts to update the service
|
||||
func (r *localRuntime) Update(s *runtime.Service, opts ...runtime.UpdateOption) error {
|
||||
func (r *localRuntime) Update(resource runtime.Resource, opts ...runtime.UpdateOption) error {
|
||||
var options runtime.UpdateOptions
|
||||
for _, o := range opts {
|
||||
o(&options)
|
||||
}
|
||||
if len(options.Namespace) == 0 {
|
||||
options.Namespace = defaultNamespace
|
||||
|
||||
// Handle the various different types of resources:
|
||||
switch resource.Type() {
|
||||
case runtime.TypeNamespace:
|
||||
// noop (Namespace is not supported by local)
|
||||
return nil
|
||||
case runtime.TypeNetworkPolicy:
|
||||
// noop (NetworkPolicy is not supported by local)
|
||||
return nil
|
||||
case runtime.TypeService:
|
||||
|
||||
// Assert the resource back into a *runtime.Service
|
||||
s, ok := resource.(*runtime.Service)
|
||||
if !ok {
|
||||
return runtime.ErrInvalidResource
|
||||
}
|
||||
|
||||
if len(options.Entrypoint) > 0 {
|
||||
s.Source = filepath.Join(s.Source, options.Entrypoint)
|
||||
}
|
||||
|
||||
if len(options.Namespace) == 0 {
|
||||
options.Namespace = defaultNamespace
|
||||
}
|
||||
|
||||
r.Lock()
|
||||
srvs, ok := r.namespaces[options.Namespace]
|
||||
r.Unlock()
|
||||
@@ -333,10 +390,31 @@ func (r *localRuntime) Update(s *runtime.Service, opts ...runtime.UpdateOption)
|
||||
service.Source = s.Source
|
||||
service.Exec.Dir = s.Source
|
||||
return service.Start()
|
||||
|
||||
default:
|
||||
return runtime.ErrInvalidResource
|
||||
}
|
||||
}
|
||||
|
||||
// Delete removes the service from the runtime and stops it
|
||||
func (r *localRuntime) Delete(s *runtime.Service, opts ...runtime.DeleteOption) error {
|
||||
func (r *localRuntime) Delete(resource runtime.Resource, opts ...runtime.DeleteOption) error {
|
||||
|
||||
// Handle the various different types of resources:
|
||||
switch resource.Type() {
|
||||
case runtime.TypeNamespace:
|
||||
// noop (Namespace is not supported by local)
|
||||
return nil
|
||||
case runtime.TypeNetworkPolicy:
|
||||
// noop (NetworkPolicy is not supported by local)
|
||||
return nil
|
||||
case runtime.TypeService:
|
||||
|
||||
// Assert the resource back into a *runtime.Service
|
||||
s, ok := resource.(*runtime.Service)
|
||||
if !ok {
|
||||
return runtime.ErrInvalidResource
|
||||
}
|
||||
|
||||
r.Lock()
|
||||
defer r.Unlock()
|
||||
|
||||
@@ -376,6 +454,9 @@ func (r *localRuntime) Delete(s *runtime.Service, opts ...runtime.DeleteOption)
|
||||
delete(srvs, service.key())
|
||||
r.namespaces[options.Namespace] = srvs
|
||||
return nil
|
||||
default:
|
||||
return runtime.ErrInvalidResource
|
||||
}
|
||||
}
|
||||
|
||||
// Start starts the runtime
|
||||
@@ -461,13 +542,3 @@ func Entrypoint(dir string) (string, error) {
|
||||
return "", errors.New("More than one entrypoint found")
|
||||
}
|
||||
}
|
||||
|
||||
func (r *localRuntime) CreateNamespace(ns string) error {
|
||||
// noop
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r *localRuntime) DeleteNamespace(ns string) error {
|
||||
// noop
|
||||
return nil
|
||||
}
|
||||
|
117
runtime/resource.go
Normal file
117
runtime/resource.go
Normal file
@@ -0,0 +1,117 @@
|
||||
// Package runtime is a service runtime manager
|
||||
package runtime
|
||||
|
||||
import "fmt"
|
||||
|
||||
const (
|
||||
TypeNamespace = "namespace"
|
||||
TypeNetworkPolicy = "networkpolicy"
|
||||
TypeService = "service"
|
||||
)
|
||||
|
||||
// Resource represents any resource handled by runtime
|
||||
type Resource interface {
|
||||
String() string
|
||||
Type() string
|
||||
}
|
||||
|
||||
// Namespace represents a logical namespace for organising resources
|
||||
type Namespace struct {
|
||||
// Name of the namespace
|
||||
Name string
|
||||
}
|
||||
|
||||
// NewNamespace mints a new namespace
|
||||
func NewNamespace(name string) (*Namespace, error) {
|
||||
if name == "" {
|
||||
return nil, ErrInvalidResource
|
||||
}
|
||||
return &Namespace{
|
||||
Name: name,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// String implements Resource
|
||||
func (r *Namespace) String() string {
|
||||
return r.Name
|
||||
}
|
||||
|
||||
// Type implements Resource
|
||||
func (*Namespace) Type() string {
|
||||
return TypeNamespace
|
||||
}
|
||||
|
||||
// NetworkPolicy represents an ACL of label pairs allowing ignress to a namespace
|
||||
type NetworkPolicy struct {
|
||||
// The labels allowed ingress by this policy
|
||||
AllowedLabels map[string]string
|
||||
// Name of the network policy
|
||||
Name string
|
||||
// Namespace the network policy belongs to
|
||||
Namespace string
|
||||
}
|
||||
|
||||
// NewNetworkPolicy mints a new networkpolicy
|
||||
func NewNetworkPolicy(name, namespace string, allowedLabels map[string]string) (*NetworkPolicy, error) {
|
||||
if name == "" || namespace == "" {
|
||||
return nil, ErrInvalidResource
|
||||
}
|
||||
if allowedLabels == nil {
|
||||
allowedLabels = map[string]string{
|
||||
"origin": "micro",
|
||||
}
|
||||
}
|
||||
return &NetworkPolicy{
|
||||
AllowedLabels: allowedLabels,
|
||||
Name: name,
|
||||
Namespace: namespace,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// String implements Resource
|
||||
func (r *NetworkPolicy) String() string {
|
||||
return fmt.Sprintf("%s.%s", r.Namespace, r.Name)
|
||||
}
|
||||
|
||||
// Type implements Resource
|
||||
func (*NetworkPolicy) Type() string {
|
||||
return TypeNetworkPolicy
|
||||
}
|
||||
|
||||
// Service represents a Micro service running within a namespace
|
||||
type Service struct {
|
||||
// Name of the service
|
||||
Name string
|
||||
// Version of the service
|
||||
Version string
|
||||
// url location of source
|
||||
Source string
|
||||
// Metadata stores metadata
|
||||
Metadata map[string]string
|
||||
// Status of the service
|
||||
Status ServiceStatus
|
||||
}
|
||||
|
||||
// NewService mints a new service
|
||||
func NewService(name, version string) (*Service, error) {
|
||||
if name == "" {
|
||||
return nil, ErrInvalidResource
|
||||
}
|
||||
if version == "" {
|
||||
version = "latest"
|
||||
}
|
||||
return &Service{
|
||||
Name: name,
|
||||
Version: version,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// String implements Resource
|
||||
func (r *Service) String() string {
|
||||
return fmt.Sprintf("service://%s@%s:%s", r.Metadata["namespace"], r.Name, r.Version)
|
||||
}
|
||||
|
||||
// Type implements Resource
|
||||
func (*Service) Type() string {
|
||||
return TypeService
|
||||
}
|
65
runtime/resource_test.go
Normal file
65
runtime/resource_test.go
Normal file
@@ -0,0 +1,65 @@
|
||||
package runtime
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
func TestResources(t *testing.T) {
|
||||
|
||||
// Namespace:
|
||||
assert.Equal(t, TypeNamespace, new(Namespace).Type())
|
||||
namespace, err := NewNamespace("")
|
||||
assert.Error(t, err)
|
||||
assert.Equal(t, ErrInvalidResource, err)
|
||||
assert.Nil(t, namespace)
|
||||
|
||||
namespace, err = NewNamespace("test-namespace")
|
||||
assert.NoError(t, err)
|
||||
assert.NotNil(t, namespace)
|
||||
assert.Equal(t, TypeNamespace, namespace.Type())
|
||||
assert.Equal(t, "test-namespace", namespace.String())
|
||||
|
||||
// NetworkPolicy:
|
||||
assert.Equal(t, TypeNetworkPolicy, new(NetworkPolicy).Type())
|
||||
networkPolicy, err := NewNetworkPolicy("", "", nil)
|
||||
assert.Error(t, err)
|
||||
assert.Equal(t, ErrInvalidResource, err)
|
||||
assert.Nil(t, networkPolicy)
|
||||
|
||||
networkPolicy, err = NewNetworkPolicy("test", "", nil)
|
||||
assert.Error(t, err)
|
||||
assert.Equal(t, ErrInvalidResource, err)
|
||||
assert.Nil(t, networkPolicy)
|
||||
|
||||
networkPolicy, err = NewNetworkPolicy("", "test", nil)
|
||||
assert.Error(t, err)
|
||||
assert.Equal(t, ErrInvalidResource, err)
|
||||
assert.Nil(t, networkPolicy)
|
||||
|
||||
networkPolicy, err = NewNetworkPolicy("ingress", "test", nil)
|
||||
assert.NoError(t, err)
|
||||
assert.NotNil(t, networkPolicy)
|
||||
assert.Equal(t, TypeNetworkPolicy, networkPolicy.Type())
|
||||
assert.Equal(t, "test.ingress", networkPolicy.String())
|
||||
assert.Len(t, networkPolicy.AllowedLabels, 1)
|
||||
|
||||
networkPolicy, err = NewNetworkPolicy("ingress", "test", map[string]string{"foo": "bar", "bar": "foo"})
|
||||
assert.Len(t, networkPolicy.AllowedLabels, 2)
|
||||
|
||||
// Service:
|
||||
assert.Equal(t, TypeService, new(Service).Type())
|
||||
service, err := NewService("", "")
|
||||
assert.Error(t, err)
|
||||
assert.Equal(t, ErrInvalidResource, err)
|
||||
assert.Nil(t, service)
|
||||
|
||||
service, err = NewService("test-service", "oldest")
|
||||
service.Metadata = map[string]string{"namespace": "testing"}
|
||||
assert.NoError(t, err)
|
||||
assert.NotNil(t, service)
|
||||
assert.Equal(t, TypeService, service.Type())
|
||||
assert.Equal(t, "service://testing@test-service:oldest", service.String())
|
||||
assert.Equal(t, "oldest", service.Version)
|
||||
}
|
@@ -8,6 +8,7 @@ import (
|
||||
|
||||
var (
|
||||
ErrAlreadyExists = errors.New("already exists")
|
||||
ErrInvalidResource = errors.New("invalid resource")
|
||||
ErrNotFound = errors.New("not found")
|
||||
)
|
||||
|
||||
@@ -15,20 +16,16 @@ var (
|
||||
type Runtime interface {
|
||||
// Init initializes runtime
|
||||
Init(...Option) error
|
||||
// CreateNamespace creates a new namespace in the runtime
|
||||
CreateNamespace(string) error
|
||||
// DeleteNamespace deletes a namespace in the runtime
|
||||
DeleteNamespace(string) error
|
||||
// Create registers a service
|
||||
Create(*Service, ...CreateOption) error
|
||||
// Read returns the service
|
||||
// Create a resource
|
||||
Create(Resource, ...CreateOption) error
|
||||
// Read a resource
|
||||
Read(...ReadOption) ([]*Service, error)
|
||||
// Update the service in place
|
||||
Update(*Service, ...UpdateOption) error
|
||||
// Remove a service
|
||||
Delete(*Service, ...DeleteOption) error
|
||||
// Logs returns the logs for a service
|
||||
Logs(*Service, ...LogsOption) (Logs, error)
|
||||
// Update a resource
|
||||
Update(Resource, ...UpdateOption) error
|
||||
// Delete a resource
|
||||
Delete(Resource, ...DeleteOption) error
|
||||
// Logs returns the logs for a resource
|
||||
Logs(Resource, ...LogsOption) (Logs, error)
|
||||
// Start starts the runtime
|
||||
Start() error
|
||||
// Stop shuts down the runtime
|
||||
@@ -113,20 +110,6 @@ const (
|
||||
Error
|
||||
)
|
||||
|
||||
// Service is runtime service
|
||||
type Service struct {
|
||||
// Name of the service
|
||||
Name string
|
||||
// Version of the service
|
||||
Version string
|
||||
// url location of source
|
||||
Source string
|
||||
// Metadata stores metadata
|
||||
Metadata map[string]string
|
||||
// Status of the service
|
||||
Status ServiceStatus
|
||||
}
|
||||
|
||||
// Resources which are allocated to a serivce
|
||||
type Resources struct {
|
||||
// CPU is the maximum amount of CPU the service will be allocated (unit millicpu)
|
||||
|
@@ -168,6 +168,9 @@ func (r *Request) request() (*http.Request, error) {
|
||||
case "deployment":
|
||||
// /apis/apps/v1/namespaces/{namespace}/deployments/{name}
|
||||
url = fmt.Sprintf("%s/apis/apps/v1/namespaces/%s/%ss/", r.host, r.namespace, r.resource)
|
||||
case "networkpolicy", "networkpolicies":
|
||||
// /apis/networking.k8s.io/v1/namespaces/{namespace}/networkpolicies
|
||||
url = fmt.Sprintf("%s/apis/networking.k8s.io/v1/namespaces/%s/networkpolicies/", r.host, r.namespace)
|
||||
default:
|
||||
// /api/v1/namespaces/{namespace}/{resource}
|
||||
url = fmt.Sprintf("%s/api/v1/namespaces/%s/%ss/", r.host, r.namespace, r.resource)
|
||||
|
@@ -156,6 +156,8 @@ func (c *client) Update(r *Resource, opts ...UpdateOption) error {
|
||||
req.Body(r.Value.(*Deployment))
|
||||
case "pod":
|
||||
req.Body(r.Value.(*Pod))
|
||||
case "networkpolicy", "networkpolicies":
|
||||
req.Body(r.Value.(*NetworkPolicy))
|
||||
default:
|
||||
return errors.New("unsupported resource")
|
||||
}
|
||||
|
@@ -6,6 +6,8 @@ var templates = map[string]string{
|
||||
"namespace": namespaceTmpl,
|
||||
"secret": secretTmpl,
|
||||
"serviceaccount": serviceAccountTmpl,
|
||||
"networkpolicies": networkPolicyTmpl,
|
||||
"networkpolicy": networkPolicyTmpl,
|
||||
}
|
||||
|
||||
var deploymentTmpl = `
|
||||
@@ -239,3 +241,31 @@ imagePullSecrets:
|
||||
{{- end }}
|
||||
{{- end }}
|
||||
`
|
||||
|
||||
var networkPolicyTmpl = `
|
||||
apiVersion: networking.k8s.io/v1
|
||||
kind: NetworkPolicy
|
||||
metadata:
|
||||
name: "{{ .Metadata.Name }}"
|
||||
namespace: "{{ .Metadata.Namespace }}"
|
||||
labels:
|
||||
{{- with .Metadata.Labels }}
|
||||
{{- range $key, $value := . }}
|
||||
{{ $key }}: "{{ $value }}"
|
||||
{{- end }}
|
||||
{{- end }}
|
||||
spec:
|
||||
podSelector:
|
||||
matchLabels:
|
||||
ingress:
|
||||
- from: # Allow pods in this namespace to talk to each other
|
||||
- podSelector: {}
|
||||
- from: # Allow pods in the namespaces bearing the specified labels to talk to pods in this namespace:
|
||||
- namespaceSelector:
|
||||
matchLabels:
|
||||
{{- with .AllowedLabels }}
|
||||
{{- range $key, $value := . }}
|
||||
{{ $key }}: "{{ $value }}"
|
||||
{{- end }}
|
||||
{{- end }}
|
||||
`
|
||||
|
@@ -267,3 +267,9 @@ type VolumeMount struct {
|
||||
Name string `json:"name"`
|
||||
MountPath string `json:"mountPath"`
|
||||
}
|
||||
|
||||
// NetworkPolicy is a Kubernetes Namespace
|
||||
type NetworkPolicy struct {
|
||||
AllowedLabels map[string]string `json:"allowedLabels,omitempty"`
|
||||
Metadata *Metadata `json:"metadata,omitempty"`
|
||||
}
|
||||
|
Reference in New Issue
Block a user