package com.kettle;

import org.pentaho.di.core.KettleEnvironment;

import org.pentaho.di.core.database.DatabaseMeta;

import org.pentaho.di.core.exception.KettleException;

import org.pentaho.di.core.util.EnvUtil;

import org.pentaho.di.repository.RepositoryDirectoryInterface;

import org.pentaho.di.repository.kdr.KettleDatabaseRepository;

import org.pentaho.di.repository.kdr.KettleDatabaseRepositoryMeta;

import org.pentaho.di.trans.Trans;

import org.pentaho.di.trans.TransMeta;

public class ExecTrans {

    private static String repName = "KETTLE_4";

    private static String dbType = "Oracle";

    private static String dbAccess = "Native";

    private static String dbHost = "192.168.200.66";

    private static String dbName = "KYY";

    private static String dbPort = "1521";

    private static String dbUser = "KETTLE_4";

    private static String dbPass = "a1b2c3";

    /**

     * JAVA 调用 Kettle，执行Trans

     *

     * @param args

     * @throws KettleException

     */

    public static void main(String[] args) {

        // TODO Auto-generated method stub

        // 配置转换名称

        String transName = "Test_imp_emp";

        // 配置Excel的文件路径及名称

        String fileName = "E:\\template\\export\\成绩导入模板1203.xls";

        // 配置错误Excel的文件路径及名称

        String errorFileName = "E:\\template\\export\\error_成绩导入";

        // 配置数据库连接

        execTransForDB(transName, fileName, errorFileName);

    }

    /**

     * 配置数据源 调用资源库中的相关job 、transfer

     */

    public static void execTransForDB(String transName, String fileName, String errorFileName) {

        try {

            KettleEnvironment.init();

            // 创建DB资源库

            KettleDatabaseRepository repository = new KettleDatabaseRepository();

            DatabaseMeta databaseMeta = new DatabaseMeta(repName, dbType,

                    dbAccess, dbHost, dbName, dbPort, dbUser, dbPass);

            // 选择资源库

            KettleDatabaseRepositoryMeta kettleDatabaseRepositoryMeta = new KettleDatabaseRepositoryMeta(

                    "Kettle", "Kettle", "Transformation description",

                    databaseMeta);

            repository.init(kettleDatabaseRepositoryMeta);

            // 连接资源库

            repository.connect("admin", "admin");

            RepositoryDirectoryInterface directoryInterface = repository

                    .loadRepositoryDirectoryTree();

            // 选择转换

            TransMeta transMeta = repository.loadTransformation(transName,

                    directoryInterface, null, true, null);

            Trans trans = new Trans(transMeta);

            trans.setVariable("fileName", fileName);

            trans.setVariable("errorFileName", errorFileName);

            trans.execute(null);

            trans.waitUntilFinished();// 等待直到数据结束

            if (trans.getErrors() > 0) {

                System.out.println("transformation error");

            } else {

                System.out.println("transformation successfully");

            }

        } catch (KettleException e) {

            e.printStackTrace();

        }

    }

}