文档章节

Beetlsql扩展之自定义Mapper

闲大赋
 闲大赋
发布于 2017/09/26 21:59
字数 1423
阅读 177
收藏 0

BeetlSql 有一个BaseMapper,提供了很多内置的Dao操作,如增删改查等10几个方法,用户只需要些一个类继承此接口便能很快的完成一个Dao,比如

public interface UserDao extends BaseMapper<User>{
}

UserDao没有包含任何方法,但集成了BaseMapper,因此具备了很多内置的Dao操作。如果这些操作不满足要求,可以使用BeetlSql的MD文档来维护复杂的SQL,比如,一个根据角色查询所有用户的操作

@SqlResource("common.user")
public interface UserDao extends BaseMapper<User>{
    List<User>  getUsersByRole(String role);
}

注解@SqlResource注明了sql文件的位置,位于common/user.md 文件,其中user.md 包含了sql查询

getUsersByRole
===

select u.* from user,user_role,role where ....

这是一个BeetlSql的基本概念,在这里不详细说明,有兴趣的可以参考官网文档 现在有这样一个需求,BaseMapper包含了太多的内置方法,有时候程序员对其中的某些方法并不敢兴趣,有的内置方法则是不允许调用的,比如deleteById 方法,总是有一次物理删除操作。应该在企业应用系统里禁止这种调用,那么,BeetlSql能否定制BaseMapper呢?

答案是能定制。只需要新写一个接口,通过SQLManager注册为BaseMapper即可,然后,为接口的每个方法添加实现即可,实际上,自带的BaseMapper就是这样实现的。

如下是一个自定义的BaseMapper,为了简单起见,Dao只提供三个接口

public interface MyBaseMapper<T>  {
	public T single(Object key);
	public T single(Object key,List attrs);
	public List<T> allData();
}

第一个方法single和第三个all方法都是原有的BaseMapper自带,第二个方法是根据主键获取对象,但只获取部分属性。我们将在博客最后讲述如何实现,现在,我们要实现allData方法。

allData方法返回目标对象对应的数据所有值,这在原有的BaseMapper已经实现,是通过org.beetl.sql.core.mapper.internal.AllAmi 实现,因此我们只需要复用这个实现。

首先,告诉SQLManager,我要添加一个新的BaseMapper

SQLManager 	sql =   ....
MapperConfig config = sql.setBaseMapper(MyBaseMapper.class);

其次,添加allData实现

config.getBuilder().addAmi("allData", new AllAmi());

这样,就非常容易的定义了一个allData内置的Dao方法,使用了已经实现的AllAmi方法,AllAmi方法也非常简单,直接使用SQLManager.all方法实现,其源码如下

public class AllAmi implements MapperInvoke {

    @Override
    public Object call(SQLManager sm, Class entityClass, String sqlId, Method m, Object[] args) {
        if (args == null || args.length == 0) {
            return sm.all(entityClass);
        }

        return sm.all(entityClass, (Integer) args[0], (Integer) args[1]);
    }

}

这样,你就具备初步自定义Dao的功能,比如如下代码

MyUserMapper dao = sql.getMapper(MyUserMapper.class);
List<User> users =dao.allData();

现在,面临新的挑战,实现一个根据主键查询,但仅仅返回需要的字段,也就是实现如下方法

public T single(Object key,List attrs);

这比较复杂,因为在BeetlSQL里并未内置,需要我们自己完成后注册,完成思路很简单,生成类似如下sql语句,并交给SQLManager 执行即可,比如

select id,name where id = #id#

我们写一个新的扩展,叫做SingleAmiExt(参考了自带的SingleAmi),其脚手架如下

@Override
public Object call(SQLManager sm, Class entityClass, String sqlId, Method m, Object[] args) {
	if (args.length == 1) {
		return sm.single(entityClass, args[0]);
	}
	List attrs = (List) args[1];
	String sql  = getSingleSelect(entityClass,sm,attrs);

	Map paras = this.setIdsParas(sm, args[0], entityClass);
	List list = sm.execute(sql, entityClass, paras);
	if(list.size()==0) {
		return null;
	}else {
		return list.get(0);
	}

}

这段代码前俩行先判断参数,如果只有一个参数,则认为没有属性过滤,直接调用内置的single方法即可。否则,需要自己拼写一个Sql模板,通过SQLManager.execute 来获取

getSingleSelect方法用于生成目标sql模板,代码如下

	private String getSingleSelect(Class cls, SQLManager sm, List attrs) {
		NameConversion nameConversion = sm.getNc();
		String condition = appendIdCondition(sm, cls);
		StringBuilder sb = new StringBuilder("select ");
	
	for (Object o : attrs) {
		String attr = (String) o;
		String col = nameConversion.getColName(cls, attr);
		sb.append(col).append(" ,");
	
	}
	// 去掉最后一逗号
	sb.setLength(sb.length() - 1);
	sb.append(" from ").append(nameConversion.getTableName(cls)).append(condition);
		return sb.toString();
	}
	
	/* 参考了AbstractDBStyle的内置代码生成办法 */
	protected String appendIdCondition(SQLManager sm, Class<?> cls) {
	
		AbstractDBStyle style = (AbstractDBStyle) sm.getDbStyle();
		MetadataManager metadataManager = sm.getMetaDataManager();
		NameConversion nameConversion = sm.getNc();
		String tableName = nameConversion.getTableName(cls);
		StringBuilder condition = new StringBuilder(" where ");
	TableDesc table = metadataManager.getTable(tableName);
	ClassDesc classDesc = table.getClassDesc(cls, nameConversion);
	
	List<String> colIds = classDesc.getIdCols();
	List<String> propertieIds = classDesc.getIdAttrs();
	Iterator<String> colIt = colIds.iterator();
	Iterator<String> propertieIt = propertieIds.iterator();
	if (colIt.hasNext() && propertieIt.hasNext()) {
		String colId = colIt.next();
		String properId = propertieIt.next();
		condition.append(style.getKeyWordHandler().getCol(colId)).append(" = ").append(style.HOLDER_START)
				.append(properId).append(style.HOLDER_END);
		while (colIt.hasNext() && propertieIt.hasNext()) {
			colId = colIt.next();
			properId = propertieIt.next();
			condition.append(" and ").append(style.getKeyWordHandler().getCol(colId)).append(" = ")
						.append(style.HOLDER_START).append(properId).append(style.HOLDER_END);
			}
		}
	
		return condition.toString();
	}

代码复杂是因为考虑到通用性,尤其是符合主键这里,如果你的表都是单一主键,或者主键都叫ID,那代码就只有几行

同样的setIdsParas也因为复合主键原因,也较为复杂,内容如下

	private Map setIdsParas(SQLManager sm, Object key, Class entityClass) {
		AbstractDBStyle style = (AbstractDBStyle) sm.getDbStyle();
		MetadataManager metadataManager = sm.getMetaDataManager();
		NameConversion nameConversion = sm.getNc();
		String tableName = nameConversion.getTableName(entityClass);

		TableDesc table = metadataManager.getTable(tableName);
		ClassDesc desc = table.getClassDesc(entityClass, nameConversion);
		Map paras = new HashMap();
		List<String> idAttrs = desc.getIdAttrs();
		if (idAttrs.size() == 1) {
			paras.put(idAttrs.get(0), key);
		} else {
			// 来自对象id的属性.复合主键
			Map<String, Object> map = desc.getIdMethods();
			for (int i = 0; i < idAttrs.size(); i++) {
				String idCol = idAttrs.get(i);
				String idAttr = idAttrs.get(i);
				Method m = (Method) map.get(idAttr);
				try {
					Object os = m.invoke(key, new Object[0]);
					paras.put(idAttr, os);
				} catch (Exception ex) {
					throw new BeetlSQLException(BeetlSQLException.ID_VALUE_ERROR, "无法设置复合主键:" + idCol, ex);
				}
			}

		}
		return paras;
	}

完成上诉代码后,可以添加这个实现

config.getBuilder().addAmi("single", new SingleAmiExt());

然后,可以使用了,比如按照主键查询,只关心id和name

dao.single(1,Arrays.asList("id","name"));

结束语: BaseMapper大部分扩展都较为简单,这里的SingleAmiExt复杂是考虑到复合主键,且现在的BeetlSQL版本还无法重用内置的代码生成片段导致的,叫来BeetlSQL版本会把内置代码生成独立出来,可以供第三方工具使用,这样,开发者就能轻易扩展BeetlSQL了。

关于BeetlSQL,请参考官网ibeetl.com

© 著作权归作者所有

共有 人打赏支持
闲大赋

闲大赋

粉丝 1123
博文 86
码字总数 81146
作品 10
西城
架构师
BeetlSQL 2.10.29 发布,Java Dao 工具

BeetlSQL 2.10.29 发布,更新如下: #IKPBA mapper 接口 支持default method,采用了liuzou1991 对mapper的优化,liuzhou也是提供BeetlSQL的Mapper功能作者之一 #IKPBB 代码生成优化,采用八...

闲大赋
06/27
0
0
BeetlSQL,简单和强大数据库访问工具(更新)

beetlsql 特点 BeetSql是一个全功能DAO工具, 同时具有Hibernate 优点 & Mybatis优点功能,适用于承认以SQL为中心,同时又需求工具能自动能生成大量常用的SQL的应用。 无需注解,自动生成大量...

闲大赋
2015/08/17
0
61
BeetlSQL 2.10.26 发布,Java Dao 工具  

BeetlSQL 2.10.26 发布了,本次发布修复了JDK9和10运行中的问题,以及对存储过程调用做了少许增强,支持了insert ignore into 的 内置插入 [feature] #IK7M3 insert ignore 支持 [feature] ...

闲大赋
06/04
0
0
闲.大赋/dao-benchmark

dao-benchmark 项目介绍 dao 性能测试,包含jpa,myabtis,beetlsql等Dao工具,测试了插入,修改,查询,翻页查询等常用操作性能,以此来优化BeetlSQL性能 测试过程说明 采用H2内存数据库,如...

闲.大赋
05/22
0
0
BeetlSQL 2.10.23 发布,Java Dao 工具

本次主要增加了Spring Boot 多数据源配置功能 #IJV68 beetlsql starter 支持多数据源,mapper也支持 #IJT15 SQLiteStyle 分页 SQL 错误 #IJSD7 Query 查询 其他库表有问题 #IJNOD sql文件默认...

闲大赋
05/21
0
0

没有更多内容

加载失败,请刷新页面

加载更多

下一页

day63-20180821-流利阅读笔记-待学习

性别歧视在日本:“我是女生,所以社会不让我学医” 毛西 2018-08-21 1.今日导读 大家在看病的时候,有留意过女医生的比例吗?在性别歧视现象十分严重的日本,男医生和女医生的比例达到了惊人...

aibinxiao
40分钟前
2
0
Ubuntu18.04 显卡GF-940MX安装NVIDIA-390.77

解决办法: 下面就给大家一个正确的姿势在Ubuntu上安装Nvidia驱动: (a)首先去N卡官网下载自己显卡对应的驱动:www.geforce.cn/drivers (b)下载后好放在英文路径的目录下,怎么简单怎么来...

AI_SKI
今天
4
0
深夜胡思乱想

魔兽世界 最近魔兽世界出了新版本, 周末两天升到了满级,比之前的版本体验好很多,做任务不用抢怪了,不用组队打怪也是共享拾取的。技能简化了很多,哪个亮按哪个。 运维 服务器 产品 之间的...

Firxiao
今天
1
0
MySQL 8 在 Windows 下安装及使用

MySQL 8 带来了全新的体验,比如支持 NoSQL、JSON 等,拥有比 MySQL 5.7 两倍以上的性能提升。本文讲解如何在 Windows 下安装 MySQL 8,以及基本的 MySQL 用法。 下载 下载地址 https://dev....

waylau
今天
1
0
微信第三方平台 access_token is invalid or not latest

微信第三方开发平台code换session_key说的特别容易,但是我一使用就带来无穷无尽的烦恼,搞了一整天也无济于事. 现在记录一下解决问题的过程,方便后来人参考. 我遇到的这个问题搜索了整个网络也...

自由的开源
今天
3
0

没有更多内容

加载失败,请刷新页面

加载更多

下一页

返回顶部
顶部