import quincy 17.2.0

[ceph.git] / ceph / src / arrow / go / arrow / internal / flight_integration / cmd / arrow-flight-integration-client / main.go

diff --git a/ceph/src/arrow/go/arrow/internal/flight_integration/cmd/arrow-flight-integration-client/main.go b/ceph/src/arrow/go/arrow/internal/flight_integration/cmd/arrow-flight-integration-client/main.go
new file mode 100644 (file)
index 0000000..db8abd0
--- /dev/null
+++ b/ceph/src/arrow/go/arrow/internal/flight_integration/cmd/arrow-flight-integration-client/main.go
@@ -0,0 +1,53 @@
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+// Client for use with Arrow Flight Integration tests via archery
+package main
+
+import (
+       "flag"
+       "fmt"
+       "time"
+
+       "github.com/apache/arrow/go/v6/arrow/internal/flight_integration"
+       "google.golang.org/grpc"
+)
+
+var (
+       host     = flag.String("host", "localhost", "Server host to connect to")
+       port     = flag.Int("port", 31337, "Server port to connect to")
+       path     = flag.String("path", "", "Resource path to request")
+       scenario = flag.String("scenario", "", "Integration test scenario to run")
+)
+
+const retries = 3
+
+func main() {
+       flag.Parse()
+
+       c := flight_integration.GetScenario(*scenario, *path)
+       var err error
+       for i := 0; i < retries; i++ {
+               err = c.RunClient(fmt.Sprintf("%s:%d", *host, *port), grpc.WithInsecure())
+               if err == nil {
+                       break
+               }
+               time.Sleep(time.Duration(i+1) * 500 * time.Millisecond)
+       }
+       if err != nil {
+               panic(err)
+       }
+}