Spring Data Mongodb

<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("查詢失敗");
        }
    }
}

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容