Package org.voltdb.sysprocs.saverestore

Source Code of org.voltdb.sysprocs.saverestore.SaveRestoreTestProjectBuilder

/* This file is part of VoltDB.
* Copyright (C) 2008-2010 VoltDB Inc.
*
* Permission is hereby granted, free of charge, to any person obtaining
* a copy of this software and associated documentation files (the
* "Software"), to deal in the Software without restriction, including
* without limitation the rights to use, copy, modify, merge, publish,
* distribute, sublicense, and/or sell copies of the Software, and to
* permit persons to whom the Software is furnished to do so, subject to
* the following conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
* IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR
* OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
* ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
* OTHER DEALINGS IN THE SOFTWARE.
*/

package org.voltdb.sysprocs.saverestore;

import java.io.File;
import java.io.IOException;
import java.net.URL;
import org.voltdb.catalog.Catalog;
import org.voltdb.catalog.Procedure;
import org.voltdb.compiler.VoltProjectBuilder;
import org.voltdb.utils.BuildDirectoryUtils;
import org.voltdb.utils.CatalogUtil;

public class SaveRestoreTestProjectBuilder extends VoltProjectBuilder {
    // REPLICATED
    /*
    public SaveRestoreTestProjectBuilder(String project_name) {
        super(project_name);
    }

    public static Class<?> PROCEDURES[] = new Class<?>[] { SaveRestoreSelect.class };

    public static final File ddlFile = new File("./src/frontend/org/voltdb/sysprocs/saverestore/saverestore-ddl-replicated.sql");
    public static final String jarFilename = "saverestore.jar";

    public void addDefaultProcedures() {
        addProcedures(PROCEDURES);
    }

    public void addDefaultSchema() {
        try {
            System.out.println("URL: " + ddlFile);
            addSchema(ddlFile);
        } catch (Exception e) {
            e.printStackTrace();
            System.exit(-1);
        }
    }

    @Override
    public void addAllDefaults() {
        addDefaultProcedures();
        addDefaultSchema();
    }

    // Different default set for the test of ENG-696
    // TestReplicatedSaveRestoreSysprocSuite
    // Has no partitioned tables.

    public void addAllDefaultsNoPartitioning() {
        addDefaultProcedures();
        addDefaultSchema();
    }

    public String getJARFilename() {
        return jarFilename;
    }

    public Catalog createSaveRestoreSchemaCatalog() throws IOException {
        String testDir = BuildDirectoryUtils.getBuildDirectoryPath();
        String catalogJar = testDir + File.separator + "saverestore-jni.jar";

        addDefaultSchema();
        addDefaultProcedures();

        boolean status = compile(catalogJar);
        assert (status);

        // read in the catalog
        String serializedCatalog = CatalogUtil.loadCatalogFromJar(catalogJar, null);
        assert (serializedCatalog != null);

        // create the catalog (that will be passed to the ClientInterface
        Catalog catalog = new Catalog();
        catalog.execute(serializedCatalog);

        return catalog;
    }
    */
   
    // PARTITIONED
    /*
    public SaveRestoreTestProjectBuilder(String project_name) {
        super(project_name);
    }

    public static Class<?> PROCEDURES[] = new Class<?>[] { SaveRestoreSelect.class };

    public static String partitioning[][] = new String[][] { { "PARTITION_TESTER", "PT_ID" } };

    public static final File ddlFile = new File("./src/frontend/org/voltdb/sysprocs/saverestore/saverestore-ddl-partitioned.sql");
    public static final String jarFilename = "saverestore.jar";

    public void addDefaultProcedures() {
        addProcedures(PROCEDURES);
    }

    public void addDefaultPartitioning() {
        for (String pair[] : partitioning) {
            addPartitionInfo(pair[0], pair[1]);
        }
    }

    public void addDefaultSchema() {
        try {
            System.out.println("URL: " + ddlFile);
            addSchema(ddlFile);
        } catch (Exception e) {
            e.printStackTrace();
            System.exit(-1);
        }
    }

    @Override
    public void addAllDefaults() {
        addDefaultProcedures();
        addDefaultPartitioning();
        addDefaultSchema();
    }

    // Different default set for the test of ENG-696
    // TestReplicatedSaveRestoreSysprocSuite
    // Has no partitioned tables.

    public void addAllDefaultsNoPartitioning() {
        addDefaultProcedures();
        addDefaultSchema();
    }

    public String getJARFilename() {
        return jarFilename;
    }

    public Catalog createSaveRestoreSchemaCatalog() throws IOException {
        String testDir = BuildDirectoryUtils.getBuildDirectoryPath();
        String catalogJar = testDir + File.separator + "saverestore-jni.jar";

        addDefaultSchema();
        addDefaultPartitioning();
        addDefaultProcedures();

        boolean status = compile(catalogJar);
        assert (status);

        // read in the catalog
        String serializedCatalog = CatalogUtil.loadCatalogFromJar(catalogJar, null);
        assert (serializedCatalog != null);

        // create the catalog (that will be passed to the ClientInterface
        Catalog catalog = new Catalog();
        catalog.execute(serializedCatalog);

        return catalog;
    }
    */
   
    // FULL
   
    public SaveRestoreTestProjectBuilder(String project_name) {
        super(project_name);
    }

    public static Class<?> PROCEDURES[] = new Class<?>[] { MatView.class, SaveRestoreSelect.class };

    public static String partitioning[][] = new String[][] { { "PARTITION_TESTER", "PT_ID" }, { "CHANGE_COLUMNS", "ID" }, { "JUMBO_ROW", "PKEY" } };

    public static final File ddlFile = new File("./src/frontend/org/voltdb/sysprocs/saverestore/saverestore-ddl.sql");
    public static final String jarFilename = "saverestore.jar";

    public void addDefaultProcedures() {
        addProcedures(PROCEDURES);
    }

    public void addDefaultPartitioning() {
        for (String pair[] : partitioning) {
            addPartitionInfo(pair[0], pair[1]);
        }
    }

    public void addDefaultSchema() {
        try {
            System.out.println("URL: " + ddlFile);
            addSchema(ddlFile);
        } catch (Exception e) {
            e.printStackTrace();
            System.exit(-1);
        }
    }

    @Override
    public void addAllDefaults() {
        addDefaultProcedures();
        addDefaultPartitioning();
        addDefaultSchema();
        addStmtProcedure("JumboInsert", "INSERT INTO JUMBO_ROW VALUES ( ?, ?, ?)", "JUMBO_ROW.PKEY: 0");
        addStmtProcedure("JumboSelect", "SELECT * FROM JUMBO_ROW WHERE PKEY = ?", "JUMBO_ROW.PKEY: 0");
        addStmtProcedure("JumboCount", "SELECT COUNT(*) FROM JUMBO_ROW");
    }

    // Different default set for the test of ENG-696
    // TestReplicatedSaveRestoreSysprocSuite
    // Has no partitioned tables.

    public void addAllDefaultsNoPartitioning() {
        addDefaultProcedures();
        addDefaultSchema();
    }

    public String getJARFilename() {
        return jarFilename;
    }

    public Catalog createSaveRestoreSchemaCatalog() throws IOException {
        String testDir = BuildDirectoryUtils.getBuildDirectoryPath();
        String catalogJar = testDir + File.separator + "saverestore-jni.jar";

        addDefaultSchema();
        addDefaultPartitioning();
        addDefaultProcedures();

        addStmtProcedure("JumboInsert", "INSERT INTO JUMBO_ROW VALUES ( ?, ?, ?)", "JUMBO_ROW.PKEY: 0");
        addStmtProcedure("JumboSelect", "SELECT * FROM JUMBO_ROW WHERE PKEY = ?", "JUMBO_ROW.PKEY: 0");
        addStmtProcedure("JumboCount", "SELECT COUNT(*) FROM JUMBO_ROW");

        boolean status = compile(catalogJar);
        assert (status);

        // read in the catalog
        String serializedCatalog = CatalogUtil.loadCatalogFromJar(catalogJar, null);
        assert (serializedCatalog != null);

        // create the catalog (that will be passed to the ClientInterface
        Catalog catalog = new Catalog();
        catalog.execute(serializedCatalog);

        return catalog;
    }
}
TOP

Related Classes of org.voltdb.sysprocs.saverestore.SaveRestoreTestProjectBuilder

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.