common.sh

#!/bin/bash

# Copyright Istio Authors
#
# Licensed 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.

if [[ -z "${GATEWAY_URL:-}" ]];then
  if [[ -z "${GATEWAY_SERVICE_NAME:-}" ]];then
    GATEWAY_URL=$(kubectl -n istio-system get service istio-ingressgateway -o jsonpath='{.status.loadBalancer.ingress[0].ip}' || true)
  else
    GATEWAY_URL=$(kubectl -n istio-system get service "${GATEWAY_SERVICE_NAME}" -o jsonpath='{.status.loadBalancer.ingress[0].ip}' || true)
  fi
fi

HTTPS=${HTTPS:-"false"}
H2UPGRADE=${H2UPGRADE:-"false"}

function run_test() {
  local ns=${1:?"namespaces"}
  local prefix=${2:?"prefix name for service. typically svc-"}
  local injection_label=${3:?"injection label"}
  MULTI_CLUSTER=${MULTI_CLUSTER:-"false"}
  CLUSTER1=${CLUSTER1:-"false"}
  CLUSTER2=${CLUSTER2:-"false"}
  VM_ENABLED="${VM_ENABLED:-false}"
  YAML=$(mktemp).yml
  # shellcheck disable=SC2086
  helm -n ${ns} template \
          --set serviceNamePrefix="${prefix}" \
          --set Namespace="${ns}" \
          --set domain="${DNS_DOMAIN}" \
          --set ingress="${GATEWAY_URL}" \
          --set https="${HTTPS}" \
          --set h2upgrade="${H2UPGRADE}" \
          --set multicluster.enabled="${MULTI_CLUSTER}" \
          --set multicluster.cluster1="${CLUSTER1}" \
          --set multicluster.cluster2="${CLUSTER2}" \
          --set vm.enabled="${VM_ENABLED}" \
          . > "${YAML}"
  echo "Wrote ${YAML}"

  kubectl create ns "${ns}" || true
  kubectl label namespace "${ns}" "${injection_label}" --overwrite

   if [[ -z "${DELETE}" ]];then
    sleep 3
    kubectl -n "${ns}" apply -f "${YAML}"
  else
    kubectl -n "${ns}" delete -f "${YAML}" || true
    kubectl delete ns "${ns}"
  fi
  if [[ "${VM_ENABLED}" == "true" ]]; then
    VM_NAMESPACE="${ns}" VM_APP="${prefix}0-9"  ./bootstrap-vm.sh
    VM_NAMESPACE="${ns}" VM_APP="${prefix}0-9-0"  ./bootstrap-vm.sh
  fi
}

function start_servicegraphs() {
  local nn=${1:?"number of namespaces"}
  local min=${2:-"0"}
  local injection_label=${3:-"istio-injection=enabled"}
  local PERF_NAMESPACE_DELAY=${4:-30}

   # shellcheck disable=SC2004
   for ((ii=$min; ii<$nn; ii++)) {
    ns=$(printf 'service-graph%.2d' "${ii}")
    prefix=$(printf 'svc%.2d-' "${ii}")
    if [[ -z "${DELETE}" ]];then
      ${CMD} run_test "${ns}" "${prefix}" "${injection_label}"
      ${CMD} "${WD}/loadclient/setup_test.sh" "${ns}" "${prefix}" "${injection_label}"
    else
      ${CMD} "${WD}/loadclient/setup_test.sh" "${ns}" "${prefix}" "${injection_label}"
      ${CMD} run_test "${ns}" "${prefix}" "${injection_label}"
    fi

    sleep "${PERF_NAMESPACE_DELAY}"
  }
}