package com.wind.web.common;
import com.github.pagehelper.PageHelper;
import com.wind.common.Constant;
import com.wind.common.SpringUtil;
import com.wind.mybatis.CustomMapper;
import org.apache.ibatis.session.RowBounds;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.transaction.annotation.Transactional;
import tk.mybatis.mapper.entity.Example;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Optional;
import static com.wind.common.Constant.ALL_PAGE;
public abstract class BaseService<T> {
@Autowired
protected CustomMapper<T> mapper;
/**
* 获取真实反射类型
*
* @return 反射类型
*/
Class getActualClass() {
Type type = getClass().getGenericSuperclass();
// 判断是否泛型
if (type instanceof ParameterizedType) {
// 返回表示此类型实际类型参数的Type对象的数组
Type[] types = ((ParameterizedType) type).getActualTypeArguments();
return (Class) types[0]; //将第一个泛型T对应的类返回
} else {
return (Class) type;//若没有给定泛型，则返回Object类
}
}
public Optional<T> selectByID(Long id) {
return Optional.ofNullable(mapper.selectByPrimaryKey(id));
}
/**
* 根据ID列表获取实例列表
*
* @param idList ID列表
* @return 实例列表
*/
public List<T> selectAll(List<Long> idList) {
String ids = "";
for (int index = 0; index < idList.size(); index++) {
ids += Long.toString(idList.get(index));
if (index != idList.size() - 1) {
ids += ",";
}
}
if ("".equals(ids)) {
return new ArrayList<>();
} else {
return mapper.selectByIds(ids);
}
}
/**
* 根据页数获取实例列表
*
* @param page 页数
* @return 实例列表
*/
public List<T> selectAll(int page) {
Example example = new Example(getActualClass());
example.setOrderByClause("id desc");
PageHelper.startPage(page, Constant.PAGE_SIZE);
return mapper.selectByExample(example);
}
/**
* 根据字段(字符型)和页数获取实例列表
*
* @param type 字段
* @param value 查询参数(字符型)
* @param page 页数
* @return 实例列表
*/
public List<T> selectAll(String type, String value, int page) {
if (type.toLowerCase().endsWith("id")) {
return selectAll(page, new QueryParameter(type, QueryParameterMethod.EQUAL, value, QueryParameterType.LONG));
} else {
return selectAll(page, new QueryParameter(type, QueryParameterMethod.LIKE, value, QueryParameterType.STRING));
}
}
/**
* 根据查询参数获取实例列表
*
* @param parameters 查询参数
* @return 实例列表
*/
public List<T> selectAll(QueryParameter... parameters) {
return selectAll(ALL_PAGE, parameters);
}
/**
* 根据查询参数和页数获取实例列表
*
* @param parameters 查询参数
* @param page 页数
* @return 实例列表
*/
public List<T> selectAll(int page, QueryParameter... parameters) {
if (page != ALL_PAGE) {
PageHelper.startPage(page, Constant.PAGE_SIZE);
}
return mapper.selectByExample(createExample(parameters));
}
/**
* 根据查询参数和页数获取位置最前的多个实例
*
* @param count 数量
* @param parameters 查询参数
* @return 实例列表
*/
public List<T> selectTop(int count, QueryParameter... parameters) {
return mapper.selectByExampleAndRowBounds(createExample(parameters), new RowBounds(0, count));
}
/**
* 根据查询参数和页数获取实例列表（关联查询）
*
* @param table 关联表
* @param parameters 查询参数
* @param page 页数
* @return 实例列表
*/
@SuppressWarnings("unchecked")
public List<T> selectRelatedAll(String table, int page, QueryParameter... parameters) throws Exception {
Object relatedService = SpringUtil.getBean(table + "Service");
Method[] s = relatedService.getClass().getDeclaredMethods();
Method selectMethod = relatedService.getClass().getDeclaredMethod("selectAll", QueryParameter[].class);
Object selectList = selectMethod.invoke(relatedService, new Object[]{parameters});
Method selectIdsMethod = relatedService.getClass().getDeclaredMethod("getIds", List.class);
List<Long> ids = (List<Long>) selectIdsMethod.invoke(relatedService, (List) selectList);
return selectAll(table + "Id", ids, page);
}
/**
* 根据字段(字符型)和页数获取实例列表（关联查询）
*
* @param type 字段
* @param value 查询参数(字符型)
* @param page 页数
* @return 实例列表
*/
@SuppressWarnings("unchecked")
public List<T> selectRelatedAll(String type, String value, String table, int page) throws Exception {
Object relatedService = SpringUtil.getBean(table + "Service");
Method selectMethod = relatedService.getClass().getDeclaredMethod("selectAll", String.class, String.class);
Object selectList = selectMethod.invoke(relatedService, type, value);
Method selectIdsMethod = relatedService.getClass().getDeclaredMethod("getIds", List.class);
List<Long> ids = (List<Long>) selectIdsMethod.invoke(relatedService, (List) selectList);
return selectAll(table + "Id", ids, page);
}
/**
* 根据字段(字符型)获取实例列表
*
* @param type 字段
* @param value 查询参数(字符型)
* @return 实例列表
*/
public List<T> selectAll(String type, String value) {
if (type.toLowerCase().endsWith("id")) {
return selectAll(new QueryParameter(type, QueryParameterMethod.EQUAL, value, QueryParameterType.LONG));
} else {
return selectAll(new QueryParameter(type, QueryParameterMethod.LIKE, value, QueryParameterType.STRING));
}
}
/**
* 根据字段(ID列表)和页数获取实例列表
*
* @param type 字段
* @param ids ID列表
* @param page 页数
* @return 实例列表
*/
public List<T> selectAll(String type, List<Long> ids, int page) {
if (ids.isEmpty()) {
return new ArrayList<>();
} else {
Example example = new Example(getActualClass());
example.setOrderByClause("id desc");
Example.Criteria criteria = example.createCriteria();
criteria.andIn(type, ids);
PageHelper.startPage(page, Constant.PAGE_SIZE);
return mapper.selectByExample(example);
}
}
/**
* 根据字段(ID列表)获取实例列表
*
* @param type 字段
* @param ids ID列表
* @return 实例列表
*/
public List<T> selectAll(String type, List<Long> ids) {
if (ids.isEmpty()) {
return new ArrayList<>();
} else {
Example example = new Example(getActualClass());
example.setOrderByClause("id desc");
Example.Criteria criteria = example.createCriteria();
criteria.andIn(type, ids);
return mapper.selectByExample(example);
}
}
/**
* 获取实例总数
*
* @return 实例总数
* @throws Exception 反射异常
*/
public int getCount() throws Exception {
Object object = getActualClass().newInstance();
int count = mapper.selectCount((T) object);
return count;
}
/**
* 根据查询参数获取实例总数
*
* @param parameters 查询参数
* @return 实例总数
*/
public int getCount(QueryParameter... parameters) {
int count = mapper.selectCountByExample(createExample(parameters));
return count;
}
public Example createExample(QueryParameter... parameters) {
Example example = new Example(getActualClass());
example.setOrderByClause("id desc");
Example.Criteria criteria = example.createCriteria();
for (QueryParameter parameter : parameters) {
Object value = null;
switch (parameter.getValueType()) {
case STRING:
value = parameter.getValue();
break;
case LONG:
value = Long.parseLong(parameter.getValue());
break;
case ARRAY:
value = parameter.getValue().split(",");
break;
}
switch (parameter.getMethod()) {
case LIKE:
criteria.andLike(parameter.getType(), "%" + value + "%");
break;
case EQUAL:
criteria.andEqualTo(parameter.getType(), value);
break;
case IN:
criteria.andIn(parameter.getType(), Arrays.asList((String[])value));
break;
case IS_NULL:
criteria.andIsNull(parameter.getType());
break;
case IS_NOT_NULL:
criteria.andIsNotNull(parameter.getType());
break;
}
}
return example;
}
/**
* 根据字段(字符型)获取实例总数
*
* @param type 字段
* @param value 查询参数(字符型)
* @return 实例总数
*/
public int getCount(String type, String value) {
if (type.toLowerCase().endsWith("id")) {
return getCount(new QueryParameter(type, QueryParameterMethod.EQUAL, value, QueryParameterType.LONG));
} else {
return getCount(new QueryParameter(type, QueryParameterMethod.LIKE, value, QueryParameterType.STRING));
}
}
/**
* 根据字段(字符型)获取实例总数（关联查询）
*
* @param type 字段
* @param value 查询参数(字符型)
* @return 实例总数
*/
public int getCount(String type, String value, String table) throws Exception {
Object relatedService = SpringUtil.getBean(table + "Service");
Method selectMethod = relatedService.getClass().getDeclaredMethod("selectAll", String.class, String.class);
Object selectList = selectMethod.invoke(relatedService, type, value);
Method selectIdsMethod = relatedService.getClass().getDeclaredMethod("getIds", List.class);
List<Long> ids = (List<Long>) selectIdsMethod.invoke(relatedService, selectList);
return getCount(table + "Id", ids);
}
/**
* 根据查询参数获取实例总数（关联查询）
*
* @param table 关联表
* @param parameters 查询参数
* @return 实例总数
*/
public int getCount(String table, QueryParameter... parameters) throws Exception {
Object relatedService = SpringUtil.getBean(table + "Service");
Method selectMethod = relatedService.getClass().getDeclaredMethod("selectAll", QueryParameter[].class);
Object selectList = selectMethod.invoke(relatedService, new Object[]{parameters});
Method selectIdsMethod = relatedService.getClass().getDeclaredMethod("getIds", List.class);
List<Long> ids = (List<Long>) selectIdsMethod.invoke(relatedService, selectList);
return getCount(table + "Id", ids);
}
/**
* 根据字段(ID列表)获取实例总数
*
* @param type 字段
* @param ids ID列表
* @return 实例总数
*/
public int getCount(String type, List<Long> ids) {
if (ids.isEmpty()) {
return 0;
} else {
Example example = new Example(getActualClass());
Example.Criteria criteria = example.createCriteria();
criteria.andIn(type, ids);
int count = mapper.selectCountByExample(example);
return count;
}
}
/**
* 从实例列表中提取ID列表
*
* @param list 实例列表
* @return ID列表
*/
public List<Long> getIds(List<T> list) throws Exception {
return getRelatedIds(list, "id");
}
/**
* 提取对象列表中所有特定ID字段列表
*
* @param list 对象列表
* @param filed 私有字段名
* @return 所有特定ID字段列表
* @throws Exception 反射异常
*/
public List<Long> getRelatedIds(List<T> list, String filed) throws Exception {
Class type = getActualClass();
Field field = type.getDeclaredField(filed);
field.setAccessible(true);
ArrayList<Long> result = new ArrayList<>();
for (T instance : list) {
Long e = Long.parseLong(field.get(instance).toString());
if (!result.contains(e)) {
result.add(e);
}
}
return result;
}
@Transactional
public boolean add(T t) {
return mapper.insertUseGeneratedKeys(t) > 0;
}
@Transactional
public int add(List<T> list) {
return mapper.insertList(list);
}
@Transactional
public boolean modifyById(T t) {
return mapper.updateByPrimaryKey(t) > 0;
}
@Transactional
public boolean deleteById(Long id) {
return mapper.deleteByPrimaryKey(id) > 0;
}
@Transactional
public boolean deleteAll() {
Example example = new Example(getActualClass());
return mapper.deleteByExample(example) > 0;
}
@Transactional
public boolean delete(QueryParameter... parameters) {
return mapper.deleteByExample(createExample(parameters)) > 0;
}
}
