mirror of https://github.com/apache/cloudstack.git
72 lines
2.9 KiB
Java
72 lines
2.9 KiB
Java
// 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
|
|
// 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.cloud.utils.db;
|
|
|
|
import java.io.File;
|
|
import java.io.FileNotFoundException;
|
|
import java.io.FileReader;
|
|
import java.io.IOException;
|
|
import java.sql.Connection;
|
|
import java.sql.SQLException;
|
|
|
|
import com.cloud.utils.PropertiesUtil;
|
|
|
|
public class DbTestUtils {
|
|
|
|
public static void executeScript(String file, boolean autoCommit, boolean stopOnError) {
|
|
File cleanScript = PropertiesUtil.findConfigFile(file);
|
|
if (cleanScript == null) {
|
|
throw new RuntimeException("Unable to clean the database because I can't find " + file);
|
|
}
|
|
Connection conn = TransactionLegacy.getStandaloneConnection();
|
|
ScriptRunner runner = new ScriptRunner(conn, autoCommit, stopOnError);
|
|
try(FileReader reader = new FileReader(cleanScript);)
|
|
{
|
|
runner.runScript(reader);
|
|
conn.close();
|
|
} catch (FileNotFoundException e) {
|
|
throw new RuntimeException("Unable to read " + file, e);
|
|
}catch (IOException e) {
|
|
throw new RuntimeException("Unable to read " + file, e);
|
|
} catch (SQLException e) {
|
|
throw new RuntimeException("Unable to close DB connection", e);
|
|
}
|
|
}
|
|
|
|
public static void executeUsageScript(String file, boolean autoCommit, boolean stopOnError) {
|
|
File cleanScript = PropertiesUtil.findConfigFile(file);
|
|
if (cleanScript == null) {
|
|
throw new RuntimeException("Unable to clean the database because I can't find " + file);
|
|
}
|
|
Connection conn = TransactionLegacy.getStandaloneUsageConnection();
|
|
ScriptRunner runner = new ScriptRunner(conn, autoCommit, stopOnError);
|
|
try(FileReader reader = new FileReader(cleanScript);) {
|
|
runner.runScript(reader);
|
|
} catch (IOException e){
|
|
throw new RuntimeException("executeUsageScript:Exception:"+e.getMessage(),e);
|
|
}catch (SQLException e){
|
|
throw new RuntimeException("executeUsageScript:Exception:"+e.getMessage(),e);
|
|
}
|
|
try {
|
|
conn.close();
|
|
} catch (SQLException e) {
|
|
throw new RuntimeException("Unable to close DB connection", e);
|
|
}
|
|
}
|
|
|
|
}
|