浏览代码

Modify Go Echo client and server to also support TLS connections.

Tim Burks 9 年之前
父节点
当前提交
57d5bf581b
共有 2 个文件被更改,包括 8 次插入7 次删除
  1. 2 2
      Examples/Echo/Go/go/src/client/client.go
  2. 6 5
      Examples/Echo/Go/go/src/server/server.go

+ 2 - 2
Examples/Echo/Go/go/src/client/client.go

@@ -29,7 +29,6 @@ import (
 )
 
 const (
-	useSSL         = false
 	defaultMessage = "hello"
 )
 
@@ -38,13 +37,14 @@ func main() {
 	var stream = flag.Int("s", 0, "send multiple messages by streaming")
 	var message = flag.String("m", defaultMessage, "the message to send")
 	var address = flag.String("a", "", "address of the echo server to use")
+	var useTLS = flag.Bool("tls", false, "Use tls for connections.")
 
 	flag.Parse()
 
 	// Set up a connection to the server.
 	var conn *grpc.ClientConn
 	var err error
-	if !useSSL {
+	if !*useTLS {
 		if *address == "" {
 			*address = "localhost:8080"
 		}

+ 6 - 5
Examples/Echo/Go/go/src/server/server.go

@@ -15,6 +15,7 @@
 package main
 
 import (
+	"flag"
 	"fmt"
 	"io"
 	"log"
@@ -26,10 +27,6 @@ import (
 	"google.golang.org/grpc/credentials"
 )
 
-const (
-	useSSL = false
-)
-
 // [START echoserver]
 type EchoServer struct{}
 
@@ -71,10 +68,14 @@ func (s *EchoServer) Update(stream pb.Echo_UpdateServer) error {
 
 // [START main]
 func main() {
+	var useTLS = flag.Bool("tls", false, "Use tls for connections.")
+
+	flag.Parse()
+
 	var err error
 	var lis net.Listener
 	var grpcServer *grpc.Server
-	if !useSSL {
+	if !*useTLS {
 		lis, err = net.Listen("tcp", ":8080")
 		if err != nil {
 			log.Fatalf("failed to listen: %v", err)