add admission

pkg/registry/admission.go

@@ -0,0 +1,65 @@
+package registry
+
+import (
+	"sync"
+
+	"github.com/go-gost/gost/pkg/admission"
+)
+
+var (
+	admissionReg = &admissionRegistry{}
+)
+
+func Admission() *admissionRegistry {
+	return admissionReg
+}
+
+type admissionRegistry struct {
+	m sync.Map
+}
+
+func (r *admissionRegistry) Register(name string, admission admission.Admission) error {
+	if name == "" || admission == nil {
+		return nil
+	}
+	if _, loaded := r.m.LoadOrStore(name, admission); loaded {
+		return ErrDup
+	}
+
+	return nil
+}
+
+func (r *admissionRegistry) Unregister(name string) {
+	r.m.Delete(name)
+}
+
+func (r *admissionRegistry) IsRegistered(name string) bool {
+	_, ok := r.m.Load(name)
+	return ok
+}
+
+func (r *admissionRegistry) Get(name string) admission.Admission {
+	if name == "" {
+		return nil
+	}
+	return &admissionWrapper{name: name}
+}
+
+func (r *admissionRegistry) get(name string) admission.Admission {
+	if v, ok := r.m.Load(name); ok {
+		return v.(admission.Admission)
+	}
+	return nil
+}
+
+type admissionWrapper struct {
+	name string
+}
+
+func (w *admissionWrapper) Admit(addr string) bool {
+	p := admissionReg.get(w.name)
+	if p == nil {
+		return false
+	}
+	return p.Admit(addr)
+}

pkg/registry/service.go

@@ -18,7 +18,7 @@ type serviceRegistry struct {
 	m sync.Map
 }
 
-func (r *serviceRegistry) Register(name string, svc service.Servicer) error {
+func (r *serviceRegistry) Register(name string, svc service.Service) error {
 	if name == "" || svc == nil {
 		return nil
 	}
@@ -38,7 +38,7 @@ func (r *serviceRegistry) IsRegistered(name string) bool {
 	return ok
 }
 
-func (r *serviceRegistry) Get(name string) service.Servicer {
+func (r *serviceRegistry) Get(name string) service.Service {
 	if name == "" {
 		return nil
 	}
@@ -46,5 +46,5 @@ func (r *serviceRegistry) Get(name string) service.Servicer {
 	if !ok {
 		return nil
 	}
-	return v.(service.Servicer)
+	return v.(service.Service)
 }