站点更新：2018-01-29 12:47:06

common/server/client/main.go

@@ -0,0 +1,31 @@
+package main
+
+import (
+	"log"
+	"net/http"
+
+	"github.com/ehlxr/go-utils/common/server"
+)
+
+func main() {
+	server := server.NewServer()
+	err := server.Register(new(Hello))
+	if err != nil {
+		log.Fatal(err)
+	}
+	err = server.Start(":8080")
+	if err != nil {
+		log.Fatal(err)
+	}
+}
+
+type Hello struct {
+}
+
+func (h *Hello) Print(w http.ResponseWriter, r *http.Request) {
+	w.Write([]byte("print"))
+}
+
+func (h *Hello) Hello(w http.ResponseWriter, r *http.Request) {
+	w.Write([]byte("hello"))
+}

common/server/server.go

@@ -0,0 +1,67 @@
+package server
+
+import (
+	"fmt"
+	"log"
+	"net/http"
+	"reflect"
+	"strings"
+	"sync"
+)
+
+type Server struct {
+	name    string
+	val     reflect.Value
+	typ     reflect.Type
+	methods []reflect.Method
+	lock    sync.Mutex
+}
+
+func NewServer() *Server {
+	server := new(Server)
+	server.methods = make([]reflect.Method, 0)
+	return server
+}
+
+func (s *Server) Start(addr string) error {
+	log.Println(fmt.Sprintf("server start on addr [%s]", addr))
+	return http.ListenAndServe(addr, s)
+}
+
+func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	for _, method := range s.methods {
+		if strings.ToLower("/"+s.name+"/"+method.Name) == r.URL.Path {
+			method.Func.Call([]reflect.Value{s.val, reflect.ValueOf(w), reflect.ValueOf(r)})
+		}
+	}
+}
+
+func (s *Server) Register(service interface{}) error {
+	s.lock.Lock()
+	defer s.lock.Unlock()
+
+	s.typ = reflect.TypeOf(service)
+	s.val = reflect.ValueOf(service)
+	s.name = reflect.Indirect(s.val).Type().Name()
+	if s.name == "" {
+		return fmt.Errorf("no service name for type %s", s.typ.String())
+	}
+	for m := 0; m < s.typ.NumMethod(); m++ {
+		method := s.typ.Method(m)
+		mtype := method.Type
+		if mtype.NumIn() != 3 {
+			return fmt.Errorf("method %s has wrong number of ins: %d", method.Name, mtype.NumIn())
+		}
+		reply := mtype.In(1)
+		if reply.String() != "http.ResponseWriter" {
+			return fmt.Errorf("%s argument type not exported: %s", method.Name, reply)
+		}
+		arg := mtype.In(2)
+		if arg.String() != "*http.Request" {
+			return fmt.Errorf("%s argument type not exported: %s", method.Name, arg)
+		}
+		s.methods = append(s.methods, method)
+		log.Println("registry path:", strings.ToLower("/"+s.name+"/"+method.Name))
+	}
+	return nil
+}