1 files changed, 126 insertions, 0 deletions
diff --git a/examples/perf-bench/perf-bench.go b/examples/perf-bench/perf-bench.go
new file mode 100644
index 0000000..24d3ebb
--- /dev/null
+++ b/examples/perf-bench/perf-bench.go
@@ -0,0 +1,126 @@
+// Copyright (c) 2017 Cisco and/or its affiliates.
+//
+// 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.
+
+// Binary simple-client is an example VPP management application that exercises the
+// govpp API on real-world use-cases.
+package main
+
+import (
+	"flag"
+	"fmt"
+	"log"
+	"time"
+
+	"github.com/pkg/profile"
+	"github.com/sirupsen/logrus"
+
+	"git.fd.io/govpp.git"
+	"git.fd.io/govpp.git/api"
+	"git.fd.io/govpp.git/core"
+	"git.fd.io/govpp.git/examples/bin_api/vpe"
+)
+
+const (
+	defaultSyncRequestCount  = 1000
+	defaultAsyncRequestCount = 10000
+)
+
+func main() {
+	// parse optional flags
+	var sync, prof bool
+	var cnt int
+	flag.BoolVar(&sync, "sync", false, "run synchronous perf test")
+	flag.IntVar(&cnt, "count", 0, "count of requests to be sent to VPP")
+	flag.BoolVar(&prof, "prof", false, "generate profile data")
+	flag.Parse()
+
+	if cnt == 0 {
+		// no specific count defined - use defaults
+		if sync {
+			cnt = defaultSyncRequestCount
+		} else {
+			cnt = defaultAsyncRequestCount
+		}
+	}
+
+	if prof {
+		defer profile.Start().Stop()
+	}
+
+	// connect to VPP
+	conn, err := govpp.Connect("")
+	if err != nil {
+		log.Fatalln("Error:", err)
+	}
+	defer conn.Disconnect()
+
+	// create an API channel
+	ch, err := conn.NewAPIChannelBuffered(cnt, cnt)
+	if err != nil {
+		log.Fatalln("Error:", err)
+	}
+	defer ch.Close()
+
+	// log only errors
+	core.SetLogger(&logrus.Logger{Level: logrus.ErrorLevel})
+
+	// run the test & measure the time
+	start := time.Now()
+
+	if sync {
+		// run synchronous test
+		syncTest(ch, cnt)
+	} else {
+		// run asynchronous test
+		asyncTest(ch, cnt)
+	}
+
+	elapsed := time.Since(start)
+	fmt.Println("Test took:", elapsed)
+	fmt.Printf("Requests per second: %.0f\n", float64(cnt)/elapsed.Seconds())
+}
+
+func syncTest(ch api.Channel, cnt int) {
+	fmt.Printf("Running synchronous perf test with %d requests...\n", cnt)
+
+	for i := 0; i < cnt; i++ {
+		req := &vpe.ControlPing{}
+		reply := &vpe.ControlPingReply{}
+
+		if err := ch.SendRequest(req).ReceiveReply(reply); err != nil {
+			log.Fatalln("Error in reply:", err)
+		}
+	}
+}
+
+func asyncTest(ch api.Channel, cnt int) {
+	fmt.Printf("Running asynchronous perf test with %d requests...\n", cnt)
+
+	ctxChan := make(chan api.RequestCtx, cnt)
+
+	go func() {
+		for i := 0; i < cnt; i++ {
+			ctxChan <- ch.SendRequest(&vpe.ControlPing{})
+		}
+		close(ctxChan)
+		fmt.Printf("Sending asynchronous requests finished\n")
+	}()
+
+	for ctx := range ctxChan {
+		reply := &vpe.ControlPingReply{}
+		if err := ctx.ReceiveReply(reply); err != nil {
+			log.Fatalln("Error in reply:", err)
+		}
+	}
+}