1/*
2 * Licensed to the Apache Software Foundation (ASF) under one
3 * or more contributor license agreements. See the NOTICE file
4 * distributed with this work for additional information
5 * regarding copyright ownership. The ASF licenses this file
6 * to you under the Apache License, Version 2.0 (the
7 * "License"); you may not use this file except in compliance
8 * with the License. You may obtain a copy of the License at
9 *
10 *   http://www.apache.org/licenses/LICENSE-2.0
11 *
12 * Unless required by applicable law or agreed to in writing,
13 * software distributed under the License is distributed on an
14 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15 * KIND, either express or implied. See the License for the
16 * specific language governing permissions and limitations
17 * under the License.
18 */
19
20package main
21
22import (
23	"flag"
24	"fmt"
25	"log"
26	"net/http"
27
28	"github.com/apache/thrift/lib/go/thrift"
29	"github.com/apache/thrift/test/go/src/common"
30)
31
32var host = flag.String("host", "localhost", "Host to connect")
33var port = flag.Int64("port", 9090, "Port number to connect")
34var domain_socket = flag.String("domain-socket", "", "Domain Socket (e.g. /tmp/ThriftTest.thrift), instead of host and port")
35var transport = flag.String("transport", "buffered", "Transport: buffered, framed, http, zlib")
36var _ = flag.Bool("zlib", false, "For compatibility. Ignored.")
37var protocol = flag.String("protocol", "binary", "Protocol: binary, compact, json, header")
38var ssl = flag.Bool("ssl", false, "Encrypted Transport using SSL")
39var certPath = flag.String("certPath", "keys", "Directory that contains SSL certificates")
40
41func main() {
42	flag.Parse()
43
44	processor, serverTransport, transportFactory, protocolFactory, err := common.GetServerParams(*host, *port, *domain_socket, *transport, *protocol, *ssl, *certPath, common.PrintingHandler)
45
46	if err != nil {
47		log.Fatalf("Unable to process server params: %v", err)
48	}
49
50	if *transport == "http" {
51		http.HandleFunc("/", thrift.NewThriftHandlerFunc(processor, protocolFactory, protocolFactory))
52
53		if *ssl {
54			err := http.ListenAndServeTLS(fmt.Sprintf(":%d", *port),
55				*certPath+"/server.pem", *certPath+"/server.key", nil)
56
57			if err != nil {
58				fmt.Println(err)
59				return
60			}
61		} else {
62			http.ListenAndServe(fmt.Sprintf(":%d", *port), nil)
63		}
64	} else {
65		server := thrift.NewTSimpleServer4(processor, serverTransport, transportFactory, protocolFactory)
66		if err = server.Listen(); err != nil {
67			return
68		}
69		go server.AcceptLoop()
70		server.Serve()
71	}
72}
73