Add benchmark for doris

config/doris/sample_chbenchmark_config.xml

@@ -0,0 +1,140 @@
+<?xml version="1.0"?>
+<parameters>
+
+    <!-- Connection details -->
+    <type>MYSQL</type>
+    <driver>com.mysql.cj.jdbc.Driver</driver>
+    <url>jdbc:mysql://doris:9033/doris_100w_0215?rewriteBatchedStatements=true</url>
+    <!--    <url>jdbc:mysql://hz11-trino-arctic-0.jd.163.org:3306/icebergdb10w?rewriteBatchedStatements=true</url>-->
+    <username>root</username>
+    <password></password>
+    <isolation>TRANSACTION_READ_UNCOMMITTED</isolation>
+    <batchsize>128</batchsize>
+
+    <!-- Scale factor is the number of warehouses in TPCC -->
+    <scalefactor>100</scalefactor>
+
+    <!-- The workload -->
+    <!-- Number of terminal per workload -->
+    <terminals>10</terminals>
+
+    <!-- Extra Features (Commented Out) -->
+    <!-- Can be workload-specific -->
+    <!--    <terminals bench="tpcc">128</terminals>-->
+
+    <!-- Workload-specific options a marked with @bench=[workload_name] -->
+    <!-- Workload-specific number of terminals -->
+    <!-- <terminals bench="chbenchmark">2</terminals> -->
+
+    <works>
+        <work>
+            <!--            需要测试至少10min左右-->
+            <time>1800</time>
+
+            <!--            <rate bench="tpcc">2000</rate>-->
+            <rate>200</rate>
+            <rate bench="tpcc">disabled</rate>
+
+            <rate bench="chbenchmark4doris">unlimited</rate>
+
+            <!-- NOTE: TPCC workers won't be distributed evenly between warehouses if not all workers are active -->
+            <!--            <active_terminals>4</active_terminals>-->
+            <!--            <active_terminals bench="tpcc">128</active_terminals>-->
+
+            <!-- Specifies transaction weight for each workload. -->
+            <weights bench="tpcc">45,43,4,4,4</weights>
+            <weights bench="chbenchmark4doris">3, 2, 3, 2, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5</weights>
+
+        </work>
+    </works>
+
+
+    <!-- CH specific -->
+    <transactiontypes bench="chbenchmark4doris">
+        <transactiontype>
+            <name>Q1</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Q2</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Q3</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Q4</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Q5</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Q6</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Q7</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Q8</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Q9</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Q10</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Q11</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Q12</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Q13</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Q14</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Q15</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Q16</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Q17</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Q18</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Q19</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Q20</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Q21</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Q22</name>
+        </transactiontype>
+    </transactiontypes>
+
+    <!-- TPCC specific -->
+    <transactiontypes bench="tpcc">
+        <transactiontype>
+            <name>NewOrder</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Payment</name>
+        </transactiontype>
+        <transactiontype>
+            <name>OrderStatus</name>
+        </transactiontype>
+        <transactiontype>
+            <name>Delivery</name>
+        </transactiontype>
+        <transactiontype>
+            <name>StockLevel</name>
+        </transactiontype>
+    </transactiontypes>
+</parameters>

config/plugin.xml

@@ -19,4 +19,5 @@
     <plugin name="chbenchmarkForTrino">com.oltpbenchmark.benchmarks.chbenchmarkForTrino.chbenchmark.CHBenCHmark</plugin>
     <plugin name="chbenchmarkForSpark">com.oltpbenchmark.benchmarks.chbenchmarkForSpark.chbenchmark.CHBenCHmark</plugin>
     <plugin name="chbenchmark4impala">com.oltpbenchmark.benchmarks.chbenchmark4impala.CHBenCHmark</plugin>
+    <plugin name="chbenchmark4doris">com.oltpbenchmark.benchmarks.chbenchmark4doirs.CHBenCHmark</plugin>
 </plugins>

src/main/java/com/oltpbenchmark/benchmarks/chbenchmark4doirs/CHBenCHmark.java

@@ -0,0 +1,61 @@
+/*
+ * Copyright 2020 by OLTPBenchmark Project
+ *
+ * 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.
+ *
+ */
+
+package com.oltpbenchmark.benchmarks.chbenchmark4doirs;
+
+
+import com.oltpbenchmark.WorkloadConfiguration;
+import com.oltpbenchmark.api.BenchmarkModule;
+import com.oltpbenchmark.api.Loader;
+import com.oltpbenchmark.api.Worker;
+import com.oltpbenchmark.benchmarks.chbenchmark4doirs.queries.Q1;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.util.ArrayList;
+import java.util.List;
+
+public class CHBenCHmark extends BenchmarkModule {
+    private static final Logger LOG = LoggerFactory.getLogger(CHBenCHmark.class);
+
+    public CHBenCHmark(WorkloadConfiguration workConf) {
+        super(workConf);
+    }
+
+    protected Package getProcedurePackageImpl() {
+        return (Q1.class.getPackage());
+    }
+
+    @Override
+    protected List<Worker<? extends BenchmarkModule>> makeWorkersImpl() {
+        // HACK: Turn off terminal messages
+        List<Worker<? extends BenchmarkModule>> workers = new ArrayList<>();
+
+        int numTerminals = workConf.getTerminals();
+        LOG.info(String.format("Creating %d workers for CHBenCHMark", numTerminals));
+        for (int i = 0; i < numTerminals; i++) {
+            workers.add(new CHBenCHmarkWorker(this, i));
+        }
+
+        return workers;
+    }
+
+    protected Loader<CHBenCHmark> makeLoaderImpl() {
+        return new CHBenCHmarkLoader(this);
+    }
+
+}