main.go

// 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.

package main

import (
	"github.com/perf-tool/perf-network-go/perfn"
	"github.com/perf-tool/perf-network-go/util"
	"os"
)

func main() {
	var config perfn.Config
	envProtocolType := os.Getenv("PROTOCOL_TYPE")
	if envProtocolType == "udp" {
		config.ProtocolType = perfn.ProtocolTypeUdp
	} else if envProtocolType == "tcp" {
		config.ProtocolType = perfn.ProtocolTypeTcp
	} else if envProtocolType == "http" {
		config.ProtocolType = perfn.ProtocolTypeHttp
	}
	envCommType := os.Getenv("COMM_TYPE")
	if envCommType == "client" {
		config.CommType = perfn.CommTypeClient
	} else if envCommType == "server" {
		config.CommType = perfn.CommTypeServer
	}
	config.PrometheusMetricsDisable = util.GetEnvBool("PROMETHEUS_METRICS_DISABLE", false)
	err := perfn.Run(config)
	if err != nil {
		panic(err)
	}
}