com.abubusoft.kripton.processor.sqlite.BindTableGenerator Maven / Gradle / Ivy
/*******************************************************************************
* Copyright 2015, 2017 Francesco Benincasa ([email protected]).
*
* 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.abubusoft.kripton.processor.sqlite;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import javax.annotation.processing.Filer;
import javax.lang.model.element.AnnotationMirror;
import javax.lang.model.element.AnnotationValue;
import javax.lang.model.element.ExecutableElement;
import javax.lang.model.element.Modifier;
import javax.lang.model.element.PackageElement;
import javax.lang.model.util.Elements;
import org.apache.commons.lang3.StringUtils;
import com.abubusoft.kripton.android.ColumnType;
import com.abubusoft.kripton.android.annotation.BindDataSource;
import com.abubusoft.kripton.android.annotation.BindSqlType;
import com.abubusoft.kripton.android.sqlite.ForeignKeyAction;
import com.abubusoft.kripton.android.sqlite.SQLiteTable;
import com.abubusoft.kripton.common.CaseFormat;
import com.abubusoft.kripton.common.Converter;
import com.abubusoft.kripton.common.One;
import com.abubusoft.kripton.common.Pair;
import com.abubusoft.kripton.common.Triple;
import com.abubusoft.kripton.processor.BindDataSourceSubProcessor;
import com.abubusoft.kripton.processor.bind.BindTypeContext;
import com.abubusoft.kripton.processor.bind.JavaWriterHelper;
import com.abubusoft.kripton.processor.core.AssertKripton;
import com.abubusoft.kripton.processor.core.Finder;
import com.abubusoft.kripton.processor.core.ManagedPropertyPersistenceHelper;
import com.abubusoft.kripton.processor.core.ManagedPropertyPersistenceHelper.PersistType;
import com.abubusoft.kripton.processor.core.ModelElementVisitor;
import com.abubusoft.kripton.processor.core.ModelProperty;
import com.abubusoft.kripton.processor.core.reflect.TypeUtility;
import com.abubusoft.kripton.processor.element.GeneratedTypeElement;
import com.abubusoft.kripton.processor.exceptions.InvalidBeanTypeException;
import com.abubusoft.kripton.processor.exceptions.InvalidForeignKeyTypeException;
import com.abubusoft.kripton.processor.exceptions.NoDaoElementFound;
import com.abubusoft.kripton.processor.sqlite.core.JavadocUtility;
import com.abubusoft.kripton.processor.sqlite.grammars.jql.JQLChecker;
import com.abubusoft.kripton.processor.sqlite.grammars.jql.JQLContext;
import com.abubusoft.kripton.processor.sqlite.grammars.jql.JQLReplacerListenerImpl;
import com.abubusoft.kripton.processor.sqlite.model.SQLProperty;
import com.abubusoft.kripton.processor.sqlite.model.SQLiteDaoDefinition;
import com.abubusoft.kripton.processor.sqlite.model.SQLiteDatabaseSchema;
import com.abubusoft.kripton.processor.sqlite.model.SQLiteEntity;
import com.abubusoft.kripton.processor.sqlite.transform.SQLTransformer;
import com.abubusoft.kripton.processor.utils.AnnotationProcessorUtilis;
import com.squareup.javapoet.ArrayTypeName;
import com.squareup.javapoet.ClassName;
import com.squareup.javapoet.FieldSpec;
import com.squareup.javapoet.FieldSpec.Builder;
import com.squareup.javapoet.MethodSpec;
import com.squareup.javapoet.TypeName;
import com.squareup.javapoet.TypeSpec;
/**
*
* Generate class :{entity}Table which represents table for entity.
*
*
* @author Francesco Benincasa ([email protected])
*
*/
public class BindTableGenerator extends AbstractBuilder implements ModelElementVisitor {
/** The Constant SUFFIX. */
public static final String SUFFIX = "Table";
/** The column name to upper case converter. */
private Converter columnNameToUpperCaseConverter = CaseFormat.LOWER_CAMEL
.converterTo(CaseFormat.UPPER_UNDERSCORE);
/**
* Instantiates a new bind table generator.
*
* @param elementUtils
* the element utils
* @param filer
* the filer
* @param model
* the model
*/
public BindTableGenerator(Elements elementUtils, Filer filer, SQLiteDatabaseSchema model) {
super(elementUtils, filer, model);
}
/**
* Generate table for entities.
*
* @param elementUtils
* the element utils
* @param filer
* the filer
* @param schema
* the schema
* @param generatedEntities
* the generated entities
* @throws Exception
* the exception
*/
public static void generate(Elements elementUtils, Filer filer, SQLiteDatabaseSchema schema,
Set generatedEntities) throws Exception {
BindTableGenerator visitor = new BindTableGenerator(elementUtils, filer, schema);
List orderedEntities = BindDataSourceBuilder.orderEntitiesList(schema);
// check table names
Map tableNames=new HashMap();
for (SQLiteEntity item: orderedEntities) {
if (tableNames.containsKey(item.getTableName().toLowerCase())) {
AssertKripton.fail("Table name '%s' is mapped to entities '%s' and '%s'" , item.getTableName(), item.getName(), tableNames.get(item.getTableName()));
} else {
tableNames.put(item.getTableName().toLowerCase(), item.getName());
}
}
for (GeneratedTypeElement item: generatedEntities) {
if (tableNames.containsKey(item.getTableName().toLowerCase())) {
AssertKripton.fail("Table name '%s' is mapped to entities '%s' and '%s'" , item.getTableName(), item.getName(), tableNames.get(item.getTableName()));
} else {
tableNames.put(item.getTableName().toLowerCase(), item.getName());
}
}
for (SQLiteEntity item : orderedEntities) {
visitor.visit(schema, item);
}
// generate table for generated entity
for (GeneratedTypeElement genItem : generatedEntities) {
visitor.visit(schema, genItem);
}
}
/**
* Visit.
*
* @param schema
* the schema
* @param entity
* the entity
* @throws Exception
* the exception
*/
private void visit(SQLiteDatabaseSchema schema, GeneratedTypeElement entity) throws Exception {
int indexCounter = 0;
// generate the class name that represents the table
String classTableName = getTableClassName(entity.getSimpleName());
PackageElement pkg = elementUtils.getPackageElement(entity.packageName);
String packageName = pkg.isUnnamed() ? null : pkg.getQualifiedName().toString();
AnnotationProcessorUtilis.infoOnGeneratedClasses(BindDataSource.class, packageName, classTableName);
classBuilder = TypeSpec.classBuilder(classTableName).addModifiers(Modifier.PUBLIC)
.addSuperinterface(SQLiteTable.class);
BindTypeContext context = new BindTypeContext(classBuilder, TypeUtility.typeName(packageName, classTableName),
Modifier.STATIC, Modifier.PRIVATE);
// javadoc for class
classBuilder.addJavadoc("");
classBuilder.addJavadoc("\nEntity $L is associated to table $L\n",
entity.getSimpleName(), entity.getTableName());
classBuilder.addJavadoc("This class represents table associated to entity.\n");
classBuilder.addJavadoc("
\n");
JavadocUtility.generateJavadocGeneratedBy(classBuilder);
classBuilder.addJavadoc(" @see $T\n", TypeUtility.className(entity.getName()));
{
// @formatter:off
// table_name
FieldSpec fieldSpec = FieldSpec
.builder(String.class, "TABLE_NAME", Modifier.PUBLIC, Modifier.STATIC, Modifier.FINAL)
.initializer("\"$L\"", entity.getTableName())
.addJavadoc("Costant represents typeName of table $L\n", entity.getTableName()).build();
classBuilder.addField(fieldSpec);
// @formatter:on
}
StringBuilder bufferTable = new StringBuilder();
StringBuilder bufferForeignKey = new StringBuilder();
// shared between create table and drop table
StringBuilder bufferIndexesCreate = new StringBuilder();
StringBuilder bufferDropTable = new StringBuilder();
StringBuilder bufferIndexesDrop = new StringBuilder();
bufferTable.append("CREATE TABLE " + entity.getTableName());
// define column typeName set
String separator = "";
bufferTable.append(" (");
List uniqueConstraintColumn=new ArrayList<>();
// for each column, that need to be persisted on table
for (SQLProperty item : entity.getCollection()) {
bufferTable.append(separator);
bufferTable.append(item.columnName);
// every column is a long
if (item.getPropertyType() == null) {
bufferTable.append(" " + SQLTransformer.lookup(TypeName.LONG).getColumnTypeAsString());
} else {
bufferTable.append(
" " + SQLTransformer.lookup(item.getPropertyType().getTypeName()).getColumnTypeAsString());
}
switch (item.columnType) {
case PRIMARY_KEY:
bufferTable.append(" PRIMARY KEY AUTOINCREMENT NOT NULL");
break;
case PRIMARY_KEY_UNMANGED:
bufferTable.append(" PRIMARY KEY NOT NULL");
break;
case UNIQUE:
bufferTable.append(" UNIQUE");
break;
case INDEXED:
bufferIndexesCreate.append(String.format(" CREATE INDEX idx_%s_%s ON %s(%s);", entity.getTableName(),
item.columnName, entity.getTableName(), item.columnName));
bufferIndexesDrop.append(
String.format(" DROP INDEX IF EXISTS idx_%s_%s;", entity.getTableName(), item.columnName));
uniqueConstraintColumn.add(item.columnName);
break;
case STANDARD:
break;
}
boolean nullable = item.isNullable();
// if it is not primary key and it is not nullable, then add not
// null
if (!nullable && !(item.columnType == ColumnType.PRIMARY_KEY
|| item.columnType == ColumnType.PRIMARY_KEY_UNMANGED)) {
bufferTable.append(" NOT NULL");
}
// foreign key
String foreignClassName = item.foreignParentClassName;
if (item.isForeignKey()) {
SQLiteEntity reference = model.getEntity(foreignClassName);
if (reference == null) {
// check if we have a DAO associated into DataSource
// definition
boolean found = false;
for (SQLiteDaoDefinition daoDefinition : schema.getCollection()) {
if (daoDefinition.getEntityClassName().equals(foreignClassName)) {
found = true;
}
}
if (!found) {
throw new NoDaoElementFound(schema, TypeUtility.className(foreignClassName));
} else {
throw new InvalidBeanTypeException(item, foreignClassName);
}
}
bufferForeignKey.append(", FOREIGN KEY(" + item.columnName + ") REFERENCES " + reference.getTableName()
+ "(" + reference.getPrimaryKey().columnName + ")");
if (item.onDeleteAction != ForeignKeyAction.NO_ACTION) {
bufferForeignKey.append(" ON DELETE " + item.onDeleteAction.toString().replaceAll("_", " "));
}
if (item.onUpdateAction != ForeignKeyAction.NO_ACTION) {
bufferForeignKey.append(" ON UPDATE " + item.onUpdateAction.toString().replaceAll("_", " "));
}
// INSERT as dependency only if reference is another entity.
// Same entity can not be own dependency.
if (!entity.getClassName().equals(TypeUtility.typeName(reference.getElement()))) {
entity.referedEntities.add(reference);
}
}
separator = ", ";
}
// add foreign key
bufferTable.append(bufferForeignKey.toString());
// add contrants unique
bufferTable.append(String.format(", UNIQUE (%s)", StringUtils.join(uniqueConstraintColumn, ", ")));
bufferTable.append(");");
// add indexes creation one table
if (bufferIndexesCreate.length() > 0) {
bufferTable.append(bufferIndexesCreate.toString());
}
// add single column indexes (NOT UNIQUE)
{
Triple multiIndexes = buildIndexes(entity, false, indexCounter);
if (!StringUtils.isEmpty(multiIndexes.value0)) {
bufferTable.append(multiIndexes.value0 + ";");
bufferIndexesDrop.append(multiIndexes.value1 + ";");
}
}
{
// create table SQL
// @formatter:off
FieldSpec.Builder fieldSpec = FieldSpec.builder(String.class, "CREATE_TABLE_SQL")
.addModifiers(Modifier.STATIC, Modifier.FINAL, Modifier.PUBLIC);
// @formatter:on
// @formatter:off
fieldSpec.addJavadoc("\nDDL to create table $L\n
\n", entity.getTableName());
fieldSpec.addJavadoc("\n$L
\n", bufferTable.toString());
// @formatter:on
classBuilder.addField(fieldSpec.initializer("$S", bufferTable.toString()).build());
}
// drop table SQL
// index does not need to be dropped, they are automatically detroyed
// with tables
if (bufferIndexesDrop.length() > 0) {
bufferDropTable.append(bufferIndexesDrop.toString());
}
bufferDropTable.append("DROP TABLE IF EXISTS " + entity.getTableName() + ";");
{
// @formatter:off
FieldSpec fieldSpec = FieldSpec.builder(String.class, "DROP_TABLE_SQL")
.addModifiers(Modifier.STATIC, Modifier.FINAL, Modifier.PUBLIC)
.initializer("$S", bufferDropTable.toString())
.addJavadoc("\nDDL to drop table $L\n
\n", entity.getTableName())
.addJavadoc("\n$L
\n", bufferDropTable.toString()).build();
// @formatter:on
classBuilder.addField(fieldSpec);
}
// define column typeName set
for (ModelProperty item : entity.getCollection()) {
item.accept(this);
}
ManagedPropertyPersistenceHelper.generateFieldPersistance(context, entity.getCollection(), PersistType.BYTE,
true, Modifier.STATIC, Modifier.PUBLIC);
model.sqlForCreate.add(bufferTable.toString());
model.sqlForDrop.add(bufferDropTable.toString());
generateColumnsArray(entity);
TypeSpec typeSpec = classBuilder.build();
JavaWriterHelper.writeJava2File(filer, packageName, typeSpec);
}
/**
* Gets the table class name.
*
* @param entityName
* the entity name
* @return the table class name
*/
public static String getTableClassName(String entityName) {
return entityName + SUFFIX;
}
/**
* Table class name.
*
* @param dao
* the dao
* @param entity
* the entity
* @return the class name
*/
public static ClassName tableClassName(SQLiteDaoDefinition dao, SQLiteEntity entity) {
String entityName = BindDataSourceSubProcessor.generateEntityQualifiedName(dao, entity);
return TypeUtility.className(entityName + SUFFIX);
}
/*
* (non-Javadoc)
*
* @see com.abubusoft.kripton.processor.core.ModelElementVisitor#visit(com. abubusoft.kripton.processor.sqlite.model.SQLiteDatabaseSchema,
* com.abubusoft.kripton.processor.core.ModelClass)
*/
@Override
public void visit(SQLiteDatabaseSchema schema, SQLiteEntity entity) throws Exception {
int indexCounter = 0;
// generate the class name that represents the table
String classTableName = getTableClassName(entity.getSimpleName());
FindIndexesVisitor indexVisitor = new FindIndexesVisitor();
List annotationMirrors = entity.getElement().getAnnotationMirrors();
for (AnnotationMirror annotationMirror : annotationMirrors) {
Map elementValues = annotationMirror
.getElementValues();
if (BindSqlType.class.getName().equals(annotationMirror.getAnnotationType().toString())) {
for (Map.Entry entry : elementValues
.entrySet()) {
// The 'entry.getKey()' here is the annotation attribute
// name.
String key = entry.getKey().getSimpleName().toString();
entry.getValue().accept(indexVisitor, key);
}
}
}
PackageElement pkg = elementUtils.getPackageOf(entity.getElement());
String packageName = pkg.isUnnamed() ? null : pkg.getQualifiedName().toString();
AnnotationProcessorUtilis.infoOnGeneratedClasses(BindDataSource.class, packageName, classTableName);
classBuilder = TypeSpec.classBuilder(classTableName).addModifiers(Modifier.PUBLIC)
.addSuperinterface(SQLiteTable.class);
BindTypeContext context = new BindTypeContext(classBuilder, TypeUtility.typeName(packageName, classTableName),
Modifier.STATIC, Modifier.PRIVATE);
// javadoc for class
classBuilder.addJavadoc("");
classBuilder.addJavadoc("\nEntity $L is associated to table $L\n",
entity.getSimpleName(), entity.getTableName());
classBuilder.addJavadoc("This class represents table associated to entity.\n");
classBuilder.addJavadoc("
\n");
JavadocUtility.generateJavadocGeneratedBy(classBuilder);
classBuilder.addJavadoc(" @see $T\n", TypeUtility.className(entity.getName()));
{
// @formatter:off
// table_name
FieldSpec fieldSpec = FieldSpec
.builder(String.class, "TABLE_NAME", Modifier.PUBLIC, Modifier.STATIC, Modifier.FINAL)
.initializer("\"$L\"", entity.getTableName())
.addJavadoc("Costant represents typeName of table $L\n", entity.getTableName()).build();
classBuilder.addField(fieldSpec);
// @formatter:on
}
StringBuilder bufferTable = new StringBuilder();
StringBuilder bufferForeignKey = new StringBuilder();
// shared between create table and drop table
StringBuilder bufferIndexesCreate = new StringBuilder();
StringBuilder bufferDropTable = new StringBuilder();
StringBuilder bufferIndexesDrop = new StringBuilder();
bufferTable.append("CREATE TABLE " + entity.getTableName());
// define column typeName set
String separator = "";
bufferTable.append(" (");
// for each column, that need to be persisted on table
for (SQLProperty item : entity.getCollection()) {
bufferTable.append(separator);
bufferTable.append(item.columnName);
bufferTable.append(" " + SQLTransformer.columnTypeAsString(item));
switch (item.columnType) {
case PRIMARY_KEY:
case PRIMARY_KEY_UNMANGED:
bufferTable.append(" PRIMARY KEY");
if (!item.isType(String.class) && item.columnType == ColumnType.PRIMARY_KEY) {
bufferTable.append(" AUTOINCREMENT");
}
bufferTable.append(" NOT NULL");
break;
case UNIQUE:
bufferTable.append(" UNIQUE");
break;
case INDEXED:
bufferIndexesCreate.append(String.format(" CREATE INDEX idx_%s_%s ON %s(%s);", entity.getTableName(),
item.columnName, entity.getTableName(), item.columnName));
bufferIndexesDrop.append(
String.format(" DROP INDEX IF EXISTS idx_%s_%s;", entity.getTableName(), item.columnName));
break;
case STANDARD:
break;
}
boolean nullable = item.isNullable();
// if it is not primary key and it is not nullable, then add not
// null
if (!nullable && !(item.columnType == ColumnType.PRIMARY_KEY
|| item.columnType == ColumnType.PRIMARY_KEY_UNMANGED)) {
bufferTable.append(" NOT NULL");
}
// foreign key
String foreignClassName = item.foreignParentClassName;
if (item.isForeignKey()) {
SQLiteEntity reference = model.getEntity(foreignClassName);
if (reference == null) {
// check if we have a DAO associated into DataSource
// definition
boolean found = false;
for (SQLiteDaoDefinition daoDefinition : schema.getCollection()) {
if (daoDefinition.getEntityClassName().equals(foreignClassName)) {
found = true;
}
}
if (!found) {
throw new NoDaoElementFound(schema, TypeUtility.className(foreignClassName));
} else {
throw new InvalidBeanTypeException(item, foreignClassName);
}
}
// foreign key can ben used only with column type
// long/Long
if (!TypeUtility.isTypeIncludedIn(item.getPropertyType().getTypeName(), Long.class, Long.TYPE,
String.class)) {
throw new InvalidForeignKeyTypeException(item);
}
bufferForeignKey.append(", FOREIGN KEY(" + item.columnName + ") REFERENCES " + reference.getTableName()
+ "(" + reference.getPrimaryKey().columnName + ")");
if (item.onDeleteAction != ForeignKeyAction.NO_ACTION) {
bufferForeignKey.append(" ON DELETE " + item.onDeleteAction.toString().replaceAll("_", " "));
}
if (item.onUpdateAction != ForeignKeyAction.NO_ACTION) {
bufferForeignKey.append(" ON UPDATE " + item.onUpdateAction.toString().replaceAll("_", " "));
}
// INSERT as dependency only if reference is another entity.
// Same entity can not be own dependency.
if (!entity.equals(reference)) {
entity.referedEntities.add(reference);
}
}
separator = ", ";
}
// add foreign key
bufferTable.append(bufferForeignKey.toString());
// add multicolumn indexes (UNIQUE)
{
Triple multiIndexes = buldIndexes(entity, indexVisitor.getUniqueIndexes(), true,
indexCounter);
// add constraint unique (if present)
bufferTable.append(multiIndexes.value2);
// close the table definition
bufferTable.append(");");
if (!StringUtils.isEmpty(multiIndexes.value0)) {
bufferTable.append(multiIndexes.value0 + ";");
bufferIndexesDrop.append(multiIndexes.value1 + ";");
}
}
// add indexes creation one table
if (bufferIndexesCreate.length() > 0) {
bufferTable.append(bufferIndexesCreate.toString());
}
// add multicolumn indexes (NOT UNIQUE)
{
Pair multiIndexes = buldIndexes(entity, indexVisitor.getNotUniqueIndexes(), false,
indexCounter);
if (!StringUtils.isEmpty(multiIndexes.value0)) {
bufferTable.append(multiIndexes.value0 + ";");
bufferIndexesDrop.append(multiIndexes.value1 + ";");
}
}
{
// create table SQL
// @formatter:off
FieldSpec.Builder fieldSpec = FieldSpec.builder(String.class, "CREATE_TABLE_SQL")
.addModifiers(Modifier.STATIC, Modifier.FINAL, Modifier.PUBLIC);
// @formatter:on
// @formatter:off
fieldSpec.addJavadoc("\nDDL to create table $L\n
\n", entity.getTableName());
fieldSpec.addJavadoc("\n$L
\n", bufferTable.toString());
// @formatter:on
classBuilder.addField(fieldSpec.initializer("$S", bufferTable.toString()).build());
}
// drop table SQL
// index does not need to be dropped, they are automatically detroyed
// with tables
if (bufferIndexesDrop.length() > 0) {
bufferDropTable.append(bufferIndexesDrop.toString());
}
bufferDropTable.append("DROP TABLE IF EXISTS " + entity.getTableName() + ";");
{
// @formatter:off
FieldSpec fieldSpec = FieldSpec.builder(String.class, "DROP_TABLE_SQL")
.addModifiers(Modifier.STATIC, Modifier.FINAL, Modifier.PUBLIC)
.initializer("$S", bufferDropTable.toString())
.addJavadoc("\nDDL to drop table $L\n
\n", entity.getTableName())
.addJavadoc("\n$L
\n", bufferDropTable.toString()).build();
// @formatter:on
classBuilder.addField(fieldSpec);
}
// define column typeName set
for (ModelProperty item : entity.getCollection()) {
item.accept(this);
}
ManagedPropertyPersistenceHelper.generateFieldPersistance(context, entity.getCollection(), PersistType.BYTE,
true, Modifier.STATIC, Modifier.PUBLIC);
model.sqlForCreate.add(bufferTable.toString());
model.sqlForDrop.add(bufferDropTable.toString());
generateColumnsArray(entity);
TypeSpec typeSpec = classBuilder.build();
JavaWriterHelper.writeJava2File(filer, packageName, typeSpec);
}
/**
* generate columns array.
*
* @param entity
* the entity
*/
private void generateColumnsArray(Finder entity) {
// generate columns array
Builder sp = FieldSpec.builder(ArrayTypeName.of(String.class), "COLUMNS", Modifier.STATIC, Modifier.PRIVATE,
Modifier.FINAL);
String s = "";
StringBuilder buffer = new StringBuilder();
for (SQLProperty property : entity.getCollection()) {
buffer.append(s + "COLUMN_" + columnNameToUpperCaseConverter.convert(property.getName()));
s = ", ";
}
classBuilder.addField(sp.addJavadoc("Columns array\n").initializer("{" + buffer.toString() + "}").build());
classBuilder.addMethod(MethodSpec.methodBuilder("columns").addModifiers(Modifier.PUBLIC)
.addJavadoc("Columns array\n").addAnnotation(Override.class).returns(ArrayTypeName.of(String.class))
.addStatement("return COLUMNS").build());
classBuilder.addMethod(MethodSpec.methodBuilder("name").addModifiers(Modifier.PUBLIC).addJavadoc("table name\n")
.addAnnotation(Override.class).returns(TypeName.get(String.class)).addStatement("return TABLE_NAME")
.build());
}
/**
* Buld indexes.
*
* @param entity
* the entity
* @param List
* @param unique
* the unique
* @param counter
* the counter
* @return the pair
*/
public static Triple buldIndexes(final SQLiteEntity entity,
ArrayList, Boolean>> indexList, boolean unique, int counter) {
Triple result = new Triple<>();
result.value0 = "";
result.value1 = "";
result.value2 = "";
if (indexList.size() == 0)
return result;
String uniqueString;
if (unique) {
uniqueString = "UNIQUE ";
} else {
uniqueString = "";
}
List listCreateIndex = new ArrayList<>();
List listDropIndex = new ArrayList<>();
List listUniqueConstraint = new ArrayList<>();
for (Pair, Boolean> index : indexList) {
final List listUniqueFields = new ArrayList<>();
String createIndex = String.format(" CREATE %sINDEX idx_%s_%s on %s (%s)", uniqueString,
entity.getTableName(), counter++, entity.getTableName(), StringUtils.join(index.value0, ", "));
String dropIndex = String.format(" DROP INDEX IF EXISTS idx_%s_%s", entity.getTableName(), counter);
final One fieldCounter = new One(0);
createIndex = JQLChecker.getInstance().replace(new JQLContext() {
@Override
public String getContextDescription() {
return "While table definition generation for entity " + entity.getName();
}
@Override
public String getName() {
// TODO Auto-generated method stub
return null;
}
@Override
public String getParentName() {
// TODO Auto-generated method stub
return null;
}
@Override
public Finder findEntityByName(String entityName) {
// TODO Auto-generated method stub
return null;
}
}, createIndex, new JQLReplacerListenerImpl(null) {
@Override
public String onColumnName(String columnName) {
fieldCounter.value0++;
SQLProperty property = entity.findPropertyByName(columnName);
AssertKripton.assertTrue(property != null, "class '%s' in @%s(indexes) use unknown property '%s'",
entity.getName(), BindSqlType.class.getSimpleName(), columnName);
listUniqueFields.add(property.columnName);
return property.columnName;
}
@Override
public String onColumnFullyQualifiedName(String tableName, String columnName) {
AssertKripton.fail("Inconsistent state");
return null;
}
});
AssertKripton.assertTrue(fieldCounter.value0 > 0,
"class '%s' have @%s(indexes) with no well formed indexes", entity.getName(),
BindSqlType.class.getSimpleName());
if (unique) {
// add unique constraint
listUniqueConstraint.add(String.format(", UNIQUE (%s)", StringUtils.join(listUniqueFields, ", ")));
}
listCreateIndex.add(createIndex);
listDropIndex.add(dropIndex);
}
result.value0 = StringUtils.join(listCreateIndex, ";");
result.value1 = StringUtils.join(listDropIndex, ";");
result.value2 = StringUtils.join(listUniqueConstraint, "");
return result;
}
/**
* A generated element can have only a primary key and two FK.
*
* @param entity
* the entity
* @param unique
* the unique
* @param counter
* the counter
* @return the pair
*/
public static Triple buildIndexes(final GeneratedTypeElement entity, boolean unique, int counter) {
Triple result = new Triple<>();
result.value0 = "";
result.value1 = "";
result.value2 = "";
List indexes = entity.index;
String uniqueString;
uniqueString = "UNIQUE ";
if (indexes == null || indexes.size() == 0)
return result;
List listCreateIndex = new ArrayList<>();
List listDropIndex = new ArrayList<>();
for (String index : indexes) {
String createIndex = String.format(" CREATE %sINDEX idx_%s_%s on %s (%s)", uniqueString,
entity.getTableName(), counter++, entity.getTableName(), index);
String dropIndex = String.format(" DROP INDEX IF EXISTS idx_%s_%s", entity.getTableName(), counter);
listCreateIndex.add(createIndex);
listDropIndex.add(dropIndex);
}
result.value0 = StringUtils.join(listCreateIndex, ";");
result.value1 = StringUtils.join(listDropIndex, ";");
return result;
}
/*
* (non-Javadoc)
*
* @see com.abubusoft.kripton.processor.core.ModelElementVisitor#visit(com. abubusoft.kripton.processor.core.ModelProperty)
*/
@Override
public void visit(SQLProperty kriptonProperty) {
// @formatter:off
FieldSpec fieldSpec = FieldSpec
.builder(String.class, "COLUMN_" + columnNameToUpperCaseConverter.convert(kriptonProperty.getName()),
Modifier.PUBLIC, Modifier.STATIC, Modifier.FINAL)
.initializer("$S", kriptonProperty.columnName)
.addJavadoc(
"Entity's property $L is associated to table column $L. This costant represents column name.\n",
kriptonProperty.getName(), kriptonProperty.columnName)
.addJavadoc("\n @see $T#$L\n", kriptonProperty.getEntityTypeName(), kriptonProperty.getName()).build();
// @formatter:on
classBuilder.addField(fieldSpec);
}
}