package main
import (
+ "context"
"flag"
"fmt"
"log"
+ "os"
"time"
"github.com/pkg/profile"
"github.com/sirupsen/logrus"
- "git.fd.io/govpp.git/adapter"
"git.fd.io/govpp.git/adapter/socketclient"
- "git.fd.io/govpp.git/adapter/vppapiclient"
+ "git.fd.io/govpp.git/adapter/statsclient"
"git.fd.io/govpp.git/api"
+ "git.fd.io/govpp.git/binapi/vpe"
"git.fd.io/govpp.git/core"
- "git.fd.io/govpp.git/examples/bin_api/vpe"
)
const (
func main() {
// parse optional flags
- var sync, prof, sock bool
+ var sync bool
var cnt int
+ var sock, prof string
flag.BoolVar(&sync, "sync", false, "run synchronous perf test")
- flag.BoolVar(&sock, "sock", false, "use socket client for VPP API")
+ flag.StringVar(&sock, "api-socket", socketclient.DefaultSocketName, "Path to VPP API socket")
+ flag.String("stats-socket", statsclient.DefaultSocketName, "Path to VPP stats socket")
flag.IntVar(&cnt, "count", 0, "count of requests to be sent to VPP")
- flag.BoolVar(&prof, "prof", false, "generate profile data")
+ flag.StringVar(&prof, "prof", "", "enable profiling mode [mem, cpu]")
flag.Parse()
if cnt == 0 {
}
}
- if prof {
- defer profile.Start().Stop()
+ switch prof {
+ case "mem":
+ defer profile.Start(profile.MemProfile, profile.MemProfileRate(1)).Stop()
+ case "cpu":
+ defer profile.Start(profile.CPUProfile).Stop()
+ case "":
+ default:
+ fmt.Printf("invalid profiling mode: %q\n", prof)
+ flag.Usage()
+ os.Exit(1)
}
- var a adapter.VppAPI
- if sock {
- a = socketclient.NewVppClient("/run/vpp-api.sock")
- } else {
- a = vppapiclient.NewVppClient("")
- }
+ a := socketclient.NewVppClient(sock)
// connect to VPP
conn, err := core.Connect(a)
if sync {
// run synchronous test
syncTest(ch, cnt)
+ //syncTest2(conn, cnt)
} else {
// run asynchronous test
asyncTest(ch, cnt)
+ //asyncTest2(conn, cnt)
}
elapsed := time.Since(start)
}
}
+func syncTest2(conn api.Connection, cnt int) {
+ fmt.Printf("Running synchronous perf test with %d requests...\n", cnt)
+
+ stream, err := conn.NewStream(context.Background())
+ if err != nil {
+ log.Fatalln("Error NewStream:", err)
+ }
+ for i := 0; i < cnt; i++ {
+ if err := stream.SendMsg(&vpe.ControlPing{}); err != nil {
+ log.Fatalln("Error SendMsg:", err)
+ }
+ if msg, err := stream.RecvMsg(); err != nil {
+ log.Fatalln("Error RecvMsg:", err)
+ } else if _, ok := msg.(*vpe.ControlPingReply); ok {
+ // ok
+ } else {
+ log.Fatalf("unexpected reply: %v", msg.GetMessageName())
+ }
+ }
+}
+
func asyncTest(ch api.Channel, cnt int) {
fmt.Printf("Running asynchronous perf test with %d requests...\n", cnt)
}
}
}
+
+func asyncTest2(conn api.Connection, cnt int) {
+ fmt.Printf("Running asynchronous perf test with %d requests...\n", cnt)
+
+ ctxChan := make(chan api.Stream, cnt)
+
+ go func() {
+ for i := 0; i < cnt; i++ {
+ stream, err := conn.NewStream(context.Background())
+ if err != nil {
+ log.Fatalln("Error NewStream:", err)
+ }
+ if err := stream.SendMsg(&vpe.ControlPing{}); err != nil {
+ log.Fatalln("Error SendMsg:", err)
+ }
+ ctxChan <- stream
+ }
+ close(ctxChan)
+ fmt.Printf("Sending asynchronous requests finished\n")
+ }()
+
+ for ctx := range ctxChan {
+ if msg, err := ctx.RecvMsg(); err != nil {
+ log.Fatalln("Error RecvMsg:", err)
+ } else if _, ok := msg.(*vpe.ControlPingReply); ok {
+ // ok
+ } else {
+ log.Fatalf("unexpected reply: %v", msg.GetMessageName())
+ }
+ ctx.Close()
+ }
+}