ceph/src/jaegertracing/thrift/tutorial/go/src/main.go

   1 package main
   2
   3 /*
   4  * Licensed to the Apache Software Foundation (ASF) under one
   5  * or more contributor license agreements. See the NOTICE file
   6  * distributed with this work for additional information
   7  * regarding copyright ownership. The ASF licenses this file
   8  * to you under the Apache License, Version 2.0 (the
   9  * "License"); you may not use this file except in compliance
  10  * with the License. You may obtain a copy of the License at
  11  *
  12  *   http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing,
  15  * software distributed under the License is distributed on an
  16  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  17  * KIND, either express or implied. See the License for the
  18  * specific language governing permissions and limitations
  19  * under the License.
  20  */
  21
  22 import (
  23         "flag"
  24         "fmt"
  25         "github.com/apache/thrift/lib/go/thrift"
  26         "os"
  27 )
  28
  29 func Usage() {
  30         fmt.Fprint(os.Stderr, "Usage of ", os.Args[0], ":\n")
  31         flag.PrintDefaults()
  32         fmt.Fprint(os.Stderr, "\n")
  33 }
  34
  35 func main() {
  36         flag.Usage = Usage
  37         server := flag.Bool("server", false, "Run server")
  38         protocol := flag.String("P", "binary", "Specify the protocol (binary, compact, json, simplejson)")
  39         framed := flag.Bool("framed", false, "Use framed transport")
  40         buffered := flag.Bool("buffered", false, "Use buffered transport")
  41         addr := flag.String("addr", "localhost:9090", "Address to listen to")
  42         secure := flag.Bool("secure", false, "Use tls secure transport")
  43
  44         flag.Parse()
  45
  46         var protocolFactory thrift.TProtocolFactory
  47         switch *protocol {
  48         case "compact":
  49                 protocolFactory = thrift.NewTCompactProtocolFactory()
  50         case "simplejson":
  51                 protocolFactory = thrift.NewTSimpleJSONProtocolFactory()
  52         case "json":
  53                 protocolFactory = thrift.NewTJSONProtocolFactory()
  54         case "binary", "":
  55                 protocolFactory = thrift.NewTBinaryProtocolFactoryDefault()
  56         default:
  57                 fmt.Fprint(os.Stderr, "Invalid protocol specified", protocol, "\n")
  58                 Usage()
  59                 os.Exit(1)
  60         }
  61
  62         var transportFactory thrift.TTransportFactory
  63         if *buffered {
  64                 transportFactory = thrift.NewTBufferedTransportFactory(8192)
  65         } else {
  66                 transportFactory = thrift.NewTTransportFactory()
  67         }
  68
  69         if *framed {
  70                 transportFactory = thrift.NewTFramedTransportFactory(transportFactory)
  71         }
  72
  73         if *server {
  74                 if err := runServer(transportFactory, protocolFactory, *addr, *secure); err != nil {
  75                         fmt.Println("error running server:", err)
  76                 }
  77         } else {
  78                 if err := runClient(transportFactory, protocolFactory, *addr, *secure); err != nil {
  79                         fmt.Println("error running client:", err)
  80                 }
  81         }
  82 }