package org.nutz.dao.impl;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.List;
import java.util.concurrent.atomic.AtomicLong;
import javax.sql.DataSource;
import org.nutz.dao.Chain;
import org.nutz.dao.Condition;
import org.nutz.dao.ConnCallback;
import org.nutz.dao.Dao;
import org.nutz.dao.DaoException;
import org.nutz.dao.FieldFilter;
import org.nutz.dao.FieldMatcher;
import org.nutz.dao.SqlManager;
import org.nutz.dao.Sqls;
import org.nutz.dao.entity.Entity;
import org.nutz.dao.entity.LinkField;
import org.nutz.dao.entity.LinkVisitor;
import org.nutz.dao.entity.Record;
import org.nutz.dao.impl.link.DoClearLinkVisitor;
import org.nutz.dao.impl.link.DoClearRelationByHostFieldLinkVisitor;
import org.nutz.dao.impl.link.DoClearRelationByLinkedFieldLinkVisitor;
import org.nutz.dao.impl.link.DoDeleteLinkVisitor;
import org.nutz.dao.impl.link.DoFetchLinkVisitor;
import org.nutz.dao.impl.link.DoInsertLinkVisitor;
import org.nutz.dao.impl.link.DoInsertRelationLinkVisitor;
import org.nutz.dao.impl.link.DoUpdateLinkVisitor;
import org.nutz.dao.impl.link.DoUpdateRelationLinkVisitor;
import org.nutz.dao.impl.sql.pojo.ConditionPItem;
import org.nutz.dao.impl.sql.pojo.PojoEachEntityCallback;
import org.nutz.dao.impl.sql.pojo.PojoEachRecordCallback;
import org.nutz.dao.impl.sql.pojo.PojoFetchEntityCallback;
import org.nutz.dao.impl.sql.pojo.PojoFetchIntCallback;
import org.nutz.dao.impl.sql.pojo.PojoFetchRecordCallback;
import org.nutz.dao.impl.sql.pojo.PojoQueryEntityCallback;
import org.nutz.dao.impl.sql.pojo.PojoQueryRecordCallback;
import org.nutz.dao.pager.Pager;
import org.nutz.dao.sql.Criteria;
import org.nutz.dao.sql.DaoStatement;
import org.nutz.dao.sql.PItem;
import org.nutz.dao.sql.Pojo;
import org.nutz.dao.sql.PojoCallback;
import org.nutz.dao.sql.Sql;
import org.nutz.dao.util.Daos;
import org.nutz.dao.util.Pojos;
import org.nutz.dao.util.cri.SqlExpressionGroup;
import org.nutz.lang.ContinueLoop;
import org.nutz.lang.Each;
import org.nutz.lang.ExitLoop;
import org.nutz.lang.Lang;
import org.nutz.lang.LoopException;
import org.nutz.lang.Strings;
import org.nutz.log.Log;
import org.nutz.log.Logs;
import org.nutz.trans.Molecule;
public class NutDao extends DaoSupport implements Dao {
private static final Log log = Logs.get();
private static final AtomicLong atomLong = new AtomicLong();
private PojoCallback _pojo_queryEntity;
private PojoCallback _pojo_fetchEntity;
private PojoCallback _pojo_eachEntity;
private PojoCallback _pojo_queryRecord;
private PojoCallback _pojo_fetchRecord;
private PojoCallback _pojo_eachRecord;
private PojoCallback _pojo_fetchInt;
protected volatile long _selfId;
// ==========================================================
// 下面是 3 个构造函数
public NutDao() {
super();
_selfId = atomLong.getAndIncrement();
// 设置默认的回调
_pojo_queryEntity = new PojoQueryEntityCallback();
_pojo_fetchEntity = new PojoFetchEntityCallback();
_pojo_eachEntity = new PojoEachEntityCallback();
_pojo_fetchInt = new PojoFetchIntCallback();
_pojo_queryRecord = new PojoQueryRecordCallback();
_pojo_fetchRecord = new PojoFetchRecordCallback();
_pojo_eachRecord = new PojoEachRecordCallback();
// check for spring DataSourceTransactionManager
for (StackTraceElement ele : Thread.currentThread().getStackTrace()) {
if (ele.getClassName().startsWith("org.springframework")) {
log.info("Note: Make sure set SpringDaoRunner if using DataSourceTransactionManager.");
break;
}
}
}
public NutDao(DataSource dataSource) {
this();
this.setDataSource(dataSource);
}
public NutDao(DataSource dataSource, SqlManager sqlManager) {
this(dataSource);
this.setSqlManager(sqlManager);
}
// 上面是 3 个构造函数
// ==========================================================
public <T> T getObject(Class<T> classOfT, ResultSet rs, FieldMatcher fm) {
return holder.getEntity(classOfT).getObject(rs, fm);
}
public <T> T insert(final T obj) {
final EntityOperator opt = _optBy(Lang.first(obj));
if (null == opt)
return null;
Lang.each(obj, false, new Each<Object>() {
public void invoke(int i, Object ele, int length) throws ExitLoop,
LoopException {
opt.addInsert(opt.entity, ele);
}
});
opt.exec();
return obj;
}
public void insert(String tableName, Chain chain) {
if (chain.isSpecial()) {
Daos.insertBySpecialChain(this, null, tableName, chain);
return;
}
EntityOperator opt = _optBy(chain.toEntityMap(tableName));
if (null == opt)
return;
opt.addInsert();
opt.exec();
}
public void insert(Class<?> classOfT, Chain chain) {
if (chain.isSpecial()) {
Daos.insertBySpecialChain(this, getEntity(classOfT), null, chain);
return;
}
EntityOperator opt = _opt(classOfT);
opt.myObj = chain;
opt.addInsertSelfOnly();
// insert(chain.toObject(classOfT));// TODO 这样的效率,未免太低了,需要改进
opt.exec();
}
public <T> T fastInsert(T obj) {
EntityOperator opt = _optBy(obj);
if (null == opt)
return null;
opt.addInsertSelfOnly();
opt.exec();
return obj;
}
public <T> T insertWith(T obj, String regex) {
// TODO 天啊,每个调用都有4个正则表达式,能快起来不?
// TODO zzh: NutEntity 会缓存正则表达式计算的结果的，会很快的
EntityOperator opt = _optBy(obj);
if (null == opt)
return null;
opt.entity.visitOne(obj, regex, doInsert(opt));
opt.addInsert();
opt.entity.visitMany(obj, regex, doInsert(opt));
opt.entity.visitManyMany(obj, regex, doInsert(opt));
opt.entity.visitManyMany(obj, regex, doInsertRelation(opt));
opt.exec();
return obj;
}
public <T> T insertLinks(T obj, String regex) {
// TODO 天啊,每个调用都有4个正则表达式,能快起来不?
// TODO zzh: NutEntity 会缓存正则表达式计算的结果的，会很快的
EntityOperator opt = _optBy(obj);
if (null == opt)
return null;
opt.entity.visitOne(obj, regex, doInsert(opt));
opt.entity.visitMany(obj, regex, doInsert(opt));
opt.entity.visitManyMany(obj, regex, doInsert(opt));
opt.entity.visitManyMany(obj, regex, doInsertRelation(opt));
opt.exec();
return obj;
}
public <T> T insertRelation(T obj, String regex) {
EntityOperator opt = _optBy(obj);
if (null == opt)
return null;
opt.entity.visitManyMany(obj, regex, doInsertRelation(opt));
opt.exec();
return obj;
}
public int update(Object obj) {
EntityOperator opt = _optBy(obj);
if (null == opt)
return 0;
opt.addUpdate();
opt.exec();
return opt.getUpdateCount();
}
public int update(final Object obj, String regex) {
Object first = Lang.first(obj);
if (null == first)
return 0;
if (Strings.isBlank(regex))
return update(obj);
Molecule<Integer> m = new Molecule<Integer>() {
public void run() {
setObj(update(obj));
}
};
FieldFilter.create(first.getClass(), regex).run(m);
return m.getObj();
}
public int updateIgnoreNull(final Object obj) {
EntityOperator opt = _optBy(obj);
if (null == opt)
return 0;
opt.addUpdateForIgnoreNull(opt.entity,
obj,
FieldFilter.get(opt.entity.getType()));
opt.exec();
return opt.getUpdateCount();
}
public int update(String tableName, Chain chain, Condition cnd) {
if (chain.isSpecial())
return Daos.updateBySpecialChain(this, null, tableName, chain, cnd);
EntityOperator opt = _optBy(chain.toEntityMap(tableName));
if (null == opt)
return 0;
opt.addUpdate(cnd);
opt.exec();
return opt.getUpdateCount();
}
public int update(Class<?> classOfT, Chain chain, Condition cnd) {
if (chain.isSpecial())
return Daos.updateBySpecialChain(this,
getEntity(classOfT),
null,
chain,
cnd);
EntityOperator opt = _opt(classOfT);
opt.addUpdate(chain, cnd);
opt.exec();
return opt.getUpdateCount();
}
public <T> T updateWith(T obj, final String regex) {
if (null == obj)
return null;
Lang.each(obj, false, new Each<Object>() {
public void invoke(int index, Object ele, int length)
throws ExitLoop, ContinueLoop, LoopException {
EntityOperator opt = _optBy(ele);
if (null == opt)
return;
opt.entity.visitOne(ele, regex, doUpdate(opt));
opt.addUpdate();
opt.entity.visitMany(ele, regex, doUpdate(opt));
opt.entity.visitManyMany(ele, regex, doUpdate(opt));
opt.exec();
}
});
return obj;
}
public <T> T updateLinks(T obj, final String regex) {
if (null == obj)
return null;
Lang.each(obj, false, new Each<Object>() {
public void invoke(int index, Object ele, int length)
throws ExitLoop, ContinueLoop, LoopException {
EntityOperator opt = _optBy(ele);
if (null == opt)
return;
opt.entity.visitOne(ele, regex, doUpdate(opt));
opt.entity.visitMany(ele, regex, doUpdate(opt));
opt.entity.visitManyMany(ele, regex, doUpdate(opt));
opt.exec();
}
});
return obj;
}
public int updateRelation(Class<?> classOfT,
String regex,
Chain chain,
Condition cnd) {
if (chain.isSpecial())
throw Lang.noImplement();
EntityOperator opt = this._opt(classOfT);
opt.entity.visitManyMany(null, regex, doUpdateRelation(opt, chain, cnd));
opt.exec();
return opt.getUpdateCount();
}
public int delete(Class<?> classOfT, long id) {
Entity<?> en = holder.getEntity(classOfT);
Pojo pojo = pojoMaker.makeDelete(en).append(Pojos.Items.cndId(en, id));
pojo.addParamsBy(id);
_exec(pojo);
return pojo.getUpdateCount();
}
public int delete(Class<?> classOfT, String name) {
Entity<?> en = holder.getEntity(classOfT);
Pojo pojo = pojoMaker.makeDelete(en)
.append(Pojos.Items.cndName(en, name))
.addParamsBy(name);
_exec(pojo);
return pojo.getUpdateCount();
}
public <T> int deletex(Class<T> classOfT, Object... pks) {
Entity<T> en = holder.getEntity(classOfT);
Pojo pojo = pojoMaker.makeDelete(en).append(Pojos.Items.cndPk(en, pks));
_exec(pojo);
return pojo.getUpdateCount();
}
public int delete(Object obj) {
EntityOperator opt = _optBy(obj);
if (null == opt)
return 0;
opt.addDeleteSelfOnly();
opt.exec();
return opt.getUpdateCount();
}
public int deleteWith(Object obj, final String regex) {
if (null == obj)
return 0;
final int[] re = new int[1];
Lang.each(obj, false, new Each<Object>() {
public void invoke(int index, Object ele, int length)
throws ExitLoop, ContinueLoop, LoopException {
EntityOperator opt = _optBy(ele);
if (null == opt)
return;
opt.entity.visitMany(ele, regex, doDelete(opt));
opt.entity.visitManyMany(ele,
regex,
doClearRelationByLinkedField(opt));
opt.entity.visitManyMany(ele, regex, doDelete(opt));
opt.addDeleteSelfOnly();
opt.entity.visitOne(ele, regex, doDelete(opt));
re[0] += opt.exec().getUpdateCount();
}
});
return re[0];
}
public int deleteLinks(Object obj, final String regex) {
if (null == obj)
return 0;
final int[] re = new int[1];
Lang.each(obj, false, new Each<Object>() {
public void invoke(int index, Object ele, int length)
throws ExitLoop, ContinueLoop, LoopException {
EntityOperator opt = _optBy(ele);
if (null == opt)
return;
opt.entity.visitMany(ele, regex, doDelete(opt));
opt.entity.visitManyMany(ele,
regex,
doClearRelationByLinkedField(opt));
opt.entity.visitManyMany(ele, regex, doDelete(opt));
opt.entity.visitOne(ele, regex, doDelete(opt));
re[0] += opt.exec().getUpdateCount();
}
});
return re[0];
}
public <T> List<T> query(Class<T> classOfT, Condition cnd, Pager pager) {
Pojo pojo = pojoMaker.makeQuery(holder.getEntity(classOfT))
.append(Pojos.Items.cnd(cnd))
.addParamsBy("*")
.setPager(pager)
.setAfter(_pojo_queryEntity);
expert.formatQuery(pojo);
_exec(pojo);
return pojo.getList(classOfT);
}
public <T> List<T> query(Class<T> classOfT, Condition cnd) {
return query(classOfT, cnd, Pojos.Items.pager(cnd));
}
public <T> int each(Class<T> classOfT,
Condition cnd,
Pager pager,
Each<T> callback) {
Pojo pojo = pojoMaker.makeQuery(holder.getEntity(classOfT))
.append(Pojos.Items.cnd(cnd))
.addParamsBy("*")
.setPager(pager)
.setAfter(_pojo_queryEntity);
expert.formatQuery(pojo);
pojo.setAfter(_pojo_eachEntity);
pojo.getContext().attr(Each.class.getName(), callback);
_exec(pojo);
return pojo.getInt();
}
public <T> int each(Class<T> classOfT, Condition cnd, Each<T> callback) {
return each(classOfT, cnd, Pojos.Items.pager(cnd), callback);
}
public List<Record> query(String tableName, Condition cnd, Pager pager) {
Pojo pojo = pojoMaker.makeQuery(tableName)
.addParamsBy("*")
.setPager(pager)
.append(Pojos.Items.cnd(cnd));
expert.formatQuery(pojo);
pojo.setAfter(_pojo_queryRecord);
_exec(pojo);
return pojo.getList(Record.class);
}
public List<Record> query(String tableName, Condition cnd) {
return query(tableName, cnd, Pojos.Items.pager(cnd));
}
public int each(String tableName,
Condition cnd,
Pager pager,
Each<Record> callback) {
Pojo pojo = pojoMaker.makeQuery(tableName)
.addParamsBy("*")
.setPager(pager)
.append(Pojos.Items.cnd(cnd));
expert.formatQuery(pojo);
pojo.setAfter(_pojo_eachRecord);
pojo.getContext().attr(Each.class.getName(), callback);
_exec(pojo);
return pojo.getInt();
}
public int each(String tableName, Condition cnd, Each<Record> callback) {
return each(tableName, cnd, Pojos.Items.pager(cnd), callback);
}
public <T> T fetch(Class<T> classOfT, long id) {
Entity<T> en = holder.getEntity(classOfT);
if (en.getIdField() == null)
throw new DaoException("Need @Id for " + classOfT);
Pojo pojo = pojoMaker.makeQuery(en)
.append(Pojos.Items.cndId(en, id))
.addParamsBy(id)
.setAfter(_pojo_fetchEntity);
_exec(pojo);
return pojo.getObject(classOfT);
}
public <T> T fetch(Class<T> classOfT, String name) {
if (name == null)
throw new IllegalArgumentException("name MUST NOT NULL!");
Entity<T> en = holder.getEntity(classOfT);
if (en.getNameField() == null)
throw new DaoException("Need @Name for " + classOfT);
Pojo pojo = pojoMaker.makeQuery(en)
.append(Pojos.Items.cndName(en, name))
.addParamsBy(name)
.setAfter(_pojo_fetchEntity);
_exec(pojo);
return pojo.getObject(classOfT);
}
public <T> T fetchx(Class<T> classOfT, Object... pks) {
Entity<T> en = holder.getEntity(classOfT);
Pojo pojo = pojoMaker.makeQuery(en)
.append(Pojos.Items.cndPk(en, pks))
.setAfter(_pojo_fetchEntity);
_exec(pojo);
return pojo.getObject(classOfT);
}
public <T> T fetch(Class<T> classOfT, Condition cnd) {
Pojo pojo = pojoMaker.makeQuery(holder.getEntity(classOfT))
.append(Pojos.Items.cnd(cnd))
.addParamsBy("*")
.setPager(createPager(1, 1))
.setAfter(_pojo_fetchEntity);
expert.formatQuery(pojo);
_exec(pojo);
return pojo.getObject(classOfT);
}
public Record fetch(String tableName, Condition cnd) {
Pojo pojo = pojoMaker.makeQuery(tableName)
.append(Pojos.Items.cnd(cnd))
.addParamsBy("*")
.setPager(createPager(1, 1))
.setAfter(_pojo_fetchRecord);
expert.formatQuery(pojo);
_exec(pojo);
return pojo.getObject(Record.class);
}
@SuppressWarnings("unchecked")
public <T> T fetch(T obj) {
Entity<?> en = holder.getEntityBy(obj);
Pojo pojo = pojoMaker.makeQuery(en)
.append(Pojos.Items.cndAuto(en, obj))
.setAfter(_pojo_fetchEntity);
_exec(pojo);
return (T) pojo.getResult();
}
public <T> T fetch(Class<T> classOfT) {
List<T> list = query(classOfT, null, createPager(1, 1));
if (null != list && !list.isEmpty())
return list.get(0);
return null;
}
public <T> T fetchLinks(T obj, final String regex) {
return fetchLinks(obj, regex, null);
}
public <T> T fetchLinks(final T obj, final String regex, final Condition cnd) {
if (null == obj)
return null;
Lang.each(obj, false, new Each<Object>() {
public void invoke(int index, Object ele, int length)
throws ExitLoop, ContinueLoop, LoopException {
EntityOperator opt = _optBy(ele);
if (null == opt)
return;
opt.entity.visitMany(ele, regex, doLinkQuery(opt, cnd));
opt.entity.visitManyMany(ele, regex, doLinkQuery(opt, cnd));
opt.entity.visitOne(ele, regex, doFetch(opt));
opt.exec();
}
});
return obj;
}
public int clear(Class<?> classOfT, Condition cnd) {
Pojo pojo = pojoMaker.makeDelete(holder.getEntity(classOfT))
.append(Pojos.Items.cnd(cnd));
_exec(pojo);
return pojo.getUpdateCount();
}
public int clear(String tableName, Condition cnd) {
Pojo pojo = pojoMaker.makeDelete(tableName)
.append(Pojos.Items.cnd(cnd));
_exec(pojo);
return pojo.getUpdateCount();
}
public int clear(Class<?> classOfT) {
return clear(classOfT, null);
}
public int clear(String tableName) {
return clear(tableName, null);
}
public <T> T clearLinks(T obj, final String regex) {
if (null == obj)
return null;
Lang.each(obj, false, new Each<Object>() {
public void invoke(int index, Object ele, int length)
throws ExitLoop, ContinueLoop, LoopException {
EntityOperator opt = _optBy(ele);
if (null == opt)
return;
opt.entity.visitMany(ele, regex, doClear(opt));
opt.entity.visitManyMany(ele,
regex,
doClearRelationByHostField(opt));
opt.entity.visitOne(ele, regex, doClear(opt));
opt.exec();
}
});
return obj;
}
public <T> Entity<T> getEntity(Class<T> classOfT) {
return holder.getEntity(classOfT);
}
public int count(Class<?> classOfT, Condition cnd) {
Entity<?> en = holder.getEntity(classOfT);
return _count(en, en.getViewName(), cnd);
}
public int count(Class<?> classOfT) {
Entity<?> en = holder.getEntity(classOfT);
return _count(en, en.getViewName(), null);
}
public int count(String tableName) {
return count(tableName, null);
}
public int count(String tableName, Condition cnd) {
return _count(null, tableName, cnd);
}
private int _count(Entity<?> en, String tableName, Condition cnd) {
// 如果有条件的话
if (null != cnd) {
Pojo pojo = pojoMaker.makeFunc(tableName, "COUNT", "*");
pojo.setEntity(en);
// 高级条件接口，直接得到 WHERE 子句
if (cnd instanceof Criteria) {
pojo.append(((Criteria) cnd).where());
}
// 否则暴力获取 WHERE 子句
else {
String str = Pojos.formatCondition(en, cnd);
if (!Strings.isBlank(str)) {
String[] ss = str.toUpperCase().split("ORDER BY");
pojo.append(Pojos.Items.wrap(str.substring(0,
ss[0].length())));
}
}
// 设置回调，并执行 SQL
pojo.setAfter(_pojo_fetchInt);
_exec(pojo);
return pojo.getInt();
}
// 没有条件，直接生成表达式
return func(tableName, "COUNT", "*");
}
public int getMaxId(Class<?> classOfT) {
Entity<?> en = holder.getEntity(classOfT);
return func(en.getViewName(), "MAX", en.getIdField().getColumnName());
}
public int func(Class<?> classOfT, String funcName, String fieldName) {
return func(classOfT, funcName, fieldName, null);
}
public int func(String tableName, String funcName, String colName) {
return func(tableName, funcName, colName, null);
}
public int func(Class<?> classOfT,
String funcName,
String colName,
Condition cnd) {
Entity<?> en = holder.getEntity(classOfT);
if (null != en.getField(colName))
colName = en.getField(colName).getColumnName();
DaoStatement pojo = pojoMaker.makeFunc(en.getViewName(),
funcName,
colName)
.append(Pojos.Items.cnd(cnd))
.setAfter(_pojo_fetchInt)
.setEntity(en);
_exec(pojo);
return pojo.getInt();
}
public int func(String tableName,
String funcName,
String colName,
Condition cnd) {
DaoStatement pojo = pojoMaker.makeFunc(tableName, funcName, colName)
.append(Pojos.Items.cnd(cnd))
.setAfter(_pojo_fetchInt);
_exec(pojo);
return pojo.getInt();
}
public Pager createPager(int pageNumber, int pageSize) {
Pager pager = new Pager();
pager.setPageNumber(pageNumber);
pager.setPageSize(pageSize);
return pager;
}
public synchronized <T> Entity<T> create(Class<T> classOfT,
boolean dropIfExists) {
Entity<T> en = holder.getEntity(classOfT);
if (exists(en.getTableName())) {
if (dropIfExists) {
expert.dropEntity(this, en);
} else {
return en;
}
}
return holder.reloadEntity(this, classOfT);
}
public boolean drop(Class<?> classOfT) {
Entity<?> en = holder.getEntity(classOfT);
if (!exists(en.getTableName()))
return false;
return expert.dropEntity(this, en);
}
public boolean drop(String tableName) {
if (!exists(tableName))
return false;
Sql sql = Sqls.createf("DROP TABLE %s", tableName);
_exec(sql);
return true;
}
public boolean exists(Class<?> classOfT) {
return exists(getEntity(classOfT).getViewName());
}
public boolean exists(final String tableName) {
final boolean[] ee = {false};
this.run(new ConnCallback() {
public void invoke(Connection conn) {
Statement stat = null;
ResultSet rs = null;
try {
stat = conn.createStatement();
// 增加不等式,减少sql执行时间
String sql = "SELECT COUNT(1) FROM "
+ tableName
+ " where 1!=1";
rs = stat.executeQuery(sql);
if (rs.next())
ee[0] = true;
}
catch (SQLException e) {}
finally {
Daos.safeClose(stat, rs);
}
}
});
return ee[0];
}
// ==========================================================
// 下面几个是快速创建映射操作回调的帮助函数
private LinkVisitor doInsert(EntityOperator opt) {
return new DoInsertLinkVisitor().opt(opt);
}
private LinkVisitor doInsertRelation(EntityOperator opt) {
return new DoInsertRelationLinkVisitor(holder).opt(opt);
}
private LinkVisitor doUpdate(EntityOperator opt) {
return new DoUpdateLinkVisitor().opt(opt);
}
private LinkVisitor doUpdateRelation(EntityOperator opt,
Chain chain,
Condition cnd) {
return new DoUpdateRelationLinkVisitor(chain.toMap(), cnd).opt(opt);
}
private LinkVisitor doClearRelationByLinkedField(EntityOperator opt) {
return new DoClearRelationByLinkedFieldLinkVisitor().opt(opt);
}
private LinkVisitor doClearRelationByHostField(EntityOperator opt) {
return new DoClearRelationByHostFieldLinkVisitor().opt(opt);
}
private LinkVisitor doDelete(EntityOperator opt) {
return new DoDeleteLinkVisitor().opt(opt);
}
private LinkVisitor doClear(EntityOperator opt) {
return new DoClearLinkVisitor().opt(opt);
}
private LinkVisitor doFetch(EntityOperator opt) {
return new DoFetchLinkVisitor().opt(opt);
}
private LinkVisitor doLinkQuery(EntityOperator opt, final Condition cnd) {
return new AbstractLinkVisitor() {
public void visit(final Object obj, final LinkField lnk) {
Pojo pojo = opt.maker().makeQuery(lnk.getLinkedEntity());
pojo.setOperatingObject(obj);
PItem[] _cndItems = Pojos.Items.cnd(lnk.createCondition(obj));
pojo.append(_cndItems);
if (cnd != null) {
if (cnd instanceof Criteria) {
Criteria cri = (Criteria) cnd;
SqlExpressionGroup seg = cri.where();
if (_cndItems.length > 0
&& seg != null
&& !seg.isEmpty()) {
seg.setTop(false);
pojo.append(Pojos.Items.wrap(" AND "));
}
pojo.append(cri);
if (cri.getPager() != null) {
pojo.setPager(cri.getPager());
expert.formatQuery(pojo);
}
}
// 普通条件
else {
pojo.append(new ConditionPItem(cnd));
}
}
pojo.setAfter(new PojoCallback() {
public Object invoke(Connection conn,
ResultSet rs,
Pojo pojo) throws SQLException {
Object value = lnk.getCallback().invoke(conn, rs, pojo);
lnk.setValue(obj, value);
return value;
}
});
pojo.setEntity(lnk.getLinkedEntity());
opt.add(pojo);
}
}.opt(opt);
}
// ==========================================================
// 下面几个是快速创建实体操作对象的帮助函数
private <T> EntityOperator _opt() {
EntityOperator opt = new EntityOperator();
opt.dao = this;
return opt;
}
<T> EntityOperator _opt(Entity<T> en) {
EntityOperator opt = _opt();
opt.entity = en;
return opt;
}
<T> EntityOperator _opt(Class<T> classOfT) {
return _opt(holder.getEntity(classOfT));
}
EntityOperator _optBy(Object obj) {
// 阻止空对象
if (null == obj)
return null;
// 对象是否有内容，这里会考虑集合与数组
Entity<?> en = holder.getEntityBy(obj);
if (null == en)
return null;
// 创建操作对象
EntityOperator re = _opt(en);
re.myObj = obj.getClass().isArray() ? Lang.array2list((Object[]) obj)
: obj;
return re;
}
// ---------------------------------------------------------------
// 专属于NutDao的一些帮助方法
/**
* 当本对象被GC的时候,打印之.
* <p/>
* 因为NutDao是线程安全的,用户反复创建NutDao的话,下面的方法将有提示作用
*/
protected void finalize() throws Throwable {
log.debugf("%s[_selfId=%d] finalize",
getClass().getSimpleName(),
_selfId);
super.finalize();
}
}