1 files changed, 75 insertions, 0 deletions
diff --git a/src/arrow/dev/benchmarking/graphql_submit.sh b/src/arrow/dev/benchmarking/graphql_submit.sh
new file mode 100755
index 000000000..2eaab9cdf
--- /dev/null
+++ b/src/arrow/dev/benchmarking/graphql_submit.sh
@@ -0,0 +1,75 @@
+#!/usr/bin/env bash
+#
+# 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.
+#
+
+OPTIONS=("machine" "benchmarks" "runs")
+
+option=${1-help}
+datafile=${2-machine.json}
+uri=${3-localhost:5000/graphql}
+
+help() {
+  cat <<HELP
+  Submit data via GraphQL
+  
+  Usage:
+  ${0} [option] [JSON_file] [URI]
+
+  Arguments:
+    option    - $(echo ${OPTIONS[@]} | sed 's/ /|/g')
+    JSON_file - path to the submission file (default 'machine.json')
+    URI       - URI to submit to (default 'localhost:5000/graphql')
+HELP
+}
+
+escape_quote() {  sed 's/"/\\"/g'; }
+
+template() {
+  cat <<TEMPLATE
+  {
+    "query": "mutation (\$jsonb: JSON!){${1}(input:{fromJsonb:\$jsonb}){${2}}}",
+    "variables": {
+      "jsonb": "$(echo $(cat ${datafile}) | escape_quote )"
+    }
+  }
+TEMPLATE
+}
+
+submit () {
+  curl -X POST -H "Content-Type: application/json"  --data @<(template $1 $2) ${uri}
+}
+
+
+case "$1" in
+  machine)
+    submit ingestMachineView integer;;
+
+  benchmarks)
+    submit ingestBenchmarkView integers;;
+
+  runs)
+    if grep -q context <(head -n2 ${2})
+    then
+      submit ingestBenchmarkRunsWithContext bigInts
+    else
+      submit ingestBenchmarkRunView bigInts
+    fi;;
+
+  *)
+    help
+    exit 1
+esac