This is an automated email from the ASF dual-hosted git repository.
dataroaring pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/doris.git
The following commit(s) were added to refs/heads/master by this push:
new f3cb0c24ee [enhancement](test) add restore action and s3 helper
methond (#12084)
f3cb0c24ee is described below
commit f3cb0c24eeb27c38ef95fa5e4aff3207951e5ded
Author: Yongqiang YANG <98214048+dataroar...@users.noreply.github.com>
AuthorDate: Wed Aug 31 23:08:23 2022 +0800
[enhancement](test) add restore action and s3 helper methond (#12084)
Co-authored-by: morrySnow <morrys...@126.com>
Co-authored-by: SWJTU-ZhangLei <1091517...@qq.com>
---
regression-test/framework/pom.xml | 5 +
.../doris/regression/action/RestoreAction.groovy | 213 +++++++++++++++++++++
.../org/apache/doris/regression/suite/Suite.groovy | 37 ++++
3 files changed, 255 insertions(+)
diff --git a/regression-test/framework/pom.xml
b/regression-test/framework/pom.xml
index a306b3b8fd..a39c15abfb 100644
--- a/regression-test/framework/pom.xml
+++ b/regression-test/framework/pom.xml
@@ -245,6 +245,11 @@ under the License.
<artifactId>commons-csv</artifactId>
<version>1.9.0</version>
</dependency>
+ <dependency>
+ <groupId>org.apache.commons</groupId>
+ <artifactId>commons-lang3</artifactId>
+ <version>3.9</version>
+ </dependency>
<dependency>
<groupId>com.google.guava</groupId>
<artifactId>guava</artifactId>
diff --git
a/regression-test/framework/src/main/groovy/org/apache/doris/regression/action/RestoreAction.groovy
b/regression-test/framework/src/main/groovy/org/apache/doris/regression/action/RestoreAction.groovy
new file mode 100644
index 0000000000..6b21a3ce2a
--- /dev/null
+++
b/regression-test/framework/src/main/groovy/org/apache/doris/regression/action/RestoreAction.groovy
@@ -0,0 +1,213 @@
+// 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.doris.regression.action
+
+import groovy.util.logging.Slf4j
+import org.apache.commons.collections.CollectionUtils
+import org.apache.commons.lang3.StringUtils
+import org.apache.doris.regression.suite.SuiteContext
+import org.apache.doris.regression.util.JdbcUtils
+
+import java.sql.ResultSetMetaData
+
+@Slf4j
+class RestoreAction implements SuiteAction {
+ public final InetSocketAddress address
+ public final String user
+ public final String password
+
+ String db
+ List<String> tables
+ String location
+ String region
+ String ak
+ String sk
+ String endpoint
+ String repository
+ String snapshot
+ String timestamp
+ int replicationNum
+ long timeout
+
+ SuiteContext context
+
+ RestoreAction(SuiteContext context) {
+ this.address = context.config.feHttpInetSocketAddress
+ this.user = context.config.feHttpUser
+ this.password = context.config.feHttpPassword
+ this.db = context.config.defaultDb
+ this.context = context
+ }
+
+ void db(String db) {
+ this.db = db
+ }
+
+ void location(String location) {
+ this.location = location
+ }
+
+ void region(String region) {
+ this.region = region
+ }
+
+ void ak(String ak) {
+ this.ak = ak
+ }
+
+ void sk(String sk) {
+ this.sk = sk
+ }
+
+ void endpoint(String endpoint) {
+ this.endpoint = endpoint
+ }
+
+ void repository(String repo) {
+ this.repository = repo
+ }
+
+ void snapshot(String snapshot) {
+ this.snapshot = snapshot
+ }
+
+ void timestamp(String timestamp) {
+ this.timestamp = timestamp
+ }
+
+ void replicationNum(int replicaNum) {
+ this.replicationNum = replicaNum
+ }
+
+ void timeout(long timeout) {
+ this.timeout = timeout
+ }
+
+ void tables(String tables) {
+ if (StringUtils.isNotEmpty(tables))
+ this.tables = tables.split(",")
+ }
+
+ @Override
+ void run() {
+ // create repo
+ createRepository()
+ // restore
+ restore()
+ // check result
+ checkRestore()
+ }
+
+ private void createRepository() {
+ String showRepoSql = """
+SHOW REPOSITORIES
+"""
+ String dropRepoSql = """
+DROP REPOSITORY ${repository}
+"""
+ List<List<Object>> showRepoRes = null
+ ResultSetMetaData meta = null
+ (showRepoRes, meta) = JdbcUtils.executeToList(context.conn,
showRepoSql)
+ for (List<Object> row : showRepoRes) {
+ if (row.size() > 1 &&
repository.equalsIgnoreCase(row[1].toString())) {
+ JdbcUtils.executeToList(context.conn, dropRepoSql)
+ break
+ }
+ }
+
+ String createRepoSql = """
+CREATE READ ONLY REPOSITORY `${repository}`
+WITH S3 ON LOCATION "${location}"
+PROPERTIES (
+ "AWS_ENDPOINT" = "${endpoint}",
+ "AWS_ACCESS_KEY" = "${ak}",
+ "AWS_SECRET_KEY" = "${sk}",
+ "AWS_REGION" = "${region}"
+)
+"""
+ JdbcUtils.executeToList(context.conn, createRepoSql)
+ }
+
+ private void restore() {
+ List<List<Object>> showTablesRes = null
+ ResultSetMetaData meta = null
+
+ if (CollectionUtils.isEmpty(tables)) {
+ tables = new ArrayList<>()
+ String showTables = """SHOW TABLES"""
+ (showTablesRes, meta) = JdbcUtils.executeToList(context.conn,
showTables)
+ for (List<Object> tableRes : showTablesRes) {
+ tables.add(tableRes[0].toString())
+ }
+ }
+ if (CollectionUtils.isNotEmpty(tables)) {
+ for (String t : tables) {
+ String dropTableSql = """DROP TABLE IF EXISTS `${t}` FORCE"""
+ JdbcUtils.executeToList(context.conn, dropTableSql)
+ log.info("drop table '${t}'")
+ }
+ }
+
+ String restoreSql = """
+RESTORE SNAPSHOT `${getRealDbName()}`.`${snapshot}`
+FROM ${repository}
+"""
+ if (CollectionUtils.isNotEmpty(tables)) {
+ restoreSql += """
+ON (
+ `${tables.join("`,\n `")}`
+)
+"""
+ }
+ restoreSql += """
+PROPERTIES (
+ "backup_timestamp" = "${timestamp}",
+ "replication_num" = "${replicationNum}",
+ "timeout" = "${timeout}"
+)
+"""
+ JdbcUtils.executeToList(context.conn, restoreSql)
+ }
+
+ private void checkRestore() {
+ String showRestoreSql = """
+SHOW RESTORE FROM `${getRealDbName()}`
+"""
+ List<List<Object>> result = null
+ ResultSetMetaData meta = null
+
+ while (true) {
+ (result, meta) = JdbcUtils.executeToList(context.conn,
showRestoreSql)
+ if (!result.empty && result[result.size() - 1].size() >= 5) {
+ def status = result[result.size() - 1][4].toString()
+ if ("CANCELLED".equalsIgnoreCase(status)) {
+ log.info("restore snapshot '${snapshot}' failed.")
+ throw new IllegalStateException("restore ${snapshot}
failed.")
+ } else if ("FINISHED".equalsIgnoreCase(status)) {
+ log.info("restore snapshot '${snapshot}' success.")
+ break
+ }
+ }
+ sleep(10000)
+ }
+ }
+
+ private String getRealDbName() {
+ return context.config.getDbNameByFile(context.file)
+ }
+}
diff --git
a/regression-test/framework/src/main/groovy/org/apache/doris/regression/suite/Suite.groovy
b/regression-test/framework/src/main/groovy/org/apache/doris/regression/suite/Suite.groovy
index c8ad15d48d..5b73eb4116 100644
---
a/regression-test/framework/src/main/groovy/org/apache/doris/regression/suite/Suite.groovy
+++
b/regression-test/framework/src/main/groovy/org/apache/doris/regression/suite/Suite.groovy
@@ -25,6 +25,7 @@ import com.google.common.collect.ImmutableList
import org.apache.doris.regression.util.DataUtils
import org.apache.doris.regression.util.OutputUtils
import org.apache.doris.regression.action.ExplainAction
+import org.apache.doris.regression.action.RestoreAction
import org.apache.doris.regression.action.StreamLoadAction
import org.apache.doris.regression.action.SuiteAction
import org.apache.doris.regression.action.TestAction
@@ -296,6 +297,38 @@ class Suite implements GroovyInterceptable {
return remotePath;
}
+ String getS3Region() {
+ String s3Region = context.config.otherConfigs.get("s3Region");
+ return s3Region
+ }
+
+ String getS3BucketName() {
+ String s3BucketName = context.config.otherConfigs.get("s3BucketName");
+ return s3BucketName
+ }
+
+ String getS3Endpoint() {
+ String s3Endpoint = context.config.otherConfigs.get("s3Endpoint");
+ return s3Endpoint
+ }
+
+ String getS3AK() {
+ String ak = context.config.otherConfigs.get("ak");
+ return ak
+ }
+
+ String getS3SK() {
+ String sk = context.config.otherConfigs.get("sk");
+ return sk
+ }
+
+ String getS3Url() {
+ String s3BucketName = context.config.otherConfigs.get("s3BucketName");
+ String s3Endpoint = context.config.otherConfigs.get("s3Endpoint");
+ String s3Url = "http://${s3BucketName}.${s3Endpoint}";
+ return s3Url
+ }
+
int getTotalLine(String filePath) {
def file = new File(filePath)
int lines = 0;
@@ -322,6 +355,10 @@ class Suite implements GroovyInterceptable {
runAction(new StreamLoadAction(context), actionSupplier)
}
+ void restore(Closure actionSupplier) {
+ runAction(new RestoreAction(context), actionSupplier)
+ }
+
void runAction(SuiteAction action, Closure actionSupplier) {
actionSupplier.setDelegate(action)
actionSupplier.setResolveStrategy(Closure.DELEGATE_FIRST)
---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscr...@doris.apache.org
For additional commands, e-mail: commits-h...@doris.apache.org
