Merge branch 'master' into pool
This commit is contained in:
		
							
								
								
									
										4
									
								
								broker/doc.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										4
									
								
								broker/doc.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,4 @@ | |||||||
|  | /* | ||||||
|  | Package broker is an interface used for asynchronous messaging. | ||||||
|  | */ | ||||||
|  | package broker | ||||||
| @@ -1,24 +1,3 @@ | |||||||
| /* |  | ||||||
| Package client provides a method to make synchronous, asynchronous and |  | ||||||
| streaming requests to services. By default json and protobuf codecs are |  | ||||||
| supported. |  | ||||||
|  |  | ||||||
| 	import "github.com/micro/go-micro/client" |  | ||||||
|  |  | ||||||
| 	c := client.NewClient() |  | ||||||
|  |  | ||||||
| 	req := c.NewRequest("go.micro.srv.greeter", "Greeter.Hello", &greeter.Request{ |  | ||||||
| 		Name: "John", |  | ||||||
| 	}) |  | ||||||
|  |  | ||||||
| 	rsp := &greeter.Response{} |  | ||||||
|  |  | ||||||
| 	if err := c.Call(context.Background(), req, rsp); err != nil { |  | ||||||
| 		return err |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	fmt.Println(rsp.Msg) |  | ||||||
| */ |  | ||||||
| package client | package client | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|   | |||||||
							
								
								
									
										23
									
								
								client/doc.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										23
									
								
								client/doc.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,23 @@ | |||||||
|  | /* | ||||||
|  | Package client is an interface for making requests. | ||||||
|  |  | ||||||
|  | It provides a method to make synchronous, asynchronous and streaming requests to services. | ||||||
|  | By default json and protobuf codecs are supported. | ||||||
|  |  | ||||||
|  |         import "github.com/micro/go-micro/client" | ||||||
|  |  | ||||||
|  |         c := client.NewClient() | ||||||
|  |  | ||||||
|  |         req := c.NewRequest("go.micro.srv.greeter", "Greeter.Hello", &greeter.Request{ | ||||||
|  |                 Name: "John", | ||||||
|  |         }) | ||||||
|  |  | ||||||
|  |         rsp := &greeter.Response{} | ||||||
|  |  | ||||||
|  |         if err := c.Call(context.Background(), req, rsp); err != nil { | ||||||
|  |                 return err | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         fmt.Println(rsp.Msg) | ||||||
|  | */ | ||||||
|  | package client | ||||||
| @@ -123,6 +123,8 @@ func (m *MockClient) CallRemote(ctx context.Context, addr string, req client.Req | |||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		v.Set(reflect.ValueOf(r.Response)) | 		v.Set(reflect.ValueOf(r.Response)) | ||||||
|  | 		 | ||||||
|  | 		return nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	return fmt.Errorf("rpc: can't find service %s", req.Method()) | 	return fmt.Errorf("rpc: can't find service %s", req.Method()) | ||||||
|   | |||||||
							
								
								
									
										4
									
								
								cmd/doc.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										4
									
								
								cmd/doc.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,4 @@ | |||||||
|  | /* | ||||||
|  | Package cmd is an interface for parsing the command line. | ||||||
|  | */ | ||||||
|  | package cmd | ||||||
							
								
								
									
										4
									
								
								codec/doc.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										4
									
								
								codec/doc.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,4 @@ | |||||||
|  | /* | ||||||
|  | Package codec is an interface for encoding messages. | ||||||
|  | */ | ||||||
|  | package codec | ||||||
							
								
								
									
										4
									
								
								errors/doc.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										4
									
								
								errors/doc.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,4 @@ | |||||||
|  | /* | ||||||
|  | Package errors is an interface for defining detailed errors. | ||||||
|  | */ | ||||||
|  | package errors | ||||||
							
								
								
									
										4
									
								
								metadata/doc.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										4
									
								
								metadata/doc.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,4 @@ | |||||||
|  | /* | ||||||
|  | Package metadata is a way of defining message headers. | ||||||
|  | */ | ||||||
|  | package metadata | ||||||
							
								
								
									
										4
									
								
								registry/doc.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										4
									
								
								registry/doc.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,4 @@ | |||||||
|  | /* | ||||||
|  | Package registry is an interface for service discovery. | ||||||
|  | */ | ||||||
|  | package registry | ||||||
| @@ -45,10 +45,6 @@ func encodeEndpoints(en []*Endpoint) []string { | |||||||
| 	var tags []string | 	var tags []string | ||||||
| 	for _, e := range en { | 	for _, e := range en { | ||||||
| 		if b, err := json.Marshal(e); err == nil { | 		if b, err := json.Marshal(e); err == nil { | ||||||
| 			// old encoding |  | ||||||
| 			// TODO: remove in 09/2016 |  | ||||||
| 			tags = append(tags, "e="+string(b)) |  | ||||||
| 			// new encoding |  | ||||||
| 			tags = append(tags, "e-"+encode(b)) | 			tags = append(tags, "e-"+encode(b)) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| @@ -100,9 +96,6 @@ func encodeMetadata(md map[string]string) []string { | |||||||
| 		if b, err := json.Marshal(map[string]string{ | 		if b, err := json.Marshal(map[string]string{ | ||||||
| 			k: v, | 			k: v, | ||||||
| 		}); err == nil { | 		}); err == nil { | ||||||
| 			// old encoding |  | ||||||
| 			// TODO: remove in 09/2016 |  | ||||||
| 			tags = append(tags, "t="+string(b)) |  | ||||||
| 			// new encoding | 			// new encoding | ||||||
| 			tags = append(tags, "t-"+encode(b)) | 			tags = append(tags, "t-"+encode(b)) | ||||||
| 		} | 		} | ||||||
| @@ -152,13 +145,7 @@ func decodeMetadata(tags []string) map[string]string { | |||||||
| } | } | ||||||
|  |  | ||||||
| func encodeVersion(v string) []string { | func encodeVersion(v string) []string { | ||||||
| 	return []string{ | 	return []string{"v-" + encode([]byte(v))} | ||||||
| 		// old encoding, |  | ||||||
| 		// TODO: remove in 09/2016 |  | ||||||
| 		"v=" + v, |  | ||||||
| 		// new encoding, |  | ||||||
| 		"v-" + encode([]byte(v)), |  | ||||||
| 	} |  | ||||||
| } | } | ||||||
|  |  | ||||||
| func decodeVersion(tags []string) (string, bool) { | func decodeVersion(tags []string) (string, bool) { | ||||||
|   | |||||||
| @@ -57,8 +57,8 @@ func TestEncodingEndpoints(t *testing.T) { | |||||||
| 		e := encodeEndpoints([]*Endpoint{ep}) | 		e := encodeEndpoints([]*Endpoint{ep}) | ||||||
|  |  | ||||||
| 		// check there are two tags; old and new | 		// check there are two tags; old and new | ||||||
| 		if len(e) != 2 { | 		if len(e) != 1 { | ||||||
| 			t.Fatalf("Expected 2 encoded tags, got %v", e) | 			t.Fatalf("Expected 1 encoded tags, got %v", e) | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		// check old encoding | 		// check old encoding | ||||||
| @@ -104,12 +104,7 @@ func TestEncodingEndpoints(t *testing.T) { | |||||||
| 		// HEX encoded | 		// HEX encoded | ||||||
| 		hencoded := encode(jencoded) | 		hencoded := encode(jencoded) | ||||||
| 		// endpoint tag | 		// endpoint tag | ||||||
| 		jepTag := "e=" + string(jencoded) |  | ||||||
| 		hepTag := "e-" + hencoded | 		hepTag := "e-" + hencoded | ||||||
|  |  | ||||||
| 		// test old |  | ||||||
| 		testEp(ep, jepTag) |  | ||||||
| 		// test new |  | ||||||
| 		testEp(ep, hepTag) | 		testEp(ep, hepTag) | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
| @@ -118,16 +113,15 @@ func TestEncodingVersion(t *testing.T) { | |||||||
| 	testData := []struct { | 	testData := []struct { | ||||||
| 		decoded string | 		decoded string | ||||||
| 		encoded string | 		encoded string | ||||||
| 		oldEncoded string |  | ||||||
| 	}{ | 	}{ | ||||||
| 		{"1.0.0", "v-789c32d433d03300040000ffff02ce00ee", "v=1.0.0"}, | 		{"1.0.0", "v-789c32d433d03300040000ffff02ce00ee"}, | ||||||
| 		{"latest", "v-789cca492c492d2e01040000ffff08cc028e", "v=latest"}, | 		{"latest", "v-789cca492c492d2e01040000ffff08cc028e"}, | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	for _, data := range testData { | 	for _, data := range testData { | ||||||
| 		e := encodeVersion(data.decoded) | 		e := encodeVersion(data.decoded) | ||||||
|  |  | ||||||
| 		if e[1] != data.encoded { | 		if e[0] != data.encoded { | ||||||
| 			t.Fatalf("Expected %s got %s", data.encoded, e) | 			t.Fatalf("Expected %s got %s", data.encoded, e) | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| @@ -148,14 +142,5 @@ func TestEncodingVersion(t *testing.T) { | |||||||
| 		if d != data.decoded { | 		if d != data.decoded { | ||||||
| 			t.Fatalf("Expected %s got %s", data.decoded, d) | 			t.Fatalf("Expected %s got %s", data.decoded, d) | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		d, ok = decodeVersion([]string{data.oldEncoded}) |  | ||||||
| 		if !ok { |  | ||||||
| 			t.Fatalf("Unexpected %t for %s", ok, data.oldEncoded) |  | ||||||
| 		} |  | ||||||
|  |  | ||||||
| 		if d != data.decoded { |  | ||||||
| 			t.Fatalf("Expected %s got %s", data.decoded, d) |  | ||||||
| 		} |  | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -90,7 +90,7 @@ func TestBlackList(t *testing.T) { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// blacklist all of it | 	// blacklist all of it | ||||||
| 	for i := 0; i < 9; i++ { | 	for i := 0; i < 20; i++ { | ||||||
| 		node, err = next() | 		node, err = next() | ||||||
| 		if err != nil { | 		if err != nil { | ||||||
| 			t.Fatal(err) | 			t.Fatal(err) | ||||||
|   | |||||||
							
								
								
									
										8
									
								
								selector/doc.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										8
									
								
								selector/doc.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,8 @@ | |||||||
|  | /* | ||||||
|  | Package selector is a way to load balance service nodes. | ||||||
|  |  | ||||||
|  | It algorithmically filter and return nodes required by the client or any other system. | ||||||
|  | Selector's implemented by Micro build on the registry but it's of optional use. One could | ||||||
|  | provide a static Selector that has a fixed pool. | ||||||
|  | */ | ||||||
|  | package selector | ||||||
| @@ -1,9 +1,3 @@ | |||||||
| /* |  | ||||||
| The Selector package provides a way to algorithmically filter and return |  | ||||||
| nodes required by the client or any other system. Selector's implemented |  | ||||||
| by Micro build on the registry but it's of optional use. One could |  | ||||||
| provide a static Selector that has a fixed pool. |  | ||||||
| */ |  | ||||||
| package selector | package selector | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|   | |||||||
| @@ -2,9 +2,6 @@ package server | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"github.com/micro/go-micro/server/debug" | 	"github.com/micro/go-micro/server/debug" | ||||||
| 	proto "github.com/micro/go-micro/server/debug/proto" |  | ||||||
|  |  | ||||||
| 	"golang.org/x/net/context" |  | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // We use this to wrap any debug handlers so we preserve the signature Debug.{Method} | // We use this to wrap any debug handlers so we preserve the signature Debug.{Method} | ||||||
| @@ -12,10 +9,6 @@ type Debug struct { | |||||||
| 	debug.DebugHandler | 	debug.DebugHandler | ||||||
| } | } | ||||||
|  |  | ||||||
| func (d *Debug) Health(ctx context.Context, req *proto.HealthRequest, rsp *proto.HealthResponse) error { |  | ||||||
| 	return d.DebugHandler.Health(ctx, req, rsp) |  | ||||||
| } |  | ||||||
|  |  | ||||||
| func registerDebugHandler(s Server) { | func registerDebugHandler(s Server) { | ||||||
| 	s.Handle(s.NewHandler(&Debug{s.Options().DebugHandler}, InternalHandler(true))) | 	s.Handle(s.NewHandler(&Debug{s.Options().DebugHandler}, InternalHandler(true))) | ||||||
| } | } | ||||||
|   | |||||||
| @@ -1,6 +1,9 @@ | |||||||
| package debug | package debug | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|  | 	"runtime" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
| 	proto "github.com/micro/go-micro/server/debug/proto" | 	proto "github.com/micro/go-micro/server/debug/proto" | ||||||
|  |  | ||||||
| 	"golang.org/x/net/context" | 	"golang.org/x/net/context" | ||||||
| @@ -12,16 +15,37 @@ import ( | |||||||
| // and /varz | // and /varz | ||||||
| type DebugHandler interface { | type DebugHandler interface { | ||||||
| 	Health(ctx context.Context, req *proto.HealthRequest, rsp *proto.HealthResponse) error | 	Health(ctx context.Context, req *proto.HealthRequest, rsp *proto.HealthResponse) error | ||||||
|  | 	Stats(ctx context.Context, req *proto.StatsRequest, rsp *proto.StatsResponse) error | ||||||
| } | } | ||||||
|  |  | ||||||
| // Our own internal handler | // Our own internal handler | ||||||
| type debug struct{} | type debug struct { | ||||||
|  | 	started int64 | ||||||
|  | } | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| 	DefaultDebugHandler DebugHandler = new(debug) | 	DefaultDebugHandler DebugHandler = newDebug() | ||||||
| ) | ) | ||||||
|  |  | ||||||
|  | func newDebug() *debug { | ||||||
|  | 	return &debug{ | ||||||
|  | 		started: time.Now().Unix(), | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| func (d *debug) Health(ctx context.Context, req *proto.HealthRequest, rsp *proto.HealthResponse) error { | func (d *debug) Health(ctx context.Context, req *proto.HealthRequest, rsp *proto.HealthResponse) error { | ||||||
| 	rsp.Status = "ok" | 	rsp.Status = "ok" | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func (d *debug) Stats(ctx context.Context, req *proto.StatsRequest, rsp *proto.StatsResponse) error { | ||||||
|  | 	var mstat runtime.MemStats | ||||||
|  | 	runtime.ReadMemStats(&mstat) | ||||||
|  |  | ||||||
|  | 	rsp.Started = uint64(d.started) | ||||||
|  | 	rsp.Uptime = uint64(time.Now().Unix() - d.started) | ||||||
|  | 	rsp.Memory = mstat.Alloc | ||||||
|  | 	rsp.Gc = mstat.PauseTotalNs | ||||||
|  | 	rsp.Threads = uint64(runtime.NumGoroutine()) | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|   | |||||||
| @@ -1,16 +1,18 @@ | |||||||
| // Code generated by protoc-gen-go. | // Code generated by protoc-gen-go. | ||||||
| // source: go-micro/server/debug/proto/debug.proto | // source: github.com/micro/go-micro/server/debug/proto/debug.proto | ||||||
| // DO NOT EDIT! | // DO NOT EDIT! | ||||||
|  |  | ||||||
| /* | /* | ||||||
| Package debug is a generated protocol buffer package. | Package debug is a generated protocol buffer package. | ||||||
|  |  | ||||||
| It is generated from these files: | It is generated from these files: | ||||||
| 	go-micro/server/debug/proto/debug.proto | 	github.com/micro/go-micro/server/debug/proto/debug.proto | ||||||
|  |  | ||||||
| It has these top-level messages: | It has these top-level messages: | ||||||
| 	HealthRequest | 	HealthRequest | ||||||
| 	HealthResponse | 	HealthResponse | ||||||
|  | 	StatsRequest | ||||||
|  | 	StatsResponse | ||||||
| */ | */ | ||||||
| package debug | package debug | ||||||
|  |  | ||||||
| @@ -23,6 +25,12 @@ var _ = proto.Marshal | |||||||
| var _ = fmt.Errorf | var _ = fmt.Errorf | ||||||
| var _ = math.Inf | var _ = math.Inf | ||||||
|  |  | ||||||
|  | // This is a compile-time assertion to ensure that this generated file | ||||||
|  | // is compatible with the proto package it is being compiled against. | ||||||
|  | // A compilation error at this line likely means your copy of the | ||||||
|  | // proto package needs to be updated. | ||||||
|  | const _ = proto.ProtoPackageIsVersion2 // please upgrade the proto package | ||||||
|  |  | ||||||
| type HealthRequest struct { | type HealthRequest struct { | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -32,6 +40,7 @@ func (*HealthRequest) ProtoMessage()               {} | |||||||
| func (*HealthRequest) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{0} } | func (*HealthRequest) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{0} } | ||||||
|  |  | ||||||
| type HealthResponse struct { | type HealthResponse struct { | ||||||
|  | 	// default: ok | ||||||
| 	Status string `protobuf:"bytes,1,opt,name=status" json:"status,omitempty"` | 	Status string `protobuf:"bytes,1,opt,name=status" json:"status,omitempty"` | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -40,18 +49,52 @@ func (m *HealthResponse) String() string            { return proto.CompactTextSt | |||||||
| func (*HealthResponse) ProtoMessage()               {} | func (*HealthResponse) ProtoMessage()               {} | ||||||
| func (*HealthResponse) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{1} } | func (*HealthResponse) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{1} } | ||||||
|  |  | ||||||
|  | type StatsRequest struct { | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *StatsRequest) Reset()                    { *m = StatsRequest{} } | ||||||
|  | func (m *StatsRequest) String() string            { return proto.CompactTextString(m) } | ||||||
|  | func (*StatsRequest) ProtoMessage()               {} | ||||||
|  | func (*StatsRequest) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{2} } | ||||||
|  |  | ||||||
|  | type StatsResponse struct { | ||||||
|  | 	// unix timestamp | ||||||
|  | 	Started uint64 `protobuf:"varint,1,opt,name=started" json:"started,omitempty"` | ||||||
|  | 	// in seconds | ||||||
|  | 	Uptime uint64 `protobuf:"varint,2,opt,name=uptime" json:"uptime,omitempty"` | ||||||
|  | 	// in bytes | ||||||
|  | 	Memory uint64 `protobuf:"varint,3,opt,name=memory" json:"memory,omitempty"` | ||||||
|  | 	// num threads | ||||||
|  | 	Threads uint64 `protobuf:"varint,4,opt,name=threads" json:"threads,omitempty"` | ||||||
|  | 	// in nanoseconds | ||||||
|  | 	Gc uint64 `protobuf:"varint,5,opt,name=gc" json:"gc,omitempty"` | ||||||
|  | } | ||||||
|  |  | ||||||
|  | func (m *StatsResponse) Reset()                    { *m = StatsResponse{} } | ||||||
|  | func (m *StatsResponse) String() string            { return proto.CompactTextString(m) } | ||||||
|  | func (*StatsResponse) ProtoMessage()               {} | ||||||
|  | func (*StatsResponse) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{3} } | ||||||
|  |  | ||||||
| func init() { | func init() { | ||||||
| 	proto.RegisterType((*HealthRequest)(nil), "HealthRequest") | 	proto.RegisterType((*HealthRequest)(nil), "HealthRequest") | ||||||
| 	proto.RegisterType((*HealthResponse)(nil), "HealthResponse") | 	proto.RegisterType((*HealthResponse)(nil), "HealthResponse") | ||||||
|  | 	proto.RegisterType((*StatsRequest)(nil), "StatsRequest") | ||||||
|  | 	proto.RegisterType((*StatsResponse)(nil), "StatsResponse") | ||||||
| } | } | ||||||
|  |  | ||||||
| var fileDescriptor0 = []byte{ | var fileDescriptor0 = []byte{ | ||||||
| 	// 107 bytes of a gzipped FileDescriptorProto | 	// 201 bytes of a gzipped FileDescriptorProto | ||||||
| 	0x1f, 0x8b, 0x08, 0x00, 0x00, 0x09, 0x6e, 0x88, 0x02, 0xff, 0xe2, 0x52, 0x4f, 0xcf, 0xd7, 0xcd, | 	0x1f, 0x8b, 0x08, 0x00, 0x00, 0x09, 0x6e, 0x88, 0x02, 0xff, 0x34, 0x8f, 0xbd, 0x6e, 0x83, 0x30, | ||||||
| 	0xcd, 0x4c, 0x2e, 0xca, 0xd7, 0x2f, 0x4e, 0x2d, 0x2a, 0x4b, 0x2d, 0xd2, 0x4f, 0x49, 0x4d, 0x2a, | 	0x14, 0x85, 0x05, 0xa5, 0x54, 0xbd, 0x2a, 0x54, 0x62, 0xa8, 0x3c, 0x56, 0x4c, 0x2c, 0xc5, 0x43, | ||||||
| 	0x4d, 0xd7, 0x2f, 0x28, 0xca, 0x2f, 0xc9, 0x87, 0xb0, 0xf5, 0xc0, 0x6c, 0x25, 0x7e, 0x2e, 0x5e, | 	0x97, 0x3e, 0x42, 0x67, 0xf2, 0x04, 0xfc, 0x5c, 0x19, 0xa4, 0x38, 0x26, 0xbe, 0xd7, 0x91, 0x32, | ||||||
| 	0x8f, 0xd4, 0xc4, 0x9c, 0x92, 0x8c, 0xa0, 0xd4, 0xc2, 0xd2, 0xd4, 0xe2, 0x12, 0x25, 0x05, 0x2e, | 	0xe7, 0xc5, 0x03, 0xb6, 0xd9, 0xce, 0xf7, 0xd9, 0xe7, 0x48, 0x17, 0xfe, 0xd4, 0xc2, 0xb3, 0x1b, | ||||||
| 	0x3e, 0x98, 0x40, 0x71, 0x41, 0x7e, 0x5e, 0x71, 0xaa, 0x10, 0x1f, 0x17, 0x5b, 0x71, 0x49, 0x62, | 	0xda, 0xd1, 0x68, 0xa9, 0x97, 0xd1, 0x1a, 0xa9, 0xcc, 0x4f, 0x08, 0x84, 0xf6, 0x86, 0x56, 0x4e, | ||||||
| 	0x49, 0x69, 0xb1, 0x04, 0xa3, 0x02, 0xa3, 0x06, 0x67, 0x12, 0x1b, 0x58, 0xa7, 0x31, 0x20, 0x00, | 	0x38, 0x38, 0x25, 0x57, 0x6b, 0xd8, 0x84, 0xdc, 0xfa, 0x5c, 0x7f, 0x42, 0xf1, 0x8f, 0xfd, 0x99, | ||||||
| 	0x00, 0xff, 0xff, 0x1c, 0xef, 0x98, 0xac, 0x64, 0x00, 0x00, 0x00, | 	0xe7, 0x0e, 0xaf, 0x0e, 0x89, 0xeb, 0x06, 0xca, 0x43, 0xd0, 0x6a, 0x2e, 0x84, 0xd5, 0x17, 0xe4, | ||||||
|  | 	0xc4, 0x3d, 0x3b, 0x12, 0xc9, 0x77, 0xd2, 0xbc, 0x77, 0x91, 0xea, 0x12, 0x3e, 0x4e, 0x5b, 0xa2, | ||||||
|  | 	0xa3, 0xf9, 0x48, 0xa0, 0x88, 0x22, 0x36, 0x05, 0xbc, 0x6d, 0x7f, 0x2d, 0xe3, 0xe4, 0xab, 0x59, | ||||||
|  | 	0x77, 0xe0, 0xbe, 0xe9, 0x56, 0x5e, 0x34, 0x8a, 0xd4, 0x3f, 0x44, 0xda, 0xbd, 0x46, 0x6d, 0xec, | ||||||
|  | 	0x5d, 0xbc, 0x04, 0x1f, 0x68, 0x5f, 0xe2, 0xd9, 0x62, 0x3f, 0x91, 0xc8, 0xc2, 0x52, 0xc4, 0xaa, | ||||||
|  | 	0x84, 0x54, 0x8d, 0xe2, 0xd5, 0xcb, 0x2d, 0x0d, 0xb9, 0xbf, 0xeb, 0xf7, 0x19, 0x00, 0x00, 0xff, | ||||||
|  | 	0xff, 0xc6, 0x75, 0x51, 0x35, 0x13, 0x01, 0x00, 0x00, | ||||||
| } | } | ||||||
|   | |||||||
| @@ -6,11 +6,29 @@ syntax = "proto3"; | |||||||
| // | // | ||||||
| // service Debug { | // service Debug { | ||||||
| // 	rpc Health(HealthRequest) returns (HealthResponse) {} | // 	rpc Health(HealthRequest) returns (HealthResponse) {} | ||||||
|  | // 	rpc Stats(StatsRequest) returns (StatsResponse) {} | ||||||
| // } | // } | ||||||
|  |  | ||||||
| message HealthRequest { | message HealthRequest { | ||||||
| } | } | ||||||
|  |  | ||||||
| message HealthResponse { | message HealthResponse { | ||||||
|  | 	// default: ok | ||||||
| 	string status = 1; | 	string status = 1; | ||||||
| } | } | ||||||
|  |  | ||||||
|  | message StatsRequest { | ||||||
|  | } | ||||||
|  |  | ||||||
|  | message StatsResponse { | ||||||
|  | 	// unix timestamp | ||||||
|  | 	uint64 started = 1; | ||||||
|  | 	// in seconds | ||||||
|  | 	uint64 uptime = 2; | ||||||
|  | 	// in bytes | ||||||
|  | 	uint64 memory = 3; | ||||||
|  | 	// num threads | ||||||
|  | 	uint64 threads = 4; | ||||||
|  | 	// total gc in nanoseconds | ||||||
|  | 	uint64 gc = 5; | ||||||
|  | } | ||||||
|   | |||||||
							
								
								
									
										31
									
								
								server/doc.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										31
									
								
								server/doc.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,31 @@ | |||||||
|  | /* | ||||||
|  | Package server is an interface for a micro server. | ||||||
|  |  | ||||||
|  | It represents a server instance in go-micro which handles synchronous | ||||||
|  | requests via handlers and asynchronous requests via subscribers that | ||||||
|  | register with a broker. | ||||||
|  |  | ||||||
|  | The server combines the all the packages in go-micro to create a whole unit | ||||||
|  | used for building applications including discovery, client/server communication | ||||||
|  | and pub/sub. | ||||||
|  |  | ||||||
|  |         import "github.com/micro/go-micro/server" | ||||||
|  |  | ||||||
|  |         type Greeter struct {} | ||||||
|  |  | ||||||
|  |         func (g *Greeter) Hello(ctx context.Context, req *greeter.Request, rsp *greeter.Response) error { | ||||||
|  |                 rsp.Msg = "Hello " + req.Name | ||||||
|  |                 return nil | ||||||
|  |         } | ||||||
|  |  | ||||||
|  |         s := server.NewServer() | ||||||
|  |  | ||||||
|  |  | ||||||
|  |         s.Handle( | ||||||
|  |                 s.NewHandler(&Greeter{}), | ||||||
|  |         ) | ||||||
|  |  | ||||||
|  |         s.Start() | ||||||
|  |  | ||||||
|  | */ | ||||||
|  | package server | ||||||
| @@ -34,6 +34,7 @@ type Subscriber interface { | |||||||
|  |  | ||||||
| type HandlerOptions struct { | type HandlerOptions struct { | ||||||
| 	Internal bool | 	Internal bool | ||||||
|  | 	Metadata map[string]map[string]string | ||||||
| } | } | ||||||
|  |  | ||||||
| type SubscriberOptions struct { | type SubscriberOptions struct { | ||||||
| @@ -41,6 +42,14 @@ type SubscriberOptions struct { | |||||||
| 	Internal bool | 	Internal bool | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // EndpointMetadata is a Handler option that allows metadata to be added to | ||||||
|  | // individual endpoints. | ||||||
|  | func EndpointMetadata(name string, md map[string]string) HandlerOption { | ||||||
|  | 	return func(o *HandlerOptions) { | ||||||
|  | 		o.Metadata[name] = md | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
| // Internal Handler options specifies that a handler is not advertised | // Internal Handler options specifies that a handler is not advertised | ||||||
| // to the discovery system. In the future this may also limit request | // to the discovery system. In the future this may also limit request | ||||||
| // to the internal network or authorised user. | // to the internal network or authorised user. | ||||||
|   | |||||||
| @@ -14,7 +14,10 @@ type rpcHandler struct { | |||||||
| } | } | ||||||
|  |  | ||||||
| func newRpcHandler(handler interface{}, opts ...HandlerOption) Handler { | func newRpcHandler(handler interface{}, opts ...HandlerOption) Handler { | ||||||
| 	var options HandlerOptions | 	options := HandlerOptions{ | ||||||
|  | 		Metadata: make(map[string]map[string]string), | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&options) | 		o(&options) | ||||||
| 	} | 	} | ||||||
| @@ -28,6 +31,11 @@ func newRpcHandler(handler interface{}, opts ...HandlerOption) Handler { | |||||||
| 	for m := 0; m < typ.NumMethod(); m++ { | 	for m := 0; m < typ.NumMethod(); m++ { | ||||||
| 		if e := extractEndpoint(typ.Method(m)); e != nil { | 		if e := extractEndpoint(typ.Method(m)); e != nil { | ||||||
| 			e.Name = name + "." + e.Name | 			e.Name = name + "." + e.Name | ||||||
|  |  | ||||||
|  | 			for k, v := range options.Metadata[e.Name] { | ||||||
|  | 				e.Metadata[k] = v | ||||||
|  | 			} | ||||||
|  |  | ||||||
| 			endpoints = append(endpoints, e) | 			endpoints = append(endpoints, e) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
|   | |||||||
| @@ -1,31 +1,3 @@ | |||||||
| /* |  | ||||||
| Server represents a server instance in go-micro which handles synchronous |  | ||||||
| requests via handlers and asynchronous requests via subscribers that |  | ||||||
| register with a broker. |  | ||||||
|  |  | ||||||
| The server combines the all the packages in go-micro to create a whole unit |  | ||||||
| used for building applications including discovery, client/server communication |  | ||||||
| and pub/sub. |  | ||||||
|  |  | ||||||
| 	import "github.com/micro/go-micro/server" |  | ||||||
|  |  | ||||||
| 	type Greeter struct {} |  | ||||||
|  |  | ||||||
| 	func (g *Greeter) Hello(ctx context.Context, req *greeter.Request, rsp *greeter.Response) error { |  | ||||||
| 		rsp.Msg = "Hello " + req.Name |  | ||||||
| 		return nil |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	s := server.NewServer() |  | ||||||
|  |  | ||||||
|  |  | ||||||
| 	s.Handle( |  | ||||||
| 		s.NewHandler(&Greeter{}), |  | ||||||
| 	) |  | ||||||
|  |  | ||||||
| 	s.Start() |  | ||||||
|  |  | ||||||
| */ |  | ||||||
| package server | package server | ||||||
|  |  | ||||||
| import ( | import ( | ||||||
|   | |||||||
							
								
								
									
										4
									
								
								transport/doc.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										4
									
								
								transport/doc.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,4 @@ | |||||||
|  | /* | ||||||
|  | Package is an interface for synchronous communication. | ||||||
|  | */ | ||||||
|  | package transport | ||||||
| @@ -409,7 +409,19 @@ func (h *httpTransport) Listen(addr string, opts ...ListenOption) (Listener, err | |||||||
|  |  | ||||||
| 		fn := func(addr string) (net.Listener, error) { | 		fn := func(addr string) (net.Listener, error) { | ||||||
| 			if config == nil { | 			if config == nil { | ||||||
| 				cert, err := mls.Certificate(addr) | 				hosts := []string{addr} | ||||||
|  |  | ||||||
|  | 				// check if its a valid host:port | ||||||
|  | 				if host, _, err := net.SplitHostPort(addr); err == nil { | ||||||
|  | 					if len(host) == 0 { | ||||||
|  | 						hosts = getIPAddrs() | ||||||
|  | 					} else { | ||||||
|  | 						hosts = []string{host} | ||||||
|  | 					} | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 				// generate a certificate | ||||||
|  | 				cert, err := mls.Certificate(hosts...) | ||||||
| 				if err != nil { | 				if err != nil { | ||||||
| 					return nil, err | 					return nil, err | ||||||
| 				} | 				} | ||||||
| @@ -440,6 +452,45 @@ func (h *httpTransport) String() string { | |||||||
| 	return "http" | 	return "http" | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func getIPAddrs() []string { | ||||||
|  | 	ifaces, err := net.Interfaces() | ||||||
|  | 	if err != nil { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	var ipAddrs []string | ||||||
|  |  | ||||||
|  | 	for _, i := range ifaces { | ||||||
|  | 		addrs, err := i.Addrs() | ||||||
|  | 		if err != nil { | ||||||
|  | 			continue | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		for _, addr := range addrs { | ||||||
|  | 			var ip net.IP | ||||||
|  | 			switch v := addr.(type) { | ||||||
|  | 			case *net.IPNet: | ||||||
|  | 				ip = v.IP | ||||||
|  | 			case *net.IPAddr: | ||||||
|  | 				ip = v.IP | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			if ip == nil { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			ip = ip.To4() | ||||||
|  | 			if ip == nil { | ||||||
|  | 				continue | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			ipAddrs = append(ipAddrs, ip.String()) | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return ipAddrs | ||||||
|  | } | ||||||
|  |  | ||||||
| func newHTTPTransport(opts ...Option) *httpTransport { | func newHTTPTransport(opts ...Option) *httpTransport { | ||||||
| 	var options Options | 	var options Options | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
|   | |||||||
| @@ -2,7 +2,11 @@ package mock | |||||||
|  |  | ||||||
| import ( | import ( | ||||||
| 	"errors" | 	"errors" | ||||||
|  | 	"fmt" | ||||||
|  | 	"math/rand" | ||||||
|  | 	"strings" | ||||||
| 	"sync" | 	"sync" | ||||||
|  | 	"time" | ||||||
|  |  | ||||||
| 	"github.com/micro/go-micro/transport" | 	"github.com/micro/go-micro/transport" | ||||||
| ) | ) | ||||||
| @@ -59,7 +63,12 @@ func (ms *mockSocket) Send(m *transport.Message) error { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (ms *mockSocket) Close() error { | func (ms *mockSocket) Close() error { | ||||||
|  | 	select { | ||||||
|  | 	case <-ms.exit: | ||||||
|  | 		return nil | ||||||
|  | 	default: | ||||||
| 		close(ms.exit) | 		close(ms.exit) | ||||||
|  | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -68,7 +77,12 @@ func (m *mockListener) Addr() string { | |||||||
| } | } | ||||||
|  |  | ||||||
| func (m *mockListener) Close() error { | func (m *mockListener) Close() error { | ||||||
|  | 	select { | ||||||
|  | 	case <-m.exit: | ||||||
|  | 		return nil | ||||||
|  | 	default: | ||||||
| 		close(m.exit) | 		close(m.exit) | ||||||
|  | 	} | ||||||
| 	return nil | 	return nil | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -126,15 +140,25 @@ func (m *mockTransport) Listen(addr string, opts ...transport.ListenOption) (tra | |||||||
| 	m.Lock() | 	m.Lock() | ||||||
| 	defer m.Unlock() | 	defer m.Unlock() | ||||||
|  |  | ||||||
| 	if _, ok := m.listeners[addr]; ok { |  | ||||||
| 		return nil, errors.New("already listening on " + addr) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	var options transport.ListenOptions | 	var options transport.ListenOptions | ||||||
| 	for _, o := range opts { | 	for _, o := range opts { | ||||||
| 		o(&options) | 		o(&options) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|  | 	parts := strings.Split(addr, ":") | ||||||
|  |  | ||||||
|  | 	// if zero port then randomly assign one | ||||||
|  | 	if len(parts) > 1 && parts[len(parts)-1] == "0" { | ||||||
|  | 		r := rand.New(rand.NewSource(time.Now().UnixNano())) | ||||||
|  | 		i := r.Intn(10000) | ||||||
|  | 		// set addr with port | ||||||
|  | 		addr = fmt.Sprintf("%s:%d", parts[:len(parts)-1], 10000+i) | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if _, ok := m.listeners[addr]; ok { | ||||||
|  | 		return nil, errors.New("already listening on " + addr) | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	listener := &mockListener{ | 	listener := &mockListener{ | ||||||
| 		opts: options, | 		opts: options, | ||||||
| 		addr: addr, | 		addr: addr, | ||||||
|   | |||||||
| @@ -58,3 +58,32 @@ func TestTransport(t *testing.T) { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| } | } | ||||||
|  |  | ||||||
|  | func TestListener(t *testing.T) { | ||||||
|  | 	tr := NewTransport() | ||||||
|  |  | ||||||
|  | 	// bind / listen on random port | ||||||
|  | 	l, err := tr.Listen(":0") | ||||||
|  | 	if err != nil { | ||||||
|  | 		t.Fatalf("Unexpected error listening %v", err) | ||||||
|  | 	} | ||||||
|  | 	defer l.Close() | ||||||
|  |  | ||||||
|  | 	// try again | ||||||
|  | 	l2, err := tr.Listen(":0") | ||||||
|  | 	if err != nil { | ||||||
|  | 		t.Fatalf("Unexpected error listening %v", err) | ||||||
|  | 	} | ||||||
|  | 	defer l2.Close() | ||||||
|  |  | ||||||
|  | 	// now make sure it still fails | ||||||
|  | 	l3, err := tr.Listen(":8080") | ||||||
|  | 	if err != nil { | ||||||
|  | 		t.Fatalf("Unexpected error listening %v", err) | ||||||
|  | 	} | ||||||
|  | 	defer l3.Close() | ||||||
|  |  | ||||||
|  | 	if _, err := tr.Listen(":8080"); err == nil { | ||||||
|  | 		t.Fatal("Expected error binding to :8080 got nil") | ||||||
|  | 	} | ||||||
|  | } | ||||||
|   | |||||||
		Reference in New Issue
	
	Block a user