com.yuehuanghun.mybatis.milu.generic.impl.GenericCountByCriteriaProviderSql Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mybatis-milu Show documentation

A mybatis orm enhance framework.

There is a newer version: 1.18.0

package com.yuehuanghun.mybatis.milu.generic.impl;

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.function.Consumer;

import org.apache.ibatis.javassist.scopedpool.SoftValueHashMap;

import com.yuehuanghun.mybatis.milu.criteria.Expression;
import com.yuehuanghun.mybatis.milu.criteria.Predicate;
import com.yuehuanghun.mybatis.milu.criteria.PredicateImpl;
import com.yuehuanghun.mybatis.milu.criteria.builder.CountSqlTemplateBuilder;
import com.yuehuanghun.mybatis.milu.generic.GenericProviderContext;
import com.yuehuanghun.mybatis.milu.generic.GenericProviderSql;
import com.yuehuanghun.mybatis.milu.tool.Constants;

public class GenericCountByCriteriaProviderSql implements GenericProviderSql {

	private final Map, Map> cache = new ConcurrentHashMap<>();
	
	@SuppressWarnings({ "rawtypes", "unchecked" })
	@Override
	public String provideSql(GenericProviderContext context, Object params) {
		Object criteria = ((Map)params).get(Constants.CRITERIA);
		Predicate predicate;
		
		if(Consumer.class.isInstance(criteria)) {
			predicate = new PredicateImpl();			
			((Consumer)criteria).accept(predicate);
		} else {
			predicate = (Predicate) criteria;
		}

		Map queryParams = new HashMap<>();
		predicate.renderParams(context, queryParams, 0);
		
		((Map)params).putAll(queryParams);

		String sqlExpression = cache.computeIfAbsent(context.getMapperType(), (clazz) -> {
			return new SoftValueHashMap<>();
		}).computeIfAbsent(predicate, (key) -> {
			return new CountSqlTemplateBuilder(context, predicate).build();
		});
		
		return sqlExpression;
	}

	@Override
	public String getMethodName() {
		return "countByCriteria";
	}

}