【go语言之thrift协议三之client端分析】

上一篇文章分析了thrift协议server端的实现，这边还是基于官方的示例去分析。

import (
	"crypto/tls"
	"flag"
	"fmt"
	"os"

	"github.com/apache/thrift/lib/go/thrift"
)

func Usage() {
   
	fmt.Fprint(os.Stderr, "Usage of ", os.Args[0], ":\n")
	flag.PrintDefaults()
	fmt.Fprint(os.Stderr, "\n")
}

func main() {
   
	flag.Usage = Usage
	server := flag.Bool("server", false, "Run server")
	protocol := flag.String("P", "compact", "Specify the protocol (binary, compact, json, simplejson)")
	framed := flag.Bool("framed", false, "Use framed transport")
	buffered := flag.Bool("buffered", false, "Use buffered transport")
	addr := flag.String("addr", "localhost:9090", "Address to listen to")
	secure := flag.Bool("secure", false, "Use tls secure transport")

	flag.Parse()

	var protocolFactory thrift.TProtocolFactory
	switch *protocol {
   
	case "compact":
		protocolFactory = thrift.NewTCompactProtocolFactoryConf(nil)
	case "simplejson":
		protocolFactory = thrift.NewTSimpleJSONProtocolFactoryConf(nil)
	case "json":
		protocolFactory = thrift.NewTJSONProtocolFactory()
	case "binary", "":
		protocolFactory = thrift.NewTBinaryProtocolFactoryConf(nil)
	default:
		fmt.Fprint(os.Stderr, "Invalid protocol specified", protocol, "\n")
		Usage()
		os.Exit(1)
	}

	var transportFactory thrift.TTransportFactory
	cfg := &thrift.TConfiguration{
   
		TLSConfig: &tls.Config{
   
			InsecureSkipVerify: true,
		},
	}
	if *buffered {
   
		transportFactory = thrift.NewTBufferedTransportFactory(8192)
	} else {
   
		transportFactory = thrift.NewTTransportFactory()
	}

	if *framed {
   
		transportFactory = thrift.NewTFramedTransportFactoryConf(transportFactory, cfg)
	}

	if *server {
   
		if err := runServer(transportFactory, protocolFactory, *addr, *secure); err != nil {
   
			fmt.Println("error running server:", err)
		}
	} else {
   
		if err := runClient(transportFactory, protocolFactory, *addr, *secure, cfg); err != nil {
   
			fmt.Println("error running client:", err)
		}