initial import

example/tracing/edge/config.json

@@ -0,0 +1,3 @@
+{
+    "Server": "localhost:3456"
+}

example/tracing/edge/main.go

@@ -0,0 +1,61 @@
+package main
+
+import (
+	"flag"
+	"log"
+	"net/http"
+
+	"zero/core/conf"
+	"zero/core/httpx"
+	"zero/core/logx"
+	"zero/core/service"
+	"zero/example/tracing/remote/portal"
+	"zero/ngin"
+	"zero/rpcx"
+)
+
+var (
+	configFile = flag.String("f", "config.json", "the config file")
+	client     *rpcx.RpcClient
+)
+
+func handle(w http.ResponseWriter, r *http.Request) {
+	conn, ok := client.Next()
+	if !ok {
+		log.Fatal("no server")
+	}
+
+	greet := portal.NewPortalClient(conn)
+	resp, err := greet.Portal(r.Context(), &portal.PortalRequest{
+		Name: "kevin",
+	})
+	if err != nil {
+		httpx.WriteJson(w, http.StatusInternalServerError, http.StatusText(http.StatusInternalServerError))
+	} else {
+		httpx.OkJson(w, resp.Response)
+	}
+}
+
+func main() {
+	flag.Parse()
+
+	var c rpcx.RpcClientConf
+	conf.MustLoad(*configFile, &c)
+	client = rpcx.MustNewClient(c)
+	engine := ngin.MustNewEngine(ngin.NgConf{
+		ServiceConf: service.ServiceConf{
+			Log: logx.LogConf{
+				Mode: "console",
+			},
+		},
+		Port: 3333,
+	})
+	defer engine.Stop()
+
+	engine.AddRoute(ngin.Route{
+		Method:  http.MethodGet,
+		Path:    "/",
+		Handler: handle,
+	})
+	engine.Start()
+}

example/tracing/portal/etc/config.json

@@ -0,0 +1,19 @@
+{
+    "Name": "portal.rpc",
+    "ListenOn": "localhost:3456",
+    "Etcd": {
+        "Hosts": [
+            "localhost:2379"
+        ],
+        "Key": "portal"
+    },
+    "UserRpc": {
+        "Etcd": {
+            "Hosts": [
+                "localhost:2379"
+            ],
+            "Key": "user"
+        }
+    },
+    "Timeout": 500
+}

example/tracing/portal/server.go

@@ -0,0 +1,67 @@
+package main
+
+import (
+	"context"
+	"errors"
+	"flag"
+
+	"zero/core/conf"
+	"zero/example/tracing/remote/portal"
+	"zero/example/tracing/remote/user"
+	"zero/rpcx"
+
+	"google.golang.org/grpc"
+)
+
+var configFile = flag.String("f", "etc/config.json", "the config file")
+
+type (
+	Config struct {
+		rpcx.RpcServerConf
+		UserRpc rpcx.RpcClientConf
+	}
+
+	PortalServer struct {
+		userRpc *rpcx.RpcClient
+	}
+)
+
+func NewPortalServer(client *rpcx.RpcClient) *PortalServer {
+	return &PortalServer{
+		userRpc: client,
+	}
+}
+
+func (gs *PortalServer) Portal(ctx context.Context, req *portal.PortalRequest) (*portal.PortalResponse, error) {
+	conn, ok := gs.userRpc.Next()
+	if !ok {
+		return nil, errors.New("internal error")
+	}
+
+	greet := user.NewUserClient(conn)
+	resp, err := greet.GetGrade(ctx, &user.UserRequest{
+		Name: req.Name,
+	})
+	if err != nil {
+		return &portal.PortalResponse{
+			Response: err.Error(),
+		}, nil
+	} else {
+		return &portal.PortalResponse{
+			Response: resp.Response,
+		}, nil
+	}
+}
+
+func main() {
+	flag.Parse()
+
+	var c Config
+	conf.MustLoad(*configFile, &c)
+
+	client := rpcx.MustNewClient(c.UserRpc)
+	server := rpcx.MustNewServer(c.RpcServerConf, func(grpcServer *grpc.Server) {
+		portal.RegisterPortalServer(grpcServer, NewPortalServer(client))
+	})
+	server.Start()
+}

example/tracing/remote/portal/portal.pb.go

@@ -0,0 +1,158 @@
+// Code generated by protoc-gen-go.
+// source: portal.proto
+// DO NOT EDIT!
+
+/*
+Package portal is a generated protocol buffer package.
+
+It is generated from these files:
+	portal.proto
+
+It has these top-level messages:
+	PortalRequest
+	PortalResponse
+*/
+package portal
+
+import proto "github.com/golang/protobuf/proto"
+import fmt "fmt"
+import math "math"
+
+import (
+	context "golang.org/x/net/context"
+	grpc "google.golang.org/grpc"
+)
+
+// Reference imports to suppress errors if they are not otherwise used.
+var _ = proto.Marshal
+var _ = fmt.Errorf
+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 PortalRequest struct {
+	Name string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty"`
+}
+
+func (m *PortalRequest) Reset()                    { *m = PortalRequest{} }
+func (m *PortalRequest) String() string            { return proto.CompactTextString(m) }
+func (*PortalRequest) ProtoMessage()               {}
+func (*PortalRequest) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{0} }
+
+func (m *PortalRequest) GetName() string {
+	if m != nil {
+		return m.Name
+	}
+	return ""
+}
+
+type PortalResponse struct {
+	Response string `protobuf:"bytes,1,opt,name=response" json:"response,omitempty"`
+}
+
+func (m *PortalResponse) Reset()                    { *m = PortalResponse{} }
+func (m *PortalResponse) String() string            { return proto.CompactTextString(m) }
+func (*PortalResponse) ProtoMessage()               {}
+func (*PortalResponse) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{1} }
+
+func (m *PortalResponse) GetResponse() string {
+	if m != nil {
+		return m.Response
+	}
+	return ""
+}
+
+func init() {
+	proto.RegisterType((*PortalRequest)(nil), "portal.PortalRequest")
+	proto.RegisterType((*PortalResponse)(nil), "portal.PortalResponse")
+}
+
+// Reference imports to suppress errors if they are not otherwise used.
+var _ context.Context
+var _ grpc.ClientConn
+
+// This is a compile-time assertion to ensure that this generated file
+// is compatible with the grpc package it is being compiled against.
+const _ = grpc.SupportPackageIsVersion4
+
+// Client API for Portal service
+
+type PortalClient interface {
+	Portal(ctx context.Context, in *PortalRequest, opts ...grpc.CallOption) (*PortalResponse, error)
+}
+
+type portalClient struct {
+	cc *grpc.ClientConn
+}
+
+func NewPortalClient(cc *grpc.ClientConn) PortalClient {
+	return &portalClient{cc}
+}
+
+func (c *portalClient) Portal(ctx context.Context, in *PortalRequest, opts ...grpc.CallOption) (*PortalResponse, error) {
+	out := new(PortalResponse)
+	err := grpc.Invoke(ctx, "/portal.Portal/Portal", in, out, c.cc, opts...)
+	if err != nil {
+		return nil, err
+	}
+	return out, nil
+}
+
+// Server API for Portal service
+
+type PortalServer interface {
+	Portal(context.Context, *PortalRequest) (*PortalResponse, error)
+}
+
+func RegisterPortalServer(s *grpc.Server, srv PortalServer) {
+	s.RegisterService(&_Portal_serviceDesc, srv)
+}
+
+func _Portal_Portal_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
+	in := new(PortalRequest)
+	if err := dec(in); err != nil {
+		return nil, err
+	}
+	if interceptor == nil {
+		return srv.(PortalServer).Portal(ctx, in)
+	}
+	info := &grpc.UnaryServerInfo{
+		Server:     srv,
+		FullMethod: "/portal.Portal/Portal",
+	}
+	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
+		return srv.(PortalServer).Portal(ctx, req.(*PortalRequest))
+	}
+	return interceptor(ctx, in, info, handler)
+}
+
+var _Portal_serviceDesc = grpc.ServiceDesc{
+	ServiceName: "portal.Portal",
+	HandlerType: (*PortalServer)(nil),
+	Methods: []grpc.MethodDesc{
+		{
+			MethodName: "Portal",
+			Handler:    _Portal_Portal_Handler,
+		},
+	},
+	Streams:  []grpc.StreamDesc{},
+	Metadata: "portal.proto",
+}
+
+func init() { proto.RegisterFile("portal.proto", fileDescriptor0) }
+
+var fileDescriptor0 = []byte{
+	// 122 bytes of a gzipped FileDescriptorProto
+	0x1f, 0x8b, 0x08, 0x00, 0x00, 0x09, 0x6e, 0x88, 0x02, 0xff, 0xe2, 0xe2, 0x29, 0xc8, 0x2f, 0x2a,
+	0x49, 0xcc, 0xd1, 0x2b, 0x28, 0xca, 0x2f, 0xc9, 0x17, 0x62, 0x83, 0xf0, 0x94, 0x94, 0xb9, 0x78,
+	0x03, 0xc0, 0xac, 0xa0, 0xd4, 0xc2, 0xd2, 0xd4, 0xe2, 0x12, 0x21, 0x21, 0x2e, 0x96, 0xbc, 0xc4,
+	0xdc, 0x54, 0x09, 0x46, 0x05, 0x46, 0x0d, 0xce, 0x20, 0x30, 0x5b, 0x49, 0x87, 0x8b, 0x0f, 0xa6,
+	0xa8, 0xb8, 0x20, 0x3f, 0xaf, 0x38, 0x55, 0x48, 0x8a, 0x8b, 0xa3, 0x08, 0xca, 0x86, 0xaa, 0x84,
+	0xf3, 0x8d, 0x1c, 0xb9, 0xd8, 0x20, 0xaa, 0x85, 0xcc, 0xe1, 0x2c, 0x51, 0x3d, 0xa8, 0xed, 0x28,
+	0x96, 0x49, 0x89, 0xa1, 0x0b, 0x43, 0x8c, 0x48, 0x62, 0x03, 0x3b, 0xd2, 0x18, 0x10, 0x00, 0x00,
+	0xff, 0xff, 0xce, 0x05, 0x16, 0xd0, 0xb4, 0x00, 0x00, 0x00,
+}

example/tracing/remote/portal/portal.proto

@@ -0,0 +1,15 @@
+syntax = "proto3";
+
+package portal;
+
+message PortalRequest {
+    string name = 1;
+}
+
+message PortalResponse {
+    string response = 1;
+}
+
+service Portal {
+    rpc Portal(PortalRequest) returns (PortalResponse);
+}

example/tracing/remote/user/user.pb.go

@@ -0,0 +1,159 @@
+// Code generated by protoc-gen-go.
+// source: user.proto
+// DO NOT EDIT!
+
+/*
+Package user is a generated protocol buffer package.
+
+It is generated from these files:
+	user.proto
+
+It has these top-level messages:
+	UserRequest
+	UserResponse
+*/
+package user
+
+import proto "github.com/golang/protobuf/proto"
+import fmt "fmt"
+import math "math"
+
+import (
+	context "golang.org/x/net/context"
+	grpc "google.golang.org/grpc"
+)
+
+// Reference imports to suppress errors if they are not otherwise used.
+var _ = proto.Marshal
+var _ = fmt.Errorf
+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 UserRequest struct {
+	Name string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty"`
+}
+
+func (m *UserRequest) Reset()                    { *m = UserRequest{} }
+func (m *UserRequest) String() string            { return proto.CompactTextString(m) }
+func (*UserRequest) ProtoMessage()               {}
+func (*UserRequest) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{0} }
+
+func (m *UserRequest) GetName() string {
+	if m != nil {
+		return m.Name
+	}
+	return ""
+}
+
+type UserResponse struct {
+	Response string `protobuf:"bytes,1,opt,name=response" json:"response,omitempty"`
+}
+
+func (m *UserResponse) Reset()                    { *m = UserResponse{} }
+func (m *UserResponse) String() string            { return proto.CompactTextString(m) }
+func (*UserResponse) ProtoMessage()               {}
+func (*UserResponse) Descriptor() ([]byte, []int) { return fileDescriptor0, []int{1} }
+
+func (m *UserResponse) GetResponse() string {
+	if m != nil {
+		return m.Response
+	}
+	return ""
+}
+
+func init() {
+	proto.RegisterType((*UserRequest)(nil), "user.UserRequest")
+	proto.RegisterType((*UserResponse)(nil), "user.UserResponse")
+}
+
+// Reference imports to suppress errors if they are not otherwise used.
+var _ context.Context
+var _ grpc.ClientConn
+
+// This is a compile-time assertion to ensure that this generated file
+// is compatible with the grpc package it is being compiled against.
+const _ = grpc.SupportPackageIsVersion4
+
+// Client API for User service
+
+type UserClient interface {
+	GetGrade(ctx context.Context, in *UserRequest, opts ...grpc.CallOption) (*UserResponse, error)
+}
+
+type userClient struct {
+	cc *grpc.ClientConn
+}
+
+func NewUserClient(cc *grpc.ClientConn) UserClient {
+	return &userClient{cc}
+}
+
+func (c *userClient) GetGrade(ctx context.Context, in *UserRequest, opts ...grpc.CallOption) (*UserResponse, error) {
+	out := new(UserResponse)
+	err := grpc.Invoke(ctx, "/user.User/GetGrade", in, out, c.cc, opts...)
+	if err != nil {
+		return nil, err
+	}
+	return out, nil
+}
+
+// Server API for User service
+
+type UserServer interface {
+	GetGrade(context.Context, *UserRequest) (*UserResponse, error)
+}
+
+func RegisterUserServer(s *grpc.Server, srv UserServer) {
+	s.RegisterService(&_User_serviceDesc, srv)
+}
+
+func _User_GetGrade_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
+	in := new(UserRequest)
+	if err := dec(in); err != nil {
+		return nil, err
+	}
+	if interceptor == nil {
+		return srv.(UserServer).GetGrade(ctx, in)
+	}
+	info := &grpc.UnaryServerInfo{
+		Server:     srv,
+		FullMethod: "/user.User/GetGrade",
+	}
+	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
+		return srv.(UserServer).GetGrade(ctx, req.(*UserRequest))
+	}
+	return interceptor(ctx, in, info, handler)
+}
+
+var _User_serviceDesc = grpc.ServiceDesc{
+	ServiceName: "user.User",
+	HandlerType: (*UserServer)(nil),
+	Methods: []grpc.MethodDesc{
+		{
+			MethodName: "GetGrade",
+			Handler:    _User_GetGrade_Handler,
+		},
+	},
+	Streams:  []grpc.StreamDesc{},
+	Metadata: "user.proto",
+}
+
+func init() { proto.RegisterFile("user.proto", fileDescriptor0) }
+
+var fileDescriptor0 = []byte{
+	// 131 bytes of a gzipped FileDescriptorProto
+	0x1f, 0x8b, 0x08, 0x00, 0x00, 0x09, 0x6e, 0x88, 0x02, 0xff, 0xe2, 0xe2, 0x2a, 0x2d, 0x4e, 0x2d,
+	0xd2, 0x2b, 0x28, 0xca, 0x2f, 0xc9, 0x17, 0x62, 0x01, 0xb1, 0x95, 0x14, 0xb9, 0xb8, 0x43, 0x8b,
+	0x53, 0x8b, 0x82, 0x52, 0x0b, 0x4b, 0x53, 0x8b, 0x4b, 0x84, 0x84, 0xb8, 0x58, 0xf2, 0x12, 0x73,
+	0x53, 0x25, 0x18, 0x15, 0x18, 0x35, 0x38, 0x83, 0xc0, 0x6c, 0x25, 0x2d, 0x2e, 0x1e, 0x88, 0x92,
+	0xe2, 0x82, 0xfc, 0xbc, 0xe2, 0x54, 0x21, 0x29, 0x2e, 0x8e, 0x22, 0x28, 0x1b, 0xaa, 0x0e, 0xce,
+	0x37, 0xb2, 0xe4, 0x62, 0x01, 0xa9, 0x15, 0x32, 0xe4, 0xe2, 0x70, 0x4f, 0x2d, 0x71, 0x2f, 0x4a,
+	0x4c, 0x49, 0x15, 0x12, 0xd4, 0x03, 0xdb, 0x8a, 0x64, 0x8d, 0x94, 0x10, 0xb2, 0x10, 0x44, 0x6b,
+	0x12, 0x1b, 0xd8, 0x59, 0xc6, 0x80, 0x00, 0x00, 0x00, 0xff, 0xff, 0x99, 0xfd, 0x8f, 0x70, 0xa4,
+	0x00, 0x00, 0x00,
+}

example/tracing/remote/user/user.proto

@@ -0,0 +1,17 @@
+syntax = "proto3";
+
+package user;
+
+message UserRequest {
+    string name = 1;
+}
+
+message UserResponse {
+    string response = 1;
+}
+
+service User {
+    rpc GetGrade(UserRequest) returns (UserResponse);
+}
+
+

example/tracing/user/etc/config.json

@@ -0,0 +1,11 @@
+{
+    "Name": "user.rpc",
+    "ListenOn": "localhost:3457",
+    "Etcd": {
+        "Hosts": [
+            "localhost:2379"
+        ],
+        "Key": "user"
+    },
+    "Timeout": 500
+}

example/tracing/user/server.go

@@ -0,0 +1,55 @@
+package main
+
+import (
+	"context"
+	"flag"
+	"fmt"
+	"os"
+	"sync"
+	"time"
+
+	"zero/core/conf"
+	"zero/example/tracing/remote/user"
+	"zero/rpcx"
+
+	"google.golang.org/grpc"
+)
+
+var configFile = flag.String("f", "etc/config.json", "the config file")
+
+type UserServer struct {
+	lock     sync.Mutex
+	alive    bool
+	downTime time.Time
+}
+
+func NewUserServer() *UserServer {
+	return &UserServer{
+		alive: true,
+	}
+}
+
+func (gs *UserServer) GetGrade(ctx context.Context, req *user.UserRequest) (*user.UserResponse, error) {
+	fmt.Println("=>", req)
+
+	hostname, err := os.Hostname()
+	if err != nil {
+		return nil, err
+	}
+
+	return &user.UserResponse{
+		Response: "hello from " + hostname,
+	}, nil
+}
+
+func main() {
+	flag.Parse()
+
+	var c rpcx.RpcServerConf
+	conf.MustLoad(*configFile, &c)
+
+	server := rpcx.MustNewServer(c, func(grpcServer *grpc.Server) {
+		user.RegisterUserServer(grpcServer, NewUserServer())
+	})
+	server.Start()
+}