perf/pulsar-perf-go.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 (
	"fmt"
	"net"
	"net/http"
	_ "net/http/pprof"

	"github.com/spf13/cobra"

	log "github.com/sirupsen/logrus"
)

type ClientArgs struct {
	ServiceURL string
}

var clientArgs ClientArgs

func main() {
	// use `go tool pprof http://localhost:3000/debug/pprof/profile` to get pprof file(cpu info)
	// use `go tool pprof http://localhost:3000/debug/pprof/heap` to get inuse_space file
	go func() {
		listenAddr := net.JoinHostPort("localhost", "3000")
		fmt.Printf("Profile server listening on %s\n", listenAddr)
		profileRedirect := http.RedirectHandler("/debug/pprof", http.StatusSeeOther)
		http.Handle("/", profileRedirect)
		err := fmt.Errorf("%v", http.ListenAndServe(listenAddr, nil))
		fmt.Println(err.Error())
	}()

	log.SetFormatter(&log.TextFormatter{
		FullTimestamp:   true,
		TimestampFormat: "15:04:05.000",
	})
	log.SetLevel(log.InfoLevel)

	initProducer()
	initConsumer()

	var rootCmd = &cobra.Command{Use: "pulsar-perf-go"}
	rootCmd.Flags().StringVarP(&clientArgs.ServiceURL, "service-url", "u",
		"pulsar://localhost:6650", "The Pulsar service URL")
	rootCmd.AddCommand(cmdProduce, cmdConsume)

	err := rootCmd.Execute()
	if err != nil {
		panic("execute root cmd error, please check.")
	}
}