ceph/src/arrow/go/arrow/internal/flight_integration/cmd/arrow-flight-integration-server/main.go

   1 // Licensed to the Apache Software Foundation (ASF) under one
   2 // or more contributor license agreements.  See the NOTICE file
   3 // distributed with this work for additional information
   4 // regarding copyright ownership.  The ASF licenses this file
   5 // to you under the Apache License, Version 2.0 (the
   6 // "License"); you may not use this file except in compliance
   7 // with the License.  You may obtain a copy of the License at
   8 //
   9 // http://www.apache.org/licenses/LICENSE-2.0
  10 //
  11 // Unless required by applicable law or agreed to in writing, software
  12 // distributed under the License is distributed on an "AS IS" BASIS,
  13 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14 // See the License for the specific language governing permissions and
  15 // limitations under the License.
  16
  17 package main
  18
  19 import (
  20         "flag"
  21         "fmt"
  22         "net"
  23         "os"
  24         "syscall"
  25
  26         "github.com/apache/arrow/go/v6/arrow/internal/flight_integration"
  27 )
  28
  29 var (
  30         port     = flag.Int("port", 31337, "Server port to listen on")
  31         scenario = flag.String("scenario", "", "Integration test scenario to run")
  32 )
  33
  34 func main() {
  35         flag.Parse()
  36
  37         s := flight_integration.GetScenario(*scenario)
  38         srv := s.MakeServer(*port)
  39         srv.SetShutdownOnSignals(syscall.SIGTERM, os.Interrupt)
  40         _, p, _ := net.SplitHostPort(srv.Addr().String())
  41         fmt.Printf("Server listening on localhost:%s\n", p)
  42         srv.Serve()
  43 }