1package 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 22import ( 23 "flag" 24 "fmt" 25 "github.com/apache/thrift/lib/go/thrift" 26 "os" 27) 28 29func Usage() { 30 fmt.Fprint(os.Stderr, "Usage of ", os.Args[0], ":\n") 31 flag.PrintDefaults() 32 fmt.Fprint(os.Stderr, "\n") 33} 34 35func 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} 83