github.com/blixtra/rkt@v0.8.1-0.20160204105720-ab0d1add1a43/Godeps/_workspace/src/google.golang.org/grpc/interop/server/server.go (about)

     1  /*
     2   *
     3   * Copyright 2014, Google Inc.
     4   * All rights reserved.
     5   *
     6   * Redistribution and use in source and binary forms, with or without
     7   * modification, are permitted provided that the following conditions are
     8   * met:
     9   *
    10   *     * Redistributions of source code must retain the above copyright
    11   * notice, this list of conditions and the following disclaimer.
    12   *     * Redistributions in binary form must reproduce the above
    13   * copyright notice, this list of conditions and the following disclaimer
    14   * in the documentation and/or other materials provided with the
    15   * distribution.
    16   *     * Neither the name of Google Inc. nor the names of its
    17   * contributors may be used to endorse or promote products derived from
    18   * this software without specific prior written permission.
    19   *
    20   * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
    21   * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
    22   * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
    23   * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
    24   * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
    25   * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
    26   * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
    27   * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
    28   * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
    29   * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
    30   * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
    31   *
    32   */
    33  
    34  package main
    35  
    36  import (
    37  	"flag"
    38  	"net"
    39  	"strconv"
    40  
    41  	"google.golang.org/grpc"
    42  	"google.golang.org/grpc/credentials"
    43  	"google.golang.org/grpc/grpclog"
    44  	"google.golang.org/grpc/interop"
    45  	testpb "google.golang.org/grpc/interop/grpc_testing"
    46  )
    47  
    48  var (
    49  	useTLS   = flag.Bool("use_tls", false, "Connection uses TLS if true, else plain TCP")
    50  	certFile = flag.String("tls_cert_file", "testdata/server1.pem", "The TLS cert file")
    51  	keyFile  = flag.String("tls_key_file", "testdata/server1.key", "The TLS key file")
    52  	port     = flag.Int("port", 10000, "The server port")
    53  )
    54  
    55  func main() {
    56  	flag.Parse()
    57  	p := strconv.Itoa(*port)
    58  	lis, err := net.Listen("tcp", ":"+p)
    59  	if err != nil {
    60  		grpclog.Fatalf("failed to listen: %v", err)
    61  	}
    62  	var opts []grpc.ServerOption
    63  	if *useTLS {
    64  		creds, err := credentials.NewServerTLSFromFile(*certFile, *keyFile)
    65  		if err != nil {
    66  			grpclog.Fatalf("Failed to generate credentials %v", err)
    67  		}
    68  		opts = []grpc.ServerOption{grpc.Creds(creds)}
    69  	}
    70  	server := grpc.NewServer(opts...)
    71  	testpb.RegisterTestServiceServer(server, interop.NewTestServer())
    72  	server.Serve(lis)
    73  }