io.datarouter.client.mysql.ddl.generate.imp.FieldSqlTableGenerator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of datarouter-mysql Show documentation
datarouter-mysql
There is a newer version: 0.0.125
/*
 * Copyright © 2009 HotPads ([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 io.datarouter.client.mysql.ddl.generate.imp;

import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

import javax.inject.Inject;

import io.datarouter.client.mysql.ddl.domain.CharSequenceSqlColumn;
import io.datarouter.client.mysql.ddl.domain.MysqlCharacterSet;
import io.datarouter.client.mysql.ddl.domain.MysqlCollation;
import io.datarouter.client.mysql.ddl.domain.MysqlRowFormat;
import io.datarouter.client.mysql.ddl.domain.MysqlTableEngine;
import io.datarouter.client.mysql.ddl.domain.SqlColumn;
import io.datarouter.client.mysql.ddl.domain.SqlIndex;
import io.datarouter.client.mysql.ddl.domain.SqlTable;
import io.datarouter.client.mysql.field.MysqlFieldCodec;
import io.datarouter.client.mysql.field.codec.factory.MysqlFieldCodecFactory;
import io.datarouter.model.field.Field;
import io.datarouter.model.field.FieldKey;
import io.datarouter.scanner.Scanner;

public class FieldSqlTableGenerator{

	@Inject
	private MysqlFieldCodecFactory fieldCodecFactory;

	public SqlTable generate(
			String tableName,
			List> primaryKeyFields,
			List> nonKeyFields,
			MysqlCollation collation,
			MysqlCharacterSet characterSet,
			MysqlRowFormat rowFormat,
			Map>> indexes,
			Map>> uniqueIndexes){

		List primaryKeyColumns = makeSqlColumns(primaryKeyFields, false);
		List primaryKeyColumnNames = Scanner.of(primaryKeyColumns).map(SqlColumn::getName).list();
		List columns = makeSqlColumns(nonKeyFields, true);
		columns.addAll(primaryKeyColumns);

		SqlIndex primaryKey = SqlIndex.createPrimaryKey(primaryKeyColumnNames);

		Set sqlIndexes = makeSqlIndexes(indexes);
		Set sqlUniqueIndexes = makeSqlIndexes(uniqueIndexes);

		//TODO set charset and collation elsewhere
		columns.stream()
				.filter(column -> column instanceof CharSequenceSqlColumn)
				.map(CharSequenceSqlColumn.class::cast)
				.forEach(column -> {
					column.setCharacterSet(characterSet);
					column.setCollation(collation);
				});

		return new SqlTable(
				tableName,
				primaryKey,
				columns,
				sqlIndexes,
				sqlUniqueIndexes,
				characterSet,
				collation,
				rowFormat,
				MysqlTableEngine.INNODB);
	}

	private List makeSqlColumns(List> fields, boolean allowNullable){
		return fields.stream()
				.map(field -> getSqlColumnDefinition(allowNullable, field))
				.collect(Collectors.toList());
	}

	private > SqlColumn getSqlColumnDefinition(boolean allowNullable, F field){
		MysqlFieldCodec codec = fieldCodecFactory.createCodec(field);
		return codec.getSqlColumnDefinition(allowNullable, field);
	}

	private List makeSqlColumnNames(List> fields){
		return fields.stream()
				.map(Field::getKey)
				.map(FieldKey::getColumnName)
				.collect(Collectors.toList());
	}

	private Set makeSqlIndexes(Map>> indexes){
		return indexes.entrySet().stream()
				.map(entry -> new SqlIndex(entry.getKey(), makeSqlColumnNames(entry.getValue())))
				.collect(Collectors.toSet());
	}

}