org.onetwo.common.db.generator.GeneratedContext Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of onetwo-dbm Show documentation
onetwo-dbm
The newest version!
package org.onetwo.common.db.generator;

import java.util.HashMap;
import java.util.Map;

import org.onetwo.common.db.generator.DbGenerator.DbTableGenerator.TableGeneratedConfig;
import org.onetwo.common.db.generator.meta.TableMeta;
import org.onetwo.common.utils.StringUtils;

import com.google.common.collect.Maps;

import lombok.EqualsAndHashCode;

/**
 * @author weishao zeng
 * 

 */

@SuppressWarnings({ "unchecked", "serial" })
@EqualsAndHashCode(callSuper=false)
public class GeneratedContext extends HashMap {
	private static final String TABLE_CONTEXT_KEY = "_tableContext";
//	private static final String TABLE_KEY = "table";
	private TableMeta table;
	private TableGeneratedConfig config;
	
	/****
	 * 主要为了兼容以前的模板
	 * @author weishao zeng
	 */
	public void initBasicContext() {
		String tableNameWithoutPrefix = config.getTableNameWithoutPrefix();
		String className = config.getClassName();
		String propertyName = config.getPropertyName();
		
		Map tableContext = Maps.newHashMap();
		tableContext.put("tableNameWithoutPrefix", tableNameWithoutPrefix);
		tableContext.put("shortTableName", tableNameWithoutPrefix);
		tableContext.put("className", className);
		tableContext.put("propertyName", propertyName);
		String localPackage = StringUtils.emptyIfNull(config.getLocalPackage());
		tableContext.put("localPackage", localPackage);
		tableContext.put("localFullPackage", config.globalGeneratedConfig().getJavaLocalPackage(localPackage));
		
		setTableContext(tableContext);
		put("table", table);
		put("config", config);
	}
	
	public void setTableContext(Map tableContext) {
		put(TABLE_CONTEXT_KEY, tableContext);
	}
	
	public Map getTableContext() {
		return (Map)get(TABLE_CONTEXT_KEY);
	}

	public TableMeta getTable() {
		return table;
	}

	public void setTable(TableMeta table) {
		this.table = table;
	}

	public TableGeneratedConfig getConfig() {
		return config;
	}

	public void setConfig(TableGeneratedConfig config) {
		this.config = config;
	}
}