com.sqlapp.gradle.plugins.GenerateSqlTask.groovy Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sqlapp-gradle-plugin Show documentation
sqlapp gradle plugin.
There is a newer version: 0.12.37
/*
 * Copyright (C) 2007-2017 Tatsuo Satoh 
 *
 * This file is part of sqlapp-gradle-plugin.
 *
 * sqlapp-gradle-plugin is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * sqlapp-gradle-plugin is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with sqlapp-gradle-plugin.  If not, see .
 */
package com.sqlapp.gradle.plugins

import com.sqlapp.data.db.command.GenerateSimpleSqlCommand
import com.sqlapp.data.db.sql.FileSqlExecutor;
import com.sqlapp.data.db.sql.SqlOperation;
import com.sqlapp.data.db.sql.SqlFactoryRegistry;
import com.sqlapp.data.db.sql.SqlType;
import com.sqlapp.data.db.sql.StandardOutSqlExecutor
import com.sqlapp.data.schemas.DbCommonObject;
import com.sqlapp.data.schemas.DefaultSchemaEqualsHandler;
import com.sqlapp.data.schemas.EqualsHandler;
import com.sqlapp.data.schemas.SchemaUtils;
import com.sqlapp.gradle.plugins.pojo.GenerateSqlPojo
import com.sqlapp.util.CommonUtils
import com.sqlapp.util.FileUtils;

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

import org.gradle.api.DefaultTask
import org.gradle.api.Plugin
import org.gradle.api.Project
import org.gradle.api.tasks.Input
import org.gradle.api.tasks.Optional;
import org.gradle.api.tasks.TaskAction;

class GenerateSqlTask extends AbstractGenerateSqlTask {

	@TaskAction
	def exec() {
		GenerateSimpleSqlCommand command=new GenerateSimpleSqlCommand();
		GenerateSqlPojo pojo=project.generateSql;
		this.pojo=pojo;
		command.sqlType=pojo.sqlType;
		command.target=SchemaUtils.readXml(getFile(pojo.targetFile));
		command.setSchemaOption(pojo.schemaOptions);
		File outputPath=getFile(pojo.outputPath);
		String encoding=pojo.encoding;
		run(command);
		if (outputPath==null){
			StandardOutSqlExecutor executor=new StandardOutSqlExecutor();
			executor.execute(command.sqlOperations);
		} else{
			if (this.debug){
				StandardOutSqlExecutor executor=new StandardOutSqlExecutor();
				executor.execute(command.sqlOperations);
			}
			int step=10;
			if (pojo.changeNumberStep!=null){
				step=pojo.changeNumberStep;
			}
			if (pojo.outputAsMultiFiles){
				if (!outputPath.exists()){
					outputPath.mkdirs()
				}
				long current=getCurrentNumber(pojo);
				String suffix=getFileSuffix(pojo);
				for(SqlOperation operation:command.sqlOperations){
					current=current+step;
					File file=new File(""+getFilename(current, pojo.numberOfDigits, toString(operation.getSqlType())+"_"+getName(operation), suffix), outputPath);
					FileSqlExecutor executor=new FileSqlExecutor(file, encoding);
					executor.execute(operation);
				}
			} else{
				FileUtils.createParentDirectory(outputPath);
				SqlOperation operation=command.sqlOperations[0];
				if (outputPath.exists()){
					if (outputPath.isDirectory()){
						long current=getCurrentNumber(pojo);
						current=current+step;
						String suffix=getFileSuffix(pojo);
						File file=new File(""+getFilename(current, pojo.numberOfDigits, toString(operation.getSqlType())+"_"+getName(operation), suffix), outputPath);
						FileSqlExecutor executor=new FileSqlExecutor(file, encoding);
						executor.execute(command.sqlOperations);
						return;
					}
				}
				FileSqlExecutor executor=new FileSqlExecutor(outputPath, encoding);
				executor.execute(command.sqlOperations);
			}
		}
	}

}