src/org/redkale/source/DataSqlSource.java

@@ -110,6 +110,15 @@ public abstract class DataSqlSource<DBChannel> extends AbstractService implement
     //创建连接池
     protected abstract PoolSource<DBChannel> createPoolSource(DataSource source, String rwtype, Properties prop);
 
+    //查询单条记录
+    protected abstract <T> CompletableFuture<T> findDB(final EntityInfo<T> info, final String sql, final boolean onlypk, final SelectColumn selects);
+
+    //查询单条记录的单个字段
+    protected abstract <T> CompletableFuture<Serializable> findColumnDB(final EntityInfo<T> info, final String sql, final boolean onlypk, final String column, final Serializable defValue);
+
+    //查询单条记录的单个字段
+    protected abstract <T> CompletableFuture<Boolean> existsDB(final EntityInfo<T> info, final String sql, final boolean onlypk);
+
     @Override
     protected ExecutorService getExecutor() {
         return executor;
@@ -481,15 +490,7 @@ public abstract class DataSqlSource<DBChannel> extends AbstractService implement
             T rs = cache.find(selects, pk);
             if (cache.isFullLoaded() || rs != null) return rs;
         }
-        if (info.isVirtualEntity()) {
-            return find(null, info, selects, pk).join();
-        } else {
-            if (isAysnc()) {
-                return readPool.pollAsync().thenCompose(conn -> find(conn, info, selects, pk)).join();
-            } else {
-                return find(readPool.poll(), info, selects, pk).join();
-            }
-        }
+        return find(info, selects, pk).join();
     }
 
     @Override
@@ -500,27 +501,244 @@ public abstract class DataSqlSource<DBChannel> extends AbstractService implement
             T rs = cache.find(selects, pk);
             if (cache.isFullLoaded() || rs != null) return CompletableFuture.completedFuture(rs);
         }
-        if (info.isVirtualEntity()) {
-            if (isAysnc()) {
-                return find(null, info, selects, pk);
-            } else {
-                return CompletableFuture.supplyAsync(() -> find(null, info, selects, pk).join(), getExecutor());
-            }
-        } else {
-            if (isAysnc()) {
-                return readPool.pollAsync().thenCompose(conn -> find(conn, info, selects, pk));
-            } else {
-                return CompletableFuture.supplyAsync(() -> find(readPool.poll(), info, selects, pk).join(), getExecutor());
-            }
-        }
+        if (isAysnc()) return find(info, selects, pk);
+        return CompletableFuture.supplyAsync(() -> find(info, selects, pk).join(), getExecutor());
     }
 
-    protected <T> CompletableFuture<T> find(final DBChannel conn, final EntityInfo<T> info, final SelectColumn selects, Serializable pk) {
-        final SelectColumn sels = selects;
-        final String sql = "SELECT " + info.getQueryColumns(null, sels) + " FROM " + info.getTable(pk) + " WHERE " + info.getPrimarySQLColumn() + " = " + FilterNode.formatToString(pk);
+    protected <T> CompletableFuture<T> find(final EntityInfo<T> info, final SelectColumn selects, Serializable pk) {
+        final String sql = "SELECT " + info.getQueryColumns(null, selects) + " FROM " + info.getTable(pk) + " WHERE " + info.getPrimarySQLColumn() + " = " + FilterNode.formatToString(pk);
         if (info.isLoggable(logger, Level.FINEST)) logger.finest(info.getType().getSimpleName() + " find sql=" + sql);
-        //待实现
-        return null;
+        return findDB(info, sql, true, selects);
+    }
+
+    @Override
+    public <T> T find(final Class<T> clazz, final String column, final Serializable key) {
+        return find(clazz, null, FilterNode.create(column, key));
+    }
+
+    @Override
+    public <T> CompletableFuture<T> findAsync(final Class<T> clazz, final String column, final Serializable key) {
+        return findAsync(clazz, null, FilterNode.create(column, key));
+    }
+
+    @Override
+    public <T> T find(final Class<T> clazz, final FilterBean bean) {
+        return find(clazz, null, FilterNodeBean.createFilterNode(bean));
+    }
+
+    @Override
+    public <T> CompletableFuture<T> findAsync(final Class<T> clazz, final FilterBean bean) {
+        return findAsync(clazz, null, FilterNodeBean.createFilterNode(bean));
+    }
+
+    @Override
+    public <T> T find(final Class<T> clazz, final FilterNode node) {
+        return find(clazz, null, node);
+    }
+
+    @Override
+    public <T> CompletableFuture<T> findAsync(final Class<T> clazz, final FilterNode node) {
+        return findAsync(clazz, null, node);
+    }
+
+    @Override
+    public <T> T find(final Class<T> clazz, final SelectColumn selects, final FilterBean bean) {
+        return find(clazz, selects, FilterNodeBean.createFilterNode(bean));
+    }
+
+    @Override
+    public <T> CompletableFuture<T> findAsync(final Class<T> clazz, final SelectColumn selects, final FilterBean bean) {
+        return findAsync(clazz, selects, FilterNodeBean.createFilterNode(bean));
+    }
+
+    @Override
+    public <T> T find(final Class<T> clazz, final SelectColumn selects, final FilterNode node) {
+        final EntityInfo<T> info = loadEntityInfo(clazz);
+        final EntityCache<T> cache = info.getCache();
+        if (cache != null && cache.isFullLoaded() && (node == null || node.isCacheUseable(this))) return cache.find(selects, node);
+        return find(info, selects, node).join();
+    }
+
+    @Override
+    public <T> CompletableFuture<T> findAsync(final Class<T> clazz, final SelectColumn selects, final FilterNode node) {
+        final EntityInfo<T> info = loadEntityInfo(clazz);
+        final EntityCache<T> cache = info.getCache();
+        if (cache != null && cache.isFullLoaded() && (node == null || node.isCacheUseable(this))) {
+            return CompletableFuture.completedFuture(cache.find(selects, node));
+        }
+        if (isAysnc()) return find(info, selects, node);
+        return CompletableFuture.supplyAsync(() -> find(info, selects, node).join(), getExecutor());
+    }
+
+    protected <T> CompletableFuture<T> find(final EntityInfo<T> info, final SelectColumn selects, final FilterNode node) {
+        final Map<Class, String> joinTabalis = node == null ? null : node.getJoinTabalis();
+        final CharSequence join = node == null ? null : node.createSQLJoin(this, false, joinTabalis, new HashSet<>(), info);
+        final CharSequence where = node == null ? null : node.createSQLExpress(info, joinTabalis);
+        final String sql = "SELECT " + info.getQueryColumns("a", selects) + " FROM " + info.getTable(node) + " a" + (join == null ? "" : join) + ((where == null || where.length() == 0) ? "" : (" WHERE " + where));
+        if (info.isLoggable(logger, Level.FINEST)) logger.finest(info.getType().getSimpleName() + " find sql=" + sql);
+        return findDB(info, sql, false, selects);
+    }
+
+    @Override
+    public <T> Serializable findColumn(final Class<T> clazz, final String column, final Serializable pk) {
+        return findColumn(clazz, column, null, pk);
+    }
+
+    @Override
+    public <T> CompletableFuture<Serializable> findColumnAsync(final Class<T> clazz, final String column, final Serializable pk) {
+        return findColumnAsync(clazz, column, null, pk);
+    }
+
+    @Override
+    public <T> Serializable findColumn(final Class<T> clazz, final String column, final FilterBean bean) {
+        return findColumn(clazz, column, null, FilterNodeBean.createFilterNode(bean));
+    }
+
+    @Override
+    public <T> CompletableFuture<Serializable> findColumnAsync(final Class<T> clazz, final String column, final FilterBean bean) {
+        return findColumnAsync(clazz, column, null, FilterNodeBean.createFilterNode(bean));
+    }
+
+    @Override
+    public <T> Serializable findColumn(final Class<T> clazz, final String column, final FilterNode node) {
+        return findColumn(clazz, column, null, node);
+    }
+
+    @Override
+    public <T> CompletableFuture<Serializable> findColumnAsync(final Class<T> clazz, final String column, final FilterNode node) {
+        return findColumnAsync(clazz, column, null, node);
+    }
+
+    @Override
+    public <T> Serializable findColumn(final Class<T> clazz, final String column, final Serializable defValue, final Serializable pk) {
+        final EntityInfo<T> info = loadEntityInfo(clazz);
+        final EntityCache<T> cache = info.getCache();
+        if (cache != null) {
+            Serializable val = cache.findColumn(column, defValue, pk);
+            if (cache.isFullLoaded() || val != null) return val;
+        }
+        return findColumn(info, column, defValue, pk).join();
+    }
+
+    @Override
+    public <T> CompletableFuture<Serializable> findColumnAsync(final Class<T> clazz, final String column, final Serializable defValue, final Serializable pk) {
+        final EntityInfo<T> info = loadEntityInfo(clazz);
+        final EntityCache<T> cache = info.getCache();
+        if (cache != null) {
+            Serializable val = cache.findColumn(column, defValue, pk);
+            if (cache.isFullLoaded() || val != null) return CompletableFuture.completedFuture(val);
+        }
+        if (isAysnc()) return findColumn(info, column, defValue, pk);
+        return CompletableFuture.supplyAsync(() -> findColumn(info, column, defValue, pk).join(), getExecutor());
+    }
+
+    protected <T> CompletableFuture<Serializable> findColumn(final EntityInfo<T> info, String column, final Serializable defValue, final Serializable pk) {
+        final String sql = "SELECT " + info.getSQLColumn(null, column) + " FROM " + info.getTable(pk) + " WHERE " + info.getPrimarySQLColumn() + " = " + FilterNode.formatToString(pk);
+        if (info.isLoggable(logger, Level.FINEST)) logger.finest(info.getType().getSimpleName() + " find sql=" + sql);
+        return findColumnDB(info, sql, true, column, defValue);
+    }
+
+    @Override
+    public <T> Serializable findColumn(final Class<T> clazz, final String column, final Serializable defValue, final FilterNode node) {
+        final EntityInfo<T> info = loadEntityInfo(clazz);
+        final EntityCache<T> cache = info.getCache();
+        if (cache != null) {
+            Serializable val = cache.findColumn(column, defValue, node);
+            if (cache.isFullLoaded() || val != null) return val;
+        }
+        return findColumn(info, column, defValue, node).join();
+    }
+
+    @Override
+    public <T> CompletableFuture<Serializable> findColumnAsync(final Class<T> clazz, final String column, final Serializable defValue, final FilterNode node) {
+        final EntityInfo<T> info = loadEntityInfo(clazz);
+        final EntityCache<T> cache = info.getCache();
+        if (cache != null) {
+            Serializable val = cache.findColumn(column, defValue, node);
+            if (cache.isFullLoaded() || val != null) return CompletableFuture.completedFuture(val);
+        }
+        if (isAysnc()) return findColumn(info, column, defValue, node);
+        return CompletableFuture.supplyAsync(() -> findColumn(info, column, defValue, node).join(), getExecutor());
+    }
+
+    protected <T> CompletableFuture<Serializable> findColumn(final EntityInfo<T> info, String column, final Serializable defValue, final FilterNode node) {
+        final Map<Class, String> joinTabalis = node == null ? null : node.getJoinTabalis();
+        final CharSequence join = node == null ? null : node.createSQLJoin(this, false, joinTabalis, new HashSet<>(), info);
+        final CharSequence where = node == null ? null : node.createSQLExpress(info, joinTabalis);
+        final String sql = "SELECT " + info.getSQLColumn("a", column) + " FROM " + info.getTable(node) + " a" + (join == null ? "" : join) + ((where == null || where.length() == 0) ? "" : (" WHERE " + where));
+        if (info.isLoggable(logger, Level.FINEST)) logger.finest(info.getType().getSimpleName() + " find sql=" + sql);
+        return findColumnDB(info, sql, false, column, defValue);
+    }
+
+    @Override
+    public <T> boolean exists(Class<T> clazz, Serializable pk) {
+        final EntityInfo<T> info = loadEntityInfo(clazz);
+        final EntityCache<T> cache = info.getCache();
+        if (cache != null) {
+            boolean rs = cache.exists(pk);
+            if (rs || cache.isFullLoaded()) return rs;
+        }
+        return exists(info, pk).join();
+    }
+
+    @Override
+    public <T> CompletableFuture<Boolean> existsAsync(final Class<T> clazz, final Serializable pk) {
+        final EntityInfo<T> info = loadEntityInfo(clazz);
+        final EntityCache<T> cache = info.getCache();
+        if (cache != null) {
+            boolean rs = cache.exists(pk);
+            if (rs || cache.isFullLoaded()) return CompletableFuture.completedFuture(rs);
+        }
+        if (isAysnc()) return exists(info, pk);
+        return CompletableFuture.supplyAsync(() -> exists(info, pk).join(), getExecutor());
+    }
+
+    protected <T> CompletableFuture<Boolean> exists(final EntityInfo<T> info, Serializable pk) {
+        final String sql = "SELECT COUNT(*) FROM " + info.getTable(pk) + " WHERE " + info.getPrimarySQLColumn() + " = " + FilterNode.formatToString(pk);
+        if (info.isLoggable(logger, Level.FINEST)) logger.finest(info.getType().getSimpleName() + " exists sql=" + sql);
+        return existsDB(info, sql, true);
+    }
+
+    @Override
+    public <T> boolean exists(final Class<T> clazz, final FilterBean bean) {
+        return exists(clazz, FilterNodeBean.createFilterNode(bean));
+    }
+
+    @Override
+    public <T> CompletableFuture<Boolean> existsAsync(final Class<T> clazz, final FilterBean bean) {
+        return existsAsync(clazz, FilterNodeBean.createFilterNode(bean));
+    }
+
+    @Override
+    public <T> boolean exists(final Class<T> clazz, final FilterNode node) {
+        final EntityInfo<T> info = loadEntityInfo(clazz);
+        final EntityCache<T> cache = info.getCache();
+        if (cache != null) {
+            boolean rs = cache.exists(node);
+            if (rs || cache.isFullLoaded()) return rs;
+        }
+        return exists(info, node).join();
+    }
+
+    @Override
+    public <T> CompletableFuture<Boolean> existsAsync(final Class<T> clazz, final FilterNode node) {
+        final EntityInfo<T> info = loadEntityInfo(clazz);
+        final EntityCache<T> cache = info.getCache();
+        if (cache != null) {
+            boolean rs = cache.exists(node);
+            if (rs || cache.isFullLoaded()) return CompletableFuture.completedFuture(rs);
+        }
+        if (isAysnc()) return exists(info, node);
+        return CompletableFuture.supplyAsync(() -> exists(info, node).join(), getExecutor());
+    }
+
+    protected <T> CompletableFuture<Boolean> exists(final EntityInfo<T> info, FilterNode node) {
+        final Map<Class, String> joinTabalis = node == null ? null : node.getJoinTabalis();
+        final CharSequence join = node == null ? null : node.createSQLJoin(this, false, joinTabalis, new HashSet<>(), info);
+        final CharSequence where = node == null ? null : node.createSQLExpress(info, joinTabalis);
+        final String sql = "SELECT COUNT(" + info.getPrimarySQLColumn("a") + ") FROM " + info.getTable(node) + " a" + (join == null ? "" : join) + ((where == null || where.length() == 0) ? "" : (" WHERE " + where));
+        if (info.isLoggable(logger, Level.FINEST)) logger.finest(info.getType().getSimpleName() + " exists sql=" + sql);
+        return existsDB(info, sql, false);
     }
 
     protected <T> Sheet<T> querySheet(final boolean readcache, final boolean needtotal, final Class<T> clazz, final SelectColumn selects, final Flipper flipper, final FilterNode node) {