tracing/jsonrpc/go-client/app/client.go - dubbo-go-samples - Git at Google

 /*
  * Licensed to the Apache Software Foundation (ASF) under one or more
  * contributor license agreements.  See the NOTICE file distributed with
  * this work for additional information regarding copyright ownership.
  * The ASF licenses this file to You under the Apache License, Version 2.0
  * (the "License"); you may not use this file except in compliance with
  * the License.  You may obtain a copy of the License at
  *
  *     http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */

 package main

 import (
 	"context"
 	"fmt"
 	"os"
 	"os/signal"
 	"syscall"
 	"time"
 )

 import (
 	_ "github.com/apache/dubbo-go/cluster/cluster_impl"
 	_ "github.com/apache/dubbo-go/cluster/loadbalance"
 	"github.com/apache/dubbo-go/common/logger"
 	_ "github.com/apache/dubbo-go/common/proxy/proxy_factory"
 	"github.com/apache/dubbo-go/config"
 	_ "github.com/apache/dubbo-go/filter/filter_impl"
 	_ "github.com/apache/dubbo-go/protocol/jsonrpc"
 	_ "github.com/apache/dubbo-go/registry/protocol"
 	_ "github.com/apache/dubbo-go/registry/zookeeper"

 	"github.com/dubbogo/gost/log"

 	"github.com/opentracing/opentracing-go"

 	zipkinot "github.com/openzipkin-contrib/zipkin-go-opentracing"

 	"github.com/openzipkin/zipkin-go"
 	zipkinhttp "github.com/openzipkin/zipkin-go/reporter/http"
 )

 var (
 	survivalTimeout int = 10e9
 )

 // they are necessary:
 // 		export CONF_CONSUMER_FILE_PATH="xxx"
 // 		export APP_LOG_CONF_FILE="xxx"
 func main() {

 	config.Load()
 	initZipkin()

 	gxlog.CInfo("\n\n\nstart to test jsonrpc")
 	user := &JsonRPCUser{}
 	ctx := context.WithValue(context.Background(), "MyTracing", "Tracing123")
 	span, spanCtx := opentracing.StartSpanFromContext(ctx, "User-Test-Service")
 	err := userProvider.GetUser(spanCtx, []interface{}{"A003"}, user)
 	span.Finish()
 	if err != nil {
 		panic(err)
 	}
 	gxlog.CInfo("response result: %v", user)

 	initSignal()
 }

 func initZipkin() {
 	// set up a span reporter
 	reporter := zipkinhttp.NewReporter("http://localhost:9411/api/v2/spans")

 	// create our local service endpoint
 	endpoint, err := zipkin.NewEndpoint("client-service", "myservice.mydomain.com:80")
 	if err != nil {
 		logger.Errorf("unable to create local endpoint: %+v\n", err)
 	}

 	// initialize our tracer
 	nativeTracer, err := zipkin.NewTracer(reporter, zipkin.WithLocalEndpoint(endpoint))
 	if err != nil {
 		logger.Errorf("unable to create tracer: %+v\n", err)
 	}

 	// use zipkin-go-opentracing to wrap our tracer
 	tracer := zipkinot.Wrap(nativeTracer)

 	// optionally set as Global OpenTracing tracer instance
 	opentracing.SetGlobalTracer(tracer)
 }

 func initSignal() {
 	signals := make(chan os.Signal, 1)
 	// It is not possible to block SIGKILL or syscall.SIGSTOP
 	signal.Notify(signals, os.Interrupt, os.Kill, syscall.SIGHUP,
 		syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGINT)
 	for {
 		sig := <-signals
 		logger.Infof("get signal %s", sig.String())
 		switch sig {
 		case syscall.SIGHUP:
 		// reload()
 		default:
 			time.AfterFunc(time.Duration(survivalTimeout), func() {
 				logger.Warnf("app exit now by force...")
 				os.Exit(1)
 			})

 			// The program exits normally or timeout forcibly exits.
 			fmt.Println("app exit now...")
 			return
 		}
 	}
 }

 func test() {
 	gxlog.CInfo("\n\n\necho")
 	res, err := userProvider.Echo(context.TODO(), "OK")
 	if err != nil {
 		gxlog.CInfo("echo - error: %v", err)
 	} else {
 		gxlog.CInfo("res: %v", res)
 	}

 	time.Sleep(3e9)

 	ctx := context.WithValue(context.Background(), "TracingID", "Tracing123")
 	gxlog.CInfo("\n\n\nstart to test jsonrpc - GetUser0")
 	ret, err := userProvider.GetUser0("A003", "Moorse")
 	if err != nil {
 		panic(err)
 	}
 	gxlog.CInfo("response result: %v", ret)

 	gxlog.CInfo("\n\n\nstart to test jsonrpc - GetUsers")
 	ret1, err := userProvider.GetUsers([]interface{}{[]interface{}{"A002", "A003"}})
 	if err != nil {
 		panic(err)
 	}
 	gxlog.CInfo("response result: %v", ret1)

 	gxlog.CInfo("\n\n\nstart to test jsonrpc - getUser")
 	user := &JsonRPCUser{}
 	err = userProvider.GetUser2(ctx, []interface{}{1}, user)
 	if err != nil {
 		panic(err)
 	}
 	gxlog.CInfo("response result: %v", user)

 	gxlog.CInfo("\n\n\nstart to test jsonrpc - GetUser3")
 	err = userProvider.GetUser3()
 	if err != nil {
 		panic(err)
 	}
 	gxlog.CInfo("succ!")

 	gxlog.CInfo("\n\n\nstart to test jsonrpc illegal method")
 	err = userProvider.GetUser1(ctx, []interface{}{"A003"}, user)
 	if err == nil {
 		panic("err is nil")
 	}
 	gxlog.CInfo("error: %v", err)
 }

 func test1() {
 	ctx := context.WithValue(context.Background(), "TracingID", "Tracing123")
 	gxlog.CInfo("\n\n\necho")
 	res, err := userProvider1.Echo(ctx, "OK")
 	if err != nil {
 		gxlog.CInfo("echo - error: %v", err)
 	} else {
 		gxlog.CInfo("res: %v", res)
 	}

 	time.Sleep(3e9)

 	gxlog.CInfo("\n\n\nstart to test jsonrpc")
 	user := &JsonRPCUser{}
 	err = userProvider1.GetUser(ctx, []interface{}{"A003"}, user)
 	if err != nil {
 		panic(err)
 	}
 	gxlog.CInfo("response result: %v", user)

 	gxlog.CInfo("\n\n\nstart to test jsonrpc - GetUser0")
 	ret, err := userProvider1.GetUser0("A003", "Moorse")
 	if err != nil {
 		panic(err)
 	}
 	gxlog.CInfo("response result: %v", ret)

 	gxlog.CInfo("\n\n\nstart to test jsonrpc - GetUsers")
 	ret1, err := userProvider1.GetUsers([]interface{}{[]interface{}{"A002", "A003"}})
 	if err != nil {
 		panic(err)
 	}
 	gxlog.CInfo("response result: %v", ret1)

 	gxlog.CInfo("\n\n\nstart to test jsonrpc - getUser")
 	user = &JsonRPCUser{}
 	err = userProvider1.GetUser2(ctx, []interface{}{1}, user)
 	if err != nil {
 		panic(err)
 	}
 	gxlog.CInfo("response result: %v", user)

 	gxlog.CInfo("\n\n\nstart to test jsonrpc - GetUser3")
 	err = userProvider1.GetUser3()
 	if err != nil {
 		panic(err)
 	}
 	gxlog.CInfo("succ!")

 	gxlog.CInfo("\n\n\nstart to test jsonrpc illegal method")
 	err = userProvider1.GetUser1(ctx, []interface{}{"A003"}, user)
 	if err == nil {
 		panic("err is nil")
 	}
 	gxlog.CInfo("error: %v", err)
 }

 func test2() {
 	ctx := context.WithValue(context.Background(), "TracingID", "Tracing123")
 	gxlog.CInfo("\n\n\necho")
 	res, err := userProvider2.Echo(ctx, "OK")
 	if err != nil {
 		gxlog.CInfo("echo - error: %v", err)
 	} else {
 		gxlog.CInfo("res: %v", res)
 	}

 	time.Sleep(3e9)

 	gxlog.CInfo("\n\n\nstart to test jsonrpc")
 	user := &JsonRPCUser{}
 	err = userProvider2.GetUser(ctx, []interface{}{"A003"}, user)
 	if err != nil {
 		panic(err)
 	}
 	gxlog.CInfo("response result: %v", user)

 	gxlog.CInfo("\n\n\nstart to test jsonrpc - GetUser0")
 	ret, err := userProvider2.GetUser0("A003", "Moorse")
 	if err != nil {
 		panic(err)
 	}
 	gxlog.CInfo("response result: %v", ret)

 	gxlog.CInfo("\n\n\nstart to test jsonrpc - GetUsers")
 	ret1, err := userProvider2.GetUsers([]interface{}{[]interface{}{"A002", "A003"}})
 	if err != nil {
 		panic(err)
 	}
 	gxlog.CInfo("response result: %v", ret1)

 	gxlog.CInfo("\n\n\nstart to test jsonrpc - getUser")
 	user = &JsonRPCUser{}
 	err = userProvider2.GetUser2(ctx, []interface{}{1}, user)
 	if err != nil {
 		panic(err)
 	}
 	gxlog.CInfo("response result: %v", user)

 	gxlog.CInfo("\n\n\nstart to test jsonrpc - GetUser3")
 	err = userProvider2.GetUser3()
 	if err != nil {
 		panic(err)
 	}
 	gxlog.CInfo("succ!")

 	gxlog.CInfo("\n\n\nstart to test jsonrpc illegal method")
 	err = userProvider2.GetUser1(ctx, []interface{}{"A003"}, user)
 	if err == nil {
 		panic("err is nil")
 	}
 	gxlog.CInfo("error: %v", err)
 }
	/*
	* Licensed to the Apache Software Foundation (ASF) under one or more
	* contributor license agreements. See the NOTICE file distributed with
	* this work for additional information regarding copyright ownership.
	* The ASF licenses this file to You under the Apache License, Version 2.0
	* (the "License"); you may not use this file except in compliance with
	* the License. You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/

	package main

	import (
	"context"
	"fmt"
	"os"
	"os/signal"
	"syscall"
	"time"
	)

	import (
	_ "github.com/apache/dubbo-go/cluster/cluster_impl"
	_ "github.com/apache/dubbo-go/cluster/loadbalance"
	"github.com/apache/dubbo-go/common/logger"
	_ "github.com/apache/dubbo-go/common/proxy/proxy_factory"
	"github.com/apache/dubbo-go/config"
	_ "github.com/apache/dubbo-go/filter/filter_impl"
	_ "github.com/apache/dubbo-go/protocol/jsonrpc"
	_ "github.com/apache/dubbo-go/registry/protocol"
	_ "github.com/apache/dubbo-go/registry/zookeeper"

	"github.com/dubbogo/gost/log"

	"github.com/opentracing/opentracing-go"

	zipkinot "github.com/openzipkin-contrib/zipkin-go-opentracing"

	"github.com/openzipkin/zipkin-go"
	zipkinhttp "github.com/openzipkin/zipkin-go/reporter/http"
	)

	var (
	survivalTimeout int = 10e9
	)

	// they are necessary:
	// export CONF_CONSUMER_FILE_PATH="xxx"
	// export APP_LOG_CONF_FILE="xxx"
	func main() {

	config.Load()
	initZipkin()

	gxlog.CInfo("\n\n\nstart to test jsonrpc")
	user := &JsonRPCUser{}
	ctx := context.WithValue(context.Background(), "MyTracing", "Tracing123")
	span, spanCtx := opentracing.StartSpanFromContext(ctx, "User-Test-Service")
	err := userProvider.GetUser(spanCtx, []interface{}{"A003"}, user)
	span.Finish()
	if err != nil {
	panic(err)
	}
	gxlog.CInfo("response result: %v", user)

	initSignal()
	}

	func initZipkin() {
	// set up a span reporter
	reporter := zipkinhttp.NewReporter("http://localhost:9411/api/v2/spans")

	// create our local service endpoint
	endpoint, err := zipkin.NewEndpoint("client-service", "myservice.mydomain.com:80")
	if err != nil {
	logger.Errorf("unable to create local endpoint: %+v\n", err)
	}

	// initialize our tracer
	nativeTracer, err := zipkin.NewTracer(reporter, zipkin.WithLocalEndpoint(endpoint))
	if err != nil {
	logger.Errorf("unable to create tracer: %+v\n", err)
	}

	// use zipkin-go-opentracing to wrap our tracer
	tracer := zipkinot.Wrap(nativeTracer)

	// optionally set as Global OpenTracing tracer instance
	opentracing.SetGlobalTracer(tracer)
	}

	func initSignal() {
	signals := make(chan os.Signal, 1)
	// It is not possible to block SIGKILL or syscall.SIGSTOP
	signal.Notify(signals, os.Interrupt, os.Kill, syscall.SIGHUP,
	syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGINT)
	for {
	sig := <-signals
	logger.Infof("get signal %s", sig.String())
	switch sig {
	case syscall.SIGHUP:
	// reload()
	default:
	time.AfterFunc(time.Duration(survivalTimeout), func() {
	logger.Warnf("app exit now by force...")
	os.Exit(1)
	})

	// The program exits normally or timeout forcibly exits.
	fmt.Println("app exit now...")
	return
	}
	}
	}

	func test() {
	gxlog.CInfo("\n\n\necho")
	res, err := userProvider.Echo(context.TODO(), "OK")
	if err != nil {
	gxlog.CInfo("echo - error: %v", err)
	} else {
	gxlog.CInfo("res: %v", res)
	}

	time.Sleep(3e9)

	ctx := context.WithValue(context.Background(), "TracingID", "Tracing123")
	gxlog.CInfo("\n\n\nstart to test jsonrpc - GetUser0")
	ret, err := userProvider.GetUser0("A003", "Moorse")
	if err != nil {
	panic(err)
	}
	gxlog.CInfo("response result: %v", ret)

	gxlog.CInfo("\n\n\nstart to test jsonrpc - GetUsers")
	ret1, err := userProvider.GetUsers([]interface{}{[]interface{}{"A002", "A003"}})
	if err != nil {
	panic(err)
	}
	gxlog.CInfo("response result: %v", ret1)

	gxlog.CInfo("\n\n\nstart to test jsonrpc - getUser")
	user := &JsonRPCUser{}
	err = userProvider.GetUser2(ctx, []interface{}{1}, user)
	if err != nil {
	panic(err)
	}
	gxlog.CInfo("response result: %v", user)

	gxlog.CInfo("\n\n\nstart to test jsonrpc - GetUser3")
	err = userProvider.GetUser3()
	if err != nil {
	panic(err)
	}
	gxlog.CInfo("succ!")

	gxlog.CInfo("\n\n\nstart to test jsonrpc illegal method")
	err = userProvider.GetUser1(ctx, []interface{}{"A003"}, user)
	if err == nil {
	panic("err is nil")
	}
	gxlog.CInfo("error: %v", err)
	}

	func test1() {
	ctx := context.WithValue(context.Background(), "TracingID", "Tracing123")
	gxlog.CInfo("\n\n\necho")
	res, err := userProvider1.Echo(ctx, "OK")
	if err != nil {
	gxlog.CInfo("echo - error: %v", err)
	} else {
	gxlog.CInfo("res: %v", res)
	}

	time.Sleep(3e9)

	gxlog.CInfo("\n\n\nstart to test jsonrpc")
	user := &JsonRPCUser{}
	err = userProvider1.GetUser(ctx, []interface{}{"A003"}, user)
	if err != nil {
	panic(err)
	}
	gxlog.CInfo("response result: %v", user)

	gxlog.CInfo("\n\n\nstart to test jsonrpc - GetUser0")
	ret, err := userProvider1.GetUser0("A003", "Moorse")
	if err != nil {
	panic(err)
	}
	gxlog.CInfo("response result: %v", ret)

	gxlog.CInfo("\n\n\nstart to test jsonrpc - GetUsers")
	ret1, err := userProvider1.GetUsers([]interface{}{[]interface{}{"A002", "A003"}})
	if err != nil {
	panic(err)
	}
	gxlog.CInfo("response result: %v", ret1)

	gxlog.CInfo("\n\n\nstart to test jsonrpc - getUser")
	user = &JsonRPCUser{}
	err = userProvider1.GetUser2(ctx, []interface{}{1}, user)
	if err != nil {
	panic(err)
	}
	gxlog.CInfo("response result: %v", user)

	gxlog.CInfo("\n\n\nstart to test jsonrpc - GetUser3")
	err = userProvider1.GetUser3()
	if err != nil {
	panic(err)
	}
	gxlog.CInfo("succ!")

	gxlog.CInfo("\n\n\nstart to test jsonrpc illegal method")
	err = userProvider1.GetUser1(ctx, []interface{}{"A003"}, user)
	if err == nil {
	panic("err is nil")
	}
	gxlog.CInfo("error: %v", err)
	}

	func test2() {
	ctx := context.WithValue(context.Background(), "TracingID", "Tracing123")
	gxlog.CInfo("\n\n\necho")
	res, err := userProvider2.Echo(ctx, "OK")
	if err != nil {
	gxlog.CInfo("echo - error: %v", err)
	} else {
	gxlog.CInfo("res: %v", res)
	}

	time.Sleep(3e9)

	gxlog.CInfo("\n\n\nstart to test jsonrpc")
	user := &JsonRPCUser{}
	err = userProvider2.GetUser(ctx, []interface{}{"A003"}, user)
	if err != nil {
	panic(err)
	}
	gxlog.CInfo("response result: %v", user)

	gxlog.CInfo("\n\n\nstart to test jsonrpc - GetUser0")
	ret, err := userProvider2.GetUser0("A003", "Moorse")
	if err != nil {
	panic(err)
	}
	gxlog.CInfo("response result: %v", ret)

	gxlog.CInfo("\n\n\nstart to test jsonrpc - GetUsers")
	ret1, err := userProvider2.GetUsers([]interface{}{[]interface{}{"A002", "A003"}})
	if err != nil {
	panic(err)
	}
	gxlog.CInfo("response result: %v", ret1)

	gxlog.CInfo("\n\n\nstart to test jsonrpc - getUser")
	user = &JsonRPCUser{}
	err = userProvider2.GetUser2(ctx, []interface{}{1}, user)
	if err != nil {
	panic(err)
	}
	gxlog.CInfo("response result: %v", user)

	gxlog.CInfo("\n\n\nstart to test jsonrpc - GetUser3")
	err = userProvider2.GetUser3()
	if err != nil {
	panic(err)
	}
	gxlog.CInfo("succ!")

	gxlog.CInfo("\n\n\nstart to test jsonrpc illegal method")
	err = userProvider2.GetUser1(ctx, []interface{}{"A003"}, user)
	if err == nil {
	panic("err is nil")
	}
	gxlog.CInfo("error: %v", err)
	}