com.tsc9526.monalisa.orm.dao.Insert Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of monalisa-orm Show documentation
Very simple database ORM
There is a newer version: 2.2.0
/*******************************************************************************************
 *	Copyright (c) 2016, zzg.zhou([email protected])
 * 
 *  Monalisa 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.

 *	This program 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 this program.  If not, see .
 *******************************************************************************************/
package com.tsc9526.monalisa.orm.dao;

import com.tsc9526.monalisa.orm.Query;
import com.tsc9526.monalisa.orm.annotation.Column;
import com.tsc9526.monalisa.orm.datasource.DBConfig;
import com.tsc9526.monalisa.orm.dialect.Dialect;
import com.tsc9526.monalisa.orm.executor.KeysExecutor;
import com.tsc9526.monalisa.orm.model.Model;
import com.tsc9526.monalisa.orm.model.ModelHolder;
import com.tsc9526.monalisa.tools.clazz.MelpClass.FGS;
import com.tsc9526.monalisa.tools.string.MelpString;

/**
 * 
 * 完成数据库模型的插入操作
 * 
 * @author zzg.zhou([email protected])
 *
 * @param  数据库模型类型
 */
@SuppressWarnings({"rawtypes"})
public class Insert{
	protected T  model;
	
	protected DBConfig db;
	
	public  Insert(T model){
		this.model=model;		 
	}
	
	public T getModel(){
		return this.model;
	}
			
	public Insert set(String name,Object value){		
		this.model.set(name,value);
		return this;
	}	
	
	public Insert use(DBConfig db){
		this.db=db;
		return this;
	}
	
	public DBConfig db(){
		return this.db==null?model.db():this.db;
	}
	/**
	 * insert到数据库
	 * 
	 * @return 成功变更的记录数
	 */
	public int insert(){			
		return insert(false);
	}
	
	/**
	 * insert到数据库
	 * 
	 * @param updateOnDuplicateKey   true: 如果插入时出现主键冲突则进行更新操作
	 * 
	 * @return 成功变更的记录数
	 */
	public int insert(boolean updateOnDuplicateKey){	 
		Dialect dialect=model.dialect();
		 
		FGS fgs=model.autoField();
		if(fgs!=null && fgs.getObject(model)==null && dialect.supportSequence()){
			Column c = fgs.getAnnotation(Column.class);
			String seq      = c==null? null : c.seq();
			if(!MelpString.isEmpty(seq)){
				String seqNextSql=dialect.getSequenceNext(seq);
				
				Query q=new Query(db());
				q.add(seqNextSql);
				Long seqNextNo = q.getResult(Long.class);
				fgs.setObject(model, seqNextNo);
				
				model.holder().setProperty(ModelHolder.PROP_SEQ_FIELD, fgs.getFieldName());
			}	
		}
		
		Query query= updateOnDuplicateKey ? dialect.insertOrUpdate(model) : dialect.insert(model);
		query.use(db());
		return query.execute(new KeysExecutor(model));  
	}
	
	 
	 
}