Function findAndMarkRuntimeMap

pkg/controller/rtctrl/runtime_dispatcher_test.go:283–332

(b *testing.B, args *benchmarkRuntimeArgs)

}

func findAndMarkRuntimeMap(b *testing.B, args *benchmarkRuntimeArgs) {
	rtMap := NewRuntimeManager(getFuncletNode(), &RuntimeManagerParameters{MaxRuntimeIdle: 10, MaxRunnerDefunct: 30})
	for i := 0; i < args.RuntimeCount; i++ {
		rtParams := &NewRuntimeParameters{
			RuntimeID:               "runtime" + strconv.Itoa(i),
			ConcurrentMode:          true,
			StreamMode:              false,
			WaitRuntimeAliveTimeout: 3,
			Resource:                &api.Resource{},
		}
		rt := rtMap.NewRuntime(rtParams)

		params := &startRunnerParams{}
		rt.initRunner(params)
		str := "commitID" + strconv.Itoa(i%args.FunctionCount)
		input := &InvocationInput{
			Configuration: &api.FunctionConfiguration{
				CommitID: &str,
				FunctionConfiguration: lambda.FunctionConfiguration{
					MemorySize: &minMemory,
				},
			},
			WithStreamMode: false,
		}

		rtMap.OccupyColdRuntime(input)
		rt.Release()
	}

	maxParallel := int(args.Concurrency) * args.RuntimeCount
	p := maxParallel / runtime.GOMAXPROCS(0)
	b.Log("Parallelism ", p*runtime.GOMAXPROCS(0))
	b.SetParallelism(p)
	b.RunParallel(func(pb *testing.PB) {
		for i := 0; pb.Next(); i++ {
			commitID := "commitID" + strconv.Itoa(i%args.FunctionCount)
			rt := rtMap.FindWarmRuntime(&InvocationInput{
				Configuration: &api.FunctionConfiguration{
					CommitID:           &commitID,
					PodConcurrentQuota: args.Concurrency,
				},
			})
			if rt != nil {
				rt.Release()
			} else {
				b.Error("empty runtime", i)
			}
		}
	})
}

func BenchmarkFindAndMark_1Conc1Func10Rt(b *testing.B) {
	findAndMarkRuntimeMap(b, &benchmarkRuntimeArgs{