63 lines
2.1 KiB
Groovy
63 lines
2.1 KiB
Groovy
// 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.
|
|
|
|
import org.apache.doris.regression.suite.Suite
|
|
|
|
Suite.metaClass.get_catalog_id = {String catalog_name /* param */ ->
|
|
String catalog_id;
|
|
def catalogs = sql """show proc '/catalogs'"""
|
|
for (catalog in catalogs) {
|
|
if (catalog[1].equals(catalog_name)) {
|
|
catalog_id = catalog[0]
|
|
break
|
|
}
|
|
}
|
|
log.info("get catalogid: " + catalog_id)
|
|
return catalog_id
|
|
}
|
|
|
|
|
|
Suite.metaClass.get_database_id = {String catalog_name, String db_name /* param */ ->
|
|
String database_id;
|
|
def catalog_id = get_catalog_id(catalog_name)
|
|
def dbs = sql """show proc '/catalogs/${catalog_id}'"""
|
|
for (db in dbs) {
|
|
if (db[1].equals(db_name)) {
|
|
database_id = db[0]
|
|
break
|
|
}
|
|
}
|
|
log.info("get database_id: " + database_id)
|
|
return database_id
|
|
}
|
|
|
|
|
|
Suite.metaClass.get_table_id = {String catalog_name, String db_name, String tb_name /* param */ ->
|
|
String table_id;
|
|
def catalog_id = get_catalog_id(catalog_name)
|
|
def database_id = get_database_id(catalog_name, db_name)
|
|
def tbs = sql """show proc '/catalogs/${catalog_id}/${database_id}'"""
|
|
for (tb in tbs) {
|
|
if (tb[1].equals(tb_name)) {
|
|
table_id = tb[0]
|
|
break
|
|
}
|
|
}
|
|
log.info("get table_id: " + table_id)
|
|
return table_id
|
|
}
|