Bench fcs zkp

cmd/benchmarking/api_bench_test.go

@@ -0,0 +1,48 @@
+/*
+Copyright IBM Corp All Rights Reserved.
+
+SPDX-License-Identifier: Apache-2.0
+*/
+package benchmarking
+
+import (
+	"path"
+	"testing"
+
+	"github.com/hyperledger-labs/fabric-smart-client/integration/benchmark/node"
+	viewregistry "github.com/hyperledger-labs/fabric-smart-client/platform/view/services/view"
+	"github.com/stretchr/testify/require"
+)
+
+func BenchmarkAPI(b *testing.B) {
+	testdataPath := b.TempDir()
+	nodeConfPath := path.Join(testdataPath, "fsc", "nodes", "test-node.0")
+
+	err := node.GenerateConfig(testdataPath)
+	require.NoError(b, err)
+
+	n, err := node.SetupNode(nodeConfPath, node.NamedFactory{
+		Name:    "zkp",
+		Factory: &TokenTxVerifyViewFactory{},
+	})
+	require.NoError(b, err)
+	defer n.Stop()
+
+	vm, err := viewregistry.GetManager(n)
+	require.NoError(b, err)
+
+	params := &TokenTxVerifyParams{}
+	proof, err := GenerateProofData(params)
+	require.NoError(b, err)
+	params.Proof, err = proof.ToWire()
+	require.NoError(b, err)
+
+	wl := node.Workload{
+		Name:    "zkp",
+		Factory: &TokenTxVerifyViewFactory{},
+		Params:  params,
+	}
+
+	b.ResetTimer()
+	node.RunAPIBenchmark(b, vm, wl)
+}

cmd/benchmarking/api_grpc_bench_test.go

@@ -0,0 +1,51 @@
+/*
+Copyright IBM Corp All Rights Reserved.
+
+SPDX-License-Identifier: Apache-2.0
+*/
+
+package benchmarking
+
+import (
+	"path"
+	"testing"
+
+	"github.com/hyperledger-labs/fabric-smart-client/integration/benchmark/node"
+	"github.com/stretchr/testify/require"
+	"google.golang.org/grpc/benchmark/flags"
+)
+
+var numConn = flags.IntSlice("numConn", []int{1, 2, 4, 8}, "Number of grpc client connections - may be a comma-separated list")
+
+// BenchmarkAPIGRPC exercises the ViewAPI via grpc client.
+// The proof is pre-computed and embedded in the workload params so every
+// gRPC request carries it.
+func BenchmarkAPIGRPC(b *testing.B) {
+	testdataPath := b.TempDir()
+	nodeConfPath := path.Join(testdataPath, "fsc", "nodes", "test-node.0")
+	clientConfPath := path.Join(nodeConfPath, "client-config.yaml")
+
+	err := node.GenerateConfig(testdataPath)
+	require.NoError(b, err)
+
+	n, err := node.SetupNode(nodeConfPath, node.NamedFactory{
+		Name:    "zkp",
+		Factory: &TokenTxVerifyViewFactory{},
+	})
+	require.NoError(b, err)
+	defer n.Stop()
+
+	params := &TokenTxVerifyParams{}
+	proof, err := GenerateProofData(params)
+	require.NoError(b, err)
+	params.Proof, err = proof.ToWire()
+	require.NoError(b, err)
+
+	wl := node.Workload{
+		Name:    "zkp",
+		Factory: &TokenTxVerifyViewFactory{},
+		Params:  params,
+	}
+
+	node.RunAPIGRPCBenchmark(b, wl, clientConfPath, *numConn)
+}

cmd/benchmarking/client/main.go

@@ -0,0 +1,70 @@
+/*
+Copyright IBM Corp All Rights Reserved.
+
+SPDX-License-Identifier: Apache-2.0
+*/
+
+package main
+
+import (
+	"flag"
+	"fmt"
+	"path"
+	"runtime"
+	"time"
+
+	"github.com/hyperledger-labs/fabric-smart-client/integration/benchmark/node"
+	"github.com/hyperledger-labs/fabric-token-sdk/cmd/benchmarking"
+	"google.golang.org/grpc/benchmark/flags"
+)
+
+var (
+	numConn      = flags.IntSlice("numConn", []int{1, 2}, "Number of grpc client connections - may be a comma-separated list")
+	numWorker    = flags.IntSlice("cpu", []int{1, 2, 4, 8}, "Number of concurrent worker - may be a comma-separated list")
+	workloadFlag = flags.StringSlice("workloads", []string{"sign"}, "Workloads to execute - may be a comma-separated list")
+	warmupDur    = flag.Duration("warmup", 5*time.Second, "Warmup duration")
+	duration     = flag.Duration("benchtime", 10*time.Second, "Duration for every execution")
+	count        = flag.Int("count", 1, "Number of executions")
+)
+
+func main() {
+	flag.Parse()
+
+	runtime.GOMAXPROCS(runtime.NumCPU())
+
+	testdataPath := "./out/testdata" // for local debugging you can set testdataPath := "out/testdata"
+	clientConfPath := path.Join(testdataPath, "fsc", "nodes", "test-node.0", "client-config.yaml")
+
+	params := &benchmarking.TokenTxVerifyParams{}
+	fmt.Println("Pre-computing ZK proof...")
+	proof, err := benchmarking.GenerateProofData(params)
+	if err != nil {
+		panic(fmt.Sprintf("failed to pre-compute proof: %v", err))
+	}
+
+	params.Proof, err = proof.ToWire()
+	if err != nil {
+		panic(fmt.Sprintf("failed to marshal proof: %v", err))
+	}
+	fmt.Println("Proof ready.")
+
+	zkpWorkload := node.Workload{
+		Name:    "zkp",
+		Factory: &benchmarking.TokenTxVerifyViewFactory{},
+		Params:  params,
+	}
+
+	selected := make([]node.Workload, 0, len(*workloadFlag))
+	selected = append(selected, zkpWorkload)
+
+	node.RunRemoteBenchmarkSuite(node.RemoteBenchmarkConfig{
+		Workloads:      selected,
+		ClientConfPath: clientConfPath,
+		ConnCounts:     *numConn,
+		WorkerCounts:   *numWorker,
+		WarmupDur:      *warmupDur,
+		BenchTime:      *duration,
+		Count:          *count,
+		BenchName:      "BenchmarkAPIGRPCRemote",
+	})
+}