package com.yeshi.buwan.dao.live;
|
|
import com.yeshi.buwan.domain.live.TVLiveCategory;
|
import com.yeshi.buwan.dao.base.MongodbBaseDao;
|
import com.yeshi.buwan.query.BaseQuery;
|
import org.springframework.data.domain.Sort;
|
import org.springframework.data.mongodb.core.query.Criteria;
|
import org.springframework.data.mongodb.core.query.Query;
|
import org.springframework.data.mongodb.core.query.Update;
|
import org.springframework.stereotype.Repository;
|
|
import java.util.ArrayList;
|
import java.util.Date;
|
import java.util.List;
|
|
@Repository
|
public class TVLiveCategoryDao extends MongodbBaseDao<TVLiveCategory> {
|
|
public void updateSelective(TVLiveCategory bean) {
|
Query query = new Query();
|
Update update = new Update();
|
query.addCriteria(Criteria.where("id").is(bean.getId()));
|
if (bean.getName() != null) {
|
update.set("name", bean.getName());
|
}
|
if (bean.getIcon() != null) {
|
update.set("icon", bean.getIcon());
|
}
|
if (bean.getWeight() != null) {
|
update.set("weight", bean.getWeight());
|
}
|
if (bean.getCreateTime() != null) {
|
update.set("createTime", bean.getCreateTime());
|
}
|
update.set("updateTime", new Date());
|
update(query, update);
|
}
|
|
private Query getQuery(DaoQuery daoQuery) {
|
Query query = BaseQuery.createBaseMongoQuery(daoQuery, daoQuery.sortList);
|
List<Criteria> andList = new ArrayList<>();
|
if (daoQuery.name != null) {
|
andList.add(Criteria.where("name").regex(daoQuery.name));
|
}
|
if(andList.size()>0) {
|
Criteria[] ands = new Criteria[andList.size()];
|
andList.toArray(ands);
|
query.addCriteria(new Criteria().andOperator(ands));
|
}
|
return query;
|
}
|
|
public List<TVLiveCategory> list(DaoQuery daoQuery) {
|
|
return findList(getQuery(daoQuery));
|
}
|
|
public long count(DaoQuery daoQuery) {
|
|
return count(getQuery(daoQuery));
|
}
|
|
|
public List<TVLiveCategory> listByIds(List<String> ids) {
|
if (ids == null || ids.size() == 0)
|
return null;
|
Criteria[] ors = new Criteria[ids.size()];
|
for (int i = 0; i < ors.length; i++) {
|
ors[i] = Criteria.where("id").is(ids.get(i));
|
}
|
Query query = new Query();
|
query.addCriteria(new Criteria().orOperator(ors));
|
return findList(query);
|
}
|
|
public static class DaoQuery extends BaseQuery {
|
public String name;
|
public String systemId;
|
public List<Sort.Order> sortList;
|
}
|
}
|