<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
<modelVersion>4.0.0</modelVersion>
<groupId>com.sun</groupId>
<artifactId>chat-records</artifactId>
<version>0.0.1-SNAPSHOT</version>
<parent>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-parent</artifactId>
<version>2.0.0.RELEASE</version>
<relativePath />
</parent>
<properties>
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
<project.reporting.outputEncoding>UTF-8</project.reporting.outputEncoding>
<java.version>1.8</java.version>
</properties>
<dependencies>
<!-- 添加spring-boot-starter-web模块依赖 -->
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-web</artifactId>
</dependency>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-data-mongodb</artifactId>
</dependency>
<dependency>
<groupId>io.springfox</groupId>
<artifactId>springfox-swagger2</artifactId>
<version>2.9.2</version>
</dependency>
<dependency>
<groupId>io.springfox</groupId>
<artifactId>springfox-swagger-ui</artifactId>
<version>2.9.2</version>
</dependency>
<dependency>
<groupId>junit</groupId>
<artifactId>junit</artifactId>
<scope>test</scope>
</dependency>
</dependencies>
<build>
<plugins>
<plugin>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-maven-plugin</artifactId>
<version>1.3.2.RELEASE</version>
<configuration>
<mainClass>com.sun.Application</mainClass>
</configuration>
<executions>
<execution>
<goals>
<goal>repackage</goal>
</goals>
</execution>
</executions>
</plugin>
</plugins>
</build>
</project>
package com.sun.repository;
import java.util.Date;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.Pageable;
import org.springframework.data.mongodb.repository.MongoRepository;
import org.springframework.data.mongodb.repository.Query;
import com.sun.entity.Message;
public interface MessageRepository extends MongoRepository<Message, Integer> {
}
MongoTemplate的使用记录
package com.sun.controller;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.UUID;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Pageable;
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.http.ResponseEntity;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import com.google.common.cache.Cache;
import com.sun.entity.Message;
import com.sun.exception.CommonException;
import com.sun.model.QueryRequest;
import com.sun.repository.MessageRepository;
import com.sun.utils.DateUtils;
import com.sun.utils.Response;
import com.sun.utils.ResponseCode;
@RestController
@RequestMapping("/record")
public class MessageController {
@Autowired
private MessageRepository messageRepository;
@Autowired
private MongoTemplate mongoTemplate;
@PostMapping("/save")
public ResponseEntity<?> saveRecord(@RequestBody Message message) {
try {
message.setId(UUID.randomUUID().toString());
message.setCreatTime(new Date());
if (messageRepository.insert(message) != null) {
return ResponseEntity.ok(Response.success());
}
return ResponseEntity.ok(Response.error("保存失败",
ResponseCode.CODE_FAIL));
} catch (Exception e) {
throw new CommonException("操作失败");
}
}
@PostMapping("/query")
public ResponseEntity<?> queryRecord(@RequestBody QueryRequest queryRequest) {
Criteria dateBetween = Criteria.where("creatTime")
.gte(DateUtils.dateToISODate(queryRequest.getStartTime()))
.lt(DateUtils.dateToISODate(queryRequest.getEndTime()));
Criteria c = new Criteria();
c.andOperator(dateBetween,
Criteria.where("userId").is(queryRequest.getUserId()), Criteria
.where("targetId").is(queryRequest.getTargetId()));
Query query = new Query();
query.addCriteria(c);
Pageable pageable = PageRequest.of(queryRequest.getPage(),
queryRequest.getSize());
query.with(pageable);
try {
List<Message> messages = mongoTemplate.find(query, Message.class);
return ResponseEntity.ok(Response.success(messages));
} catch (Exception e) {
e.printStackTrace();
throw new CommonException("查询失败");
}
}
@PostMapping("/queryAll")
public ResponseEntity<?> queryAllRecords(
@RequestBody QueryRequest queryRequest) {
Query query = new Query();
if (!StringUtils.isEmpty(queryRequest.getStartTime())
&& !StringUtils.isEmpty(queryRequest.getEndTime())) {
Criteria dateBetween = Criteria.where("creatTime")
.gte(DateUtils.dateToISODate(queryRequest.getStartTime()))
.lt(DateUtils.dateToISODate(queryRequest.getEndTime()));
query.addCriteria(dateBetween);
}
long count = mongoTemplate.count(query, Message.class);
Pageable pageable = PageRequest.of(queryRequest.getPage() - 1,
queryRequest.getSize());
//分页排序
query.with(pageable).with(new Sort(new Sort.Order(Sort.Direction.DESC,"creatTime")));
try {
List<Message> messages = mongoTemplate.find(query, Message.class);
Map<String, Object> resMap = new HashMap<String, Object>();
resMap.put("page", pageable.getPageNumber() + 1);
resMap.put("rows", pageable.getPageSize());
resMap.put("pageList", messages);
resMap.put("total", count);
return ResponseEntity.ok(Response.success(resMap));
} catch (Exception e) {
e.printStackTrace();
throw new CommonException("查询失败");
}
}
}