Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.github.aqiu202.starters.jpa.sql.AbstractQuery Maven / Gradle / Ivy
package com.github.aqiu202.starters.jpa.sql;
import com.github.aqiu202.starters.jpa.page.PageParam;
import com.github.aqiu202.starters.jpa.sql.trans.BeanTransformerAdapter;
import com.github.aqiu202.starters.jpa.sql.trans.SimpleTransformer;
import com.github.aqiu202.starters.jpa.sql.trans.inter.AnonymousResultTransformer;
import com.github.aqiu202.starters.jpa.util.JPAExpressionSingleton;
import com.github.aqiu202.util.CollectionUtils;
import com.github.aqiu202.util.StringUtils;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.StringJoiner;
import javax.persistence.EntityManager;
import javax.persistence.Query;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageImpl;
import org.springframework.data.domain.Pageable;
import org.springframework.data.domain.Sort;
public abstract class AbstractQuery extends BaseQuery {
protected final EntityManager entityManager;
protected String sql;
protected String countSql;
protected Map parameters;
protected Object[] params;
private boolean isItemParams;
private boolean isSorted = false;
protected AbstractQuery(EntityManager entityManager) {
this.entityManager = entityManager;
}
@SuppressWarnings("unchecked")
public AbstractQuery as(Class s) {
if (this.beanTransformer == null) {
this.beanTransformer = BeanTransformerAdapter.of(s);
} else if (!s.equals(this.beanTransformer.getResultClass())) {
this.beanTransformer = this.beanTransformer.as(s);
}
return (AbstractQuery) this;
}
private AbstractQuery init(String sql, boolean isItemParams) {
//解析SpEL表达式
this.sql = (String) JPAExpressionSingleton.getParser()
.parseExpression(sql, JPAExpressionSingleton.getTemplate()).getValue();
this.isItemParams = isItemParams;
return this;
}
public AbstractQuery init(String sql) {
return this.init(sql, true);
}
public AbstractQuery init(String sql, Map parameters) {
this.init(sql, false);
this.parameters = parameters;
return this;
}
public AbstractQuery init(String sql, Object... parameters) {
this.init(sql, true);
this.params = parameters;
return this;
}
public AbstractQuery countSql() {
return this.countSql(this.getCountSql(this.sql));
}
public AbstractQuery countSql(String countSql) {
this.countSql = countSql;
return this;
}
public Page pagingList(Pageable pageable) {
this.beforePagingQuery(pageable);
List list;
long total = this.total();
if (total > pageable.getOffset()) {
list = this.list();
} else {
list = new ArrayList<>();
}
return new PageImpl<>(list, pageable, total);
}
public Page pagingList(int page, int size, Sort sort) {
return pagingList(PageParam.of(page, size, sort));
}
public Page pagingList(int page, int size) {
return pagingList(page, size, null);
}
public Page pagingObjects(Pageable pageable) {
this.beforePagingQuery(pageable);
long total = this.total();
List list;
if (total > pageable.getOffset()) {
list = this.objects();
} else {
list = new ArrayList<>();
}
return new PageImpl<>(list, pageable, total);
}
public Page pagingObjects(int page, int size, Sort sort) {
return pagingObjects(PageParam.of(page, size, sort));
}
public Page pagingObjects(int page, int size) {
return pagingObjects(page, size, null);
}
public Page> pagingMapList(Pageable pageable) {
this.beforePagingQuery(pageable);
List> list;
long total = this.total();
if (total > pageable.getOffset()) {
list = this.mapList();
} else {
list = new ArrayList<>();
}
return new PageImpl<>(list, pageable, total);
}
public Page> pagingHumpMapList(Pageable pageable,
AnonymousResultTransformer anonymousResultTransformer) {
this.beforePagingQuery(pageable);
if (anonymousResultTransformer != null) {
this.mapTransformer = new SimpleTransformer(anonymousResultTransformer);
}
List> list;
long total = this.total();
if (total > pageable.getOffset()) {
list = this.humpMapList();
} else {
list = new ArrayList<>();
}
return new PageImpl<>(list, pageable, total);
}
public Page> pagingMapList(int page, int size, Sort sort) {
return pagingMapList(PageParam.of(page, size, sort));
}
public Page> pagingMapList(int page, int size) {
return pagingMapList(page, size, null);
}
public Page> pagingHumpMapList(Pageable pageable) {
return pagingHumpMapList(pageable, null);
}
public Page> pagingHumpMapList(int page, int size, Sort sort) {
return pagingHumpMapList(PageParam.of(page, size, sort), null);
}
public Page> pagingHumpMapList(int page, int size, Sort sort,
AnonymousResultTransformer anonymousResultTransformer) {
return pagingHumpMapList(PageParam.of(page, size, sort), anonymousResultTransformer);
}
public List queryList(Sort sort) {
this.prepareListQuery(this.sql, sort);
return this.list();
}
public List queryList() {
return queryList(null);
}
public List queryObjectList(Sort sort) {
this.prepareListQuery(this.sql, sort);
return this.objects();
}
public List queryObjectList() {
return queryObjectList(null);
}
public List> queryMapList() {
return this.queryMapList(null, null);
}
public List> queryMapList(Sort sort) {
this.prepareListQuery(this.sql, sort);
return this.mapList();
}
public List> queryMapList(
AnonymousResultTransformer anonymousResultTransformer) {
return queryHumpMapList(null, anonymousResultTransformer);
}
public List> queryMapList(Sort sort,
AnonymousResultTransformer anonymousResultTransformer) {
return queryHumpMapList(sort, anonymousResultTransformer);
}
public List> queryHumpMapList(Sort sort,
AnonymousResultTransformer anonymousResultTransformer) {
this.prepareListQuery(this.sql, sort);
if (anonymousResultTransformer != null) {
this.mapTransformer = new SimpleTransformer(anonymousResultTransformer);
}
return this.humpMapList();
}
public List> queryHumpMapList(Sort sort) {
return queryHumpMapList(sort, null);
}
public List> queryHumpMapList() {
return queryHumpMapList(null, null);
}
public T queryOne() {
this.prepareOneQuery(this.sql);
return this.one();
}
public Object queryObject() {
this.prepareOneQuery(this.sql);
return this.object();
}
public Map queryMap(AnonymousResultTransformer anonymousResultTransformer) {
this.prepareOneQuery(this.sql);
if (anonymousResultTransformer != null) {
this.mapTransformer = new SimpleTransformer(anonymousResultTransformer);
}
return this.humpMap();
}
public Map queryMap() {
this.prepareOneQuery(this.sql);
return this.map();
}
public Map queryHumpMap() {
this.prepareOneQuery(this.sql);
return this.humpMap();
}
public int execSql() {
this.query = this.buildQuery(this.sql);
return this.query.executeUpdate();
}
protected abstract String getCountSql(String querySql);
/**
* 添加排序
*
* @param sort 排序信息
* @author AQIU 2018年4月12日 下午5:37:22
*/
private void appendSort(Sort sort) {
if (isSorted) {
return;
}
isSorted = true;
StringJoiner joiner = new StringJoiner(",", this.sql + " order by ", "");
sort.forEach(order -> joiner.add(order.getProperty() + " " + order.getDirection().name()));
this.sql = joiner.toString();
}
private void fullParams(Query query) {
// 给条件赋上值
if (this.isItemParams) {
if (CollectionUtils.notEmpty(this.params)) {
for (int i = 0; i < this.params.length; i++) {
query.setParameter(i + 1, this.params[i]);
}
}
} else {
if (CollectionUtils.notEmpty(this.parameters)) {
for (Map.Entry entry : this.parameters.entrySet()) {
query.setParameter(entry.getKey(), entry.getValue());
}
}
}
}
private void limitOne() {
this.sql = this.sql.concat(" limit 1");
}
private void beforePagingQuery(Pageable pageable) {
this.prepareListQuery(this.sql, pageable.getSort());
if (StringUtils.isEmpty(this.countSql)) {
this.countSql();
}
this.countQuery = this.buildQuery(this.countSql);
this.pageable(pageable);
}
/**
* 列表查询之前 添加排序
*
* @param sort 排序规则
*/
private void beforeListQuery(Sort sort) {
if (sort != null) {
this.appendSort(sort);
}
}
private Query buildQuery(String sql) {
Query query = this.createQuery(sql);
this.fullParams(query);
return query;
}
private void prepareListQuery(String sql, Sort sort) {
this.beforeListQuery(sort);
this.query = this.buildQuery(sql);
}
private void prepareOneQuery(String sql) {
this.limitOne();
this.query = this.buildQuery(sql);
}
public abstract Query createQuery(String sql);
}
