[flink-benchmarks] branch master updated: [FLINK-21731] Add benchmarks for DefaultScheduler's creation, scheduling and deploying

[zh...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

zhuzh pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/flink-benchmarks.git


The following commit(s) were added to refs/heads/master by this push:
     new c5d936d  [FLINK-21731] Add benchmarks for DefaultScheduler's creation, scheduling and deploying
c5d936d is described below

commit c5d936df12b3193e0a6b1aa57c16c9a7ce822581
Author: Thesharing <cy...@outlook.com>
AuthorDate: Thu Mar 25 19:39:40 2021 +0800

    [FLINK-21731] Add benchmarks for DefaultScheduler's creation, scheduling and deploying
---
 .../e2e/CreateSchedulerBenchmarkExecutor.java      | 65 ++++++++++++++++++++++
 .../SchedulingAndDeployingBenchmarkExecutor.java   | 64 +++++++++++++++++++++
 2 files changed, 129 insertions(+)

diff --git a/src/main/java/org/apache/flink/scheduler/benchmark/e2e/CreateSchedulerBenchmarkExecutor.java b/src/main/java/org/apache/flink/scheduler/benchmark/e2e/CreateSchedulerBenchmarkExecutor.java
new file mode 100644
index 0000000..2a7d03f
--- /dev/null
+++ b/src/main/java/org/apache/flink/scheduler/benchmark/e2e/CreateSchedulerBenchmarkExecutor.java
@@ -0,0 +1,65 @@
+/*
+ * 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 org.apache.flink.scheduler.benchmark.e2e;
+
+import org.apache.flink.runtime.scheduler.benchmark.JobConfiguration;
+import org.apache.flink.runtime.scheduler.benchmark.e2e.CreateSchedulerBenchmark;
+import org.apache.flink.scheduler.benchmark.SchedulerBenchmarkBase;
+
+import org.openjdk.jmh.annotations.Benchmark;
+import org.openjdk.jmh.annotations.BenchmarkMode;
+import org.openjdk.jmh.annotations.Level;
+import org.openjdk.jmh.annotations.Mode;
+import org.openjdk.jmh.annotations.Param;
+import org.openjdk.jmh.annotations.Setup;
+import org.openjdk.jmh.annotations.TearDown;
+import org.openjdk.jmh.infra.Blackhole;
+import org.openjdk.jmh.runner.RunnerException;
+
+/**
+ * The benchmark of creating the scheduler in a STREAMING/BATCH job.
+ */
+public class CreateSchedulerBenchmarkExecutor extends SchedulerBenchmarkBase {
+
+	@Param({"BATCH", "STREAMING"})
+	private JobConfiguration jobConfiguration;
+
+	private CreateSchedulerBenchmark benchmark;
+
+	public static void main(String[] args) throws RunnerException {
+		runBenchmark(CreateSchedulerBenchmarkExecutor.class);
+	}
+
+	@Setup(Level.Trial)
+	public void setup() throws Exception {
+		benchmark = new CreateSchedulerBenchmark();
+		benchmark.setup(jobConfiguration);
+	}
+
+	@Benchmark
+	@BenchmarkMode(Mode.SingleShotTime)
+	public void createScheduler(Blackhole blackhole) throws Exception {
+		blackhole.consume(benchmark.createScheduler());
+	}
+
+	@TearDown(Level.Trial)
+	public void teardownTrial() throws Exception {
+		benchmark.teardown();
+	}
+}
diff --git a/src/main/java/org/apache/flink/scheduler/benchmark/e2e/SchedulingAndDeployingBenchmarkExecutor.java b/src/main/java/org/apache/flink/scheduler/benchmark/e2e/SchedulingAndDeployingBenchmarkExecutor.java
new file mode 100644
index 0000000..14968e6
--- /dev/null
+++ b/src/main/java/org/apache/flink/scheduler/benchmark/e2e/SchedulingAndDeployingBenchmarkExecutor.java
@@ -0,0 +1,64 @@
+/*
+ * 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 org.apache.flink.scheduler.benchmark.e2e;
+
+import org.apache.flink.runtime.scheduler.benchmark.JobConfiguration;
+import org.apache.flink.runtime.scheduler.benchmark.e2e.SchedulingAndDeployingBenchmark;
+import org.apache.flink.scheduler.benchmark.SchedulerBenchmarkBase;
+
+import org.openjdk.jmh.annotations.Benchmark;
+import org.openjdk.jmh.annotations.BenchmarkMode;
+import org.openjdk.jmh.annotations.Level;
+import org.openjdk.jmh.annotations.Mode;
+import org.openjdk.jmh.annotations.Param;
+import org.openjdk.jmh.annotations.Setup;
+import org.openjdk.jmh.annotations.TearDown;
+import org.openjdk.jmh.runner.RunnerException;
+
+/**
+ * The benchmark of scheduling and deploying tasks in a STREAMING/BATCH job.
+ */
+public class SchedulingAndDeployingBenchmarkExecutor extends SchedulerBenchmarkBase {
+
+	@Param({"BATCH", "STREAMING"})
+	private JobConfiguration jobConfiguration;
+
+	private SchedulingAndDeployingBenchmark benchmark;
+
+	public static void main(String[] args) throws RunnerException {
+		runBenchmark(SchedulingAndDeployingBenchmarkExecutor.class);
+	}
+
+	@Setup(Level.Trial)
+	public void setup() throws Exception {
+		benchmark = new SchedulingAndDeployingBenchmark();
+		benchmark.setup(jobConfiguration);
+	}
+
+	@Benchmark
+	@BenchmarkMode(Mode.SingleShotTime)
+	public void startScheduling() throws Exception {
+		benchmark.startScheduling();
+	}
+
+	@TearDown(Level.Trial)
+	public void teardownTrial() throws Exception {
+		benchmark.teardown();
+	}
+}