com.axiomine.largecollections.generator.GeneratorWritableSet.java Source code

Introduction

Here is the source code for com.axiomine.largecollections.generator.GeneratorWritableSet.java

Source

/*
 * Copyright 2014 Axiomine
 *
 * Licensed 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 com.axiomine.largecollections.generator;

import java.io.File;
import java.util.List;

import org.apache.commons.io.FileUtils;
import org.apache.commons.lang.StringUtils;

public class GeneratorWritableSet {

    /*
     * Sample invocation
     * java GeneratorPrimitivePrimitive com.axiomine.largecollections - com.axiomine.largecollections.functions com.axiomine.largecollections.functions Integer Integer
     */
    public static void main(String[] args) throws Exception {
        //Package of the new class you are generating Ex. com.mypackage
        String MY_PACKAGE = args[0];
        //Any custom imports you need (: seperated). Use - if no custom imports are included
        //Ex. java.util.*:java.lang.Random     
        String CUSTOM_IMPORTS = args[1].equals("-") ? "" : args[1];
        //Package of your Key serializer class. Use com.axiomine.bigcollections.functions
        String T = args[2];

        String tCls = T;

        if (tCls.equals("byte[]")) {
            tCls = "BytesArray";
        }
        String CLASS_NAME = tCls + "Set";
        File root = new File("");
        File outFile = new File(root.getAbsolutePath() + "/src/main/java/" + MY_PACKAGE.replaceAll("\\.", "/") + "/"
                + CLASS_NAME + ".java");

        if (outFile.exists()) {
            System.out.println(outFile.getAbsolutePath() + " already exists. Please delete it and try again");
        }
        {
            String[] imports = null;
            String importStr = "";

            if (!StringUtils.isBlank(CUSTOM_IMPORTS)) {
                CUSTOM_IMPORTS.split(":");
                for (String s : imports) {
                    importStr = "import " + s + ";\n";
                }
            }

            String program = FileUtils.readFileToString(
                    new File(root.getAbsolutePath() + "/src/main/resources/WritableSetTemplate.java"));

            program = program.replaceAll("#MY_PACKAGE#", MY_PACKAGE);
            program = program.replaceAll("#CUSTOM_IMPORTS#", importStr);

            program = program.replaceAll("#CLASS_NAME#", CLASS_NAME);
            program = program.replaceAll("#T#", T);
            program = program.replaceAll("#TCLS#", tCls);

            System.out.println(outFile.getAbsolutePath());
            FileUtils.writeStringToFile(outFile, program);
        }
    }
}