SpringBoot整合mongoDB

1 搭建mongoDB

自行搜索安裝送粱。

2 整合

2.1 創(chuàng)建SpringBoot項目

直接通過idea創(chuàng)建一個springboot項目：mongodb-demo骂澄，導入MongoDB的依賴弊琴。

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-mongodb</artifactId>
</dependency>

我這里用的springboot的版本是 2.1.9.RELEASE

2.2 MongoDB的連接

在application.yml假如mongoDB的配置

server:
  port: 9999
spring:
  data:
    mongodb:
      uri: mongodb://172.23.109.238:40001
      database: user

2.3 創(chuàng)建實體類

import lombok.Data;
import org.springframework.data.annotation.Id;
import org.springframework.data.mongodb.core.mapping.Document;

@Data
@Document("user")
public class User {

    @Id
    private String id;
    private String userName;
    private Integer sex;
    private Integer age;
}

2.4 創(chuàng)建Repository

創(chuàng)建一個UserRepository接口，繼承MongoRepository<User,String>平匈，其中泛型User為實體類啄清，String為字段id的類型。

import com.cloud.mongodb.demo.entity.User;
import org.springframework.data.mongodb.repository.MongoRepository;

public interface UserRepository extends MongoRepository<User,String> {

}

2.5 創(chuàng)建Controller

創(chuàng)建一個UserController吧碾，注入UserRepository.添加測試方法凰盔。

@RestController
@RequestMapping("/user")
public class UserController {

    @Autowired
    private UserRepository userRepository;

    @PostMapping("/save")
    public void save(@RequestBody  User user) {
        userRepository.save(user);
    }

    @GetMapping("/list")
    public List<User> list() {
        return userRepository.findAll();
    }
}

2.6 啟動項目，進行調(diào)試

請求保存接口 http://localhost:9999/user/save

{
    "userName":"小明",
    "age":18,
    "sex":1
}

請求list接口倦春，查看是否保存成功,http://localhost:9999/user/list

1586749040747.png

可以看到基本操作已經(jīng)完成户敬。

3 MongoDB事務

MongoDB在4.0之前只支持單文檔事務，想要使用(Multi-Document Transaction)多文檔事務睁本，需要升級到4.0尿庐。
MongoDB的StandAlone模式不支持事務，只有ReplicaSet(復制集)才支持事務
MongoDB使用事務之后呢堰，數(shù)據(jù)庫和集合必須提前創(chuàng)建抄瑟。

3.1 配置MongoDB復制集

可以參考這個文檔來配置。

https://docs.mongodb.com/manual/tutorial/deploy-replica-set-with-keyfile-access-control/

具體操作后面有空再補充枉疼。

3.2 配置springboot開啟事務皮假。

這里我重新創(chuàng)建一個SpringBoot工程：mongodb-plus，將一些公用的配置放到一起骂维。

3.2.1 創(chuàng)建一個自動配置類

/**
 * 自動配置
 */
@Configuration
public class MongoAutoConfigure {
    @Bean
    MongoTransactionManager transactionManager(MongoDbFactory factory){
        return new MongoTransactionManager(factory);
    }
}

3.2.2 添加配置文件

在resources文件夾下增加一個spring.factories惹资，添加自動配置類

org.springframework.boot.autoconfigure.EnableAutoConfiguration=\
    com.cloud.mongodb.plus.configure.MongoAutoConfigure

3.2.3 引用mongodb-plus

可以通過maven將mongodb-plus打成jar包，或者將mongodb-plus導入到mongodb-demo中席舍，然后mongodb-demo工程pom.xml引入mongo-plus包布轿。

<dependency>
    <groupId>com.cloud.mongodb</groupId>
    <artifactId>mongodb-plus</artifactId>
    <version>0.0.1-SNAPSHOT</version>
</dependency>

3.2.4 添加service層

添加userService,并在save方法添加事務哮笆。

public interface IUserService {

    void save(User user);

    List<User> list();
}

@Service
public class UserServiceImpl implements IUserService {

    @Autowired
    private UserRepository userRepository;

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void save(User user) {
        userRepository.save(user);
    }

    @Override
    public List<User> list() {
        return userRepository.findAll();
    }

}

3.2.5 改造UserController

去掉UserRepository注入来颤，改為UserService.

@RestController
@RequestMapping("/user")
public class UserController {

    @Autowired
    private IUserService userService;

    @PostMapping("/save")
    public void save(@RequestBody  User user) {
        userService.save(user);
    }

    @GetMapping("/list")
    public List<User> list() {
        return userService.list();
    }
}

3.2.6 測試

重啟工程，按照上面的《2 創(chuàng)建SpringBoot》自行測試..

測試完可以發(fā)現(xiàn)save方法可以正常保存稠肘。

3.2.7 制造一個異常

添加一個算術(shù)異常福铅，測試保存功能是否可以正常調(diào)用。

@Override
@Transactional(rollbackFor = Exception.class)
public void save(User user) {
    userRepository.save(user);
    int i = 1/0;
}

重啟demo工程项阴，調(diào)用保存方法滑黔，事務是否會回滾。

{
    "userName":"小明3",
    "age":18,
    "sex":1
}

可以看到环揽，控制臺報了異常略荡，同時，mongoDB沒有保存上述數(shù)據(jù)歉胶，事務開啟成功汛兜。

1586760109815.png

4 抽取公用方法

常見保存，刪掉通今，根據(jù)id獲取對象數(shù)據(jù)等方法粥谬，可以抽取成一個公用的代碼肛根，不用重復造輪子。

4.1 改造mongodb-plus工程

4.1.1 創(chuàng)建BaseEntity

在mongodb-plus工程中漏策，創(chuàng)建entity的基類派哲，主要是擴展用，這里暫時只對分頁屬性擴展掺喻。

@Data
public class PageVo {
    @Transient
    private transient Integer page = 1;
    @Transient
    private transient Integer pageSize = 10;
}

public abstract class BaseEntity<T, ID extends Serializable> extends PageVo {
}

4.1.2 創(chuàng)建BaseService

在mongodb-plus工程中芭届，創(chuàng)建baseService.

public interface IBaseService<T extends BaseEntity,ID extends Serializable> {
     void save(T t);
     T selectById(ID id);
     void deleteById(ID id);
}

public class BaseServiceImpl<T extends BaseEntity, ID extends Serializable> implements IBaseService<T, ID> {

    @Autowired
    protected MongoRepository<T, ID> baseRepository;

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void save(T t) {
        ID id = FieldCacheUtil.getId(t);
        if (id == null) {
            baseRepository.save(t);
        } else {
            T dbObj = this.selectById(id);
            CopyOptions copyOptions = CopyOptions.create().ignoreNullValue();
            BeanUtil.copyProperties(t,dbObj,copyOptions);
            baseRepository.save(dbObj);
        }
    }
    
    @Override
    public T selectById(ID id) {
        Optional<T> dbObj = baseRepository.findById(id);
        return dbObj.orElse(null);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void deleteById(ID id) {
        baseRepository.deleteById(id);
    }
}

4.2 改造 mongodb-demo工程

4.2.1 實體類

繼承BaseEntity

@EqualsAndHashCode(callSuper = true)
@Data
@Document("user")
public class User extends BaseEntity<User,String> {

    @Id
    private String id;
    private String userName;
    private Integer sex;
    private Integer age;
}

4.2.2 業(yè)務層

刪除baseService已有的方法，當然這里的list也可以抽取到BaseService中感耙。

public interface IUserService extends IBaseService<User,String> {
    List<User> list();
}

@Service
public class UserServiceImpl extends BaseServiceImpl<User,String> implements IUserService {

    @Autowired
    private UserRepository userRepository;

    @Override
    public List<User> list() {
        return userRepository.findAll();
    }

}

4.2.3 啟動是否報錯喉脖，并測試.

{
    "userName":"小明4",
    "age":18,
    "sex":1
}

可以看到mongoDB中，保存成功抑月。

1586761623312.png

4.2.4 注意

FieldCacheUtil通過反射獲取id值树叽。

public class FieldCacheUtil {
    public static final Map<Class<?>, Field> idFieldMap = new ConcurrentHashMap<>();

    public FieldCacheUtil() {
    }

    @SneakyThrows
    public static <T,ID> ID getId(T t) {
        Class cla = t.getClass();
        try {
            Field primaryKeyField = getPrimaryKeyField(cla);
            return (ID) primaryKeyField.get(t);
        } catch (IllegalAccessException var3) {
            throw new Exception("獲取id失敗");
        }
    }

    @SneakyThrows
    public static Field getPrimaryKeyField(Class<?> tc) {
        Field idField = idFieldMap.getOrDefault(tc, null);
        if (Objects.nonNull(idField)) {
            return idField;
        } else {
            Field[] allFields = tc.getDeclaredFields();
            Field[] var3 = allFields;
            int var4 = allFields.length;

            for(int var5 = 0; var5 < var4; ++var5) {
                Field field = var3[var5];
                Id tableId = field.getDeclaredAnnotation(Id.class);
                if (Objects.nonNull(tableId)) {
                    field.setAccessible(true);
                    idFieldMap.putIfAbsent(tc, field);
                    return field;
                }
            }

            throw new Exception("當前實體沒有主鍵");
        }
    }

}

5 擴展公用Repository接口

上文只是針對service進行公用方法抽取，有時候需要對Repository進行自定義操作谦絮，而這些操作基本也是公用方法题诵，所以需要對Repository進行擴展。

5.1 修改mongodb-plus工程

5.1.1 創(chuàng)建公用的Repository

在dao層中添加BaseRepository层皱，該接口為自定義Repository性锭，需要添加@NoRepositoryBean注解來說明此接口不是一個聲明的Repository，否則Spring默認當做普通的Repository來處理

@NoRepositoryBean
public interface BaseRepository<T extends BaseEntity, ID extends Serializable> extends MongoRepository<T, ID>,BaseRepositoryEnhance<T, ID>  {

}

添加一個擴展的接口BaseRepositoryEnhance叫胖，所有的擴展方法定義都在此接口上(可以合并到BaseRepository中)

public interface BaseRepositoryEnhance <T extends BaseEntity, ID extends Serializable> {

    /**
     * 分頁獲取
     * @param t
     * @return
     */
    List<T> findPage(T t);

    /**
     * 獲取當前泛型的實體信息
     * @return
     */
    MongoEntityInformation<T, ID> getCurEntityInformation();
}

添加一個實現(xiàn)類草冈，主要實現(xiàn)我們自定義的方法。

public class BaseRepositoryImpl<T extends BaseEntity, ID extends Serializable> extends SimpleMongoRepository<T,ID> implements BaseRepository<T, ID> {

    private MongoOperations mongoOperations;
    private  MongoEntityInformation<T, ID> entityInformation;

    public BaseRepositoryImpl(MongoEntityInformation<T, ID> metadata, MongoOperations mongoOperations) {
        super(metadata, mongoOperations);
        this.mongoOperations = mongoOperations;
        entityInformation = metadata;
    }


    @Override
    public List<T> findPage(T t) {

        Query query = new Query();
        // 查詢pagge
        // 從那條記錄開始
        query.skip((t.getPage() - 1) * t.getPage());
        // 取多少條記錄
        query.limit(t.getPageSize());
        List<T> ts = mongoOperations.find(query, entityInformation.getJavaType(), entityInformation.getCollectionName());
        return ts;
    }

    @Override
    public MongoEntityInformation<T, ID> getCurEntityInformation() {
        return entityInformation;
    }
}

5.1.2 配置自定義的MongoRepositoryFactoryBean

創(chuàng)建CustomMongoRepositoryFactoryBean瓮增，主要是實現(xiàn)加載自定義工廠bean

public class CustomMongoRepositoryFactoryBean <T extends Repository<S, ID>, S, ID extends Serializable> extends MongoRepositoryFactoryBean<T,S,ID> {


    /**
     * 默認構(gòu)造方法
     * @param repositoryInterface
     */
    public CustomMongoRepositoryFactoryBean(Class<? extends T> repositoryInterface) {
        super(repositoryInterface);
    }

    /**
     * 設置為自定義工廠
     * @param operations
     * @return
     */
    @Override
    protected RepositoryFactorySupport getFactoryInstance(MongoOperations operations) {
        return new CustomMongoRepositoryFactory(operations);
    }

    /**
     * 自定義工廠類
     */
    public class CustomMongoRepositoryFactory extends MongoRepositoryFactory {


        private final MongoOperations operations;

        /**
         * 默認的構(gòu)造方法
         * @param mongoOperations
         */
        public CustomMongoRepositoryFactory(MongoOperations mongoOperations) {
            super(mongoOperations);
            operations = mongoOperations;
        }


        /**
         * Create a repository instance as backing for the query proxy
         * @param information
         * @return
         */
        @Override
        protected Object getTargetRepository(RepositoryInformation information) {
            //如果為自定義的類型怎棱，返回BaseRepositoryImpl
            if(isBaseRepository(information.getRepositoryInterface())) {
                MongoEntityInformation<?, Object> entityInformation = getEntityInformation(information.getDomainType());
                return getTargetRepositoryViaReflection(BaseRepositoryImpl.class,entityInformation, operations);
            }
            return super.getTargetRepository(information);
        }

        /**
         * Returns the base class backing the actual repository instance
         * @param metadata
         * @return
         */
        @Override
        protected Class<?> getRepositoryBaseClass(RepositoryMetadata metadata) {
            //如果為自定義的類型，返回BaseRepositoryImpl
            if(isBaseRepository(metadata.getRepositoryInterface())) {
                return BaseRepositoryImpl.class;
            }
            return super.getRepositoryBaseClass(metadata);
        }

        /**
         * 判斷Repository的接口是否為BaseRepository
         * @param repositoryInterface
         * @return
         */
        private boolean isBaseRepository(Class<?> repositoryInterface) {
            return ClassUtils.isAssignable(BaseRepository.class, repositoryInterface);
        }
    }
}

同時需要在MongoAutoConfigure添加指定的FactoryBean绷跑。

/**
 * 自動配置
 */
@Configuration
@EnableMongoRepositories(basePackages = {"com.cloud"},repositoryFactoryBeanClass = CustomMongoRepositoryFactoryBean.class)
public class MongoAutoConfigure {
    @Bean
    MongoTransactionManager transactionManager(MongoDbFactory factory){
        return new MongoTransactionManager(factory);
    }
}

5.1.3 修改BaseService注入類

1.前面我們注入的是MongoRepository類型拳恋，現(xiàn)在需要改為自定義的類型BaseRepository；

2.不再需要FieldCacheUtil獲取id了砸捏，baseRepository.getCurEntityInformation()即可獲取id谬运。

public class BaseServiceImpl<T extends BaseEntity, ID extends Serializable> implements IBaseService<T, ID> {

    @Autowired
    protected BaseRepository<T, ID> baseRepository;
    
    public void save(T t) {
        ID id = baseRepository.getCurEntityInformation().getId(t);
        if (id == null) {
            baseRepository.save(t);
        } else {
            T dbObj = this.selectById(id);
            CopyOptions copyOptions = CopyOptions.create().ignoreNullValue();
            BeanUtil.copyProperties(t,dbObj,copyOptions);
            baseRepository.save(dbObj);
        }
    }
    
    @Override
    public T selectById(ID id) {
        Optional<T> dbObj = baseRepository.findById(id);
        return dbObj.orElse(null);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void deleteById(ID id) {
        baseRepository.deleteById(id);
    }
}

5.1.4 打包

5.2 修改mongodb-demo工程

5.2.1 修改UserRepository

前面我們繼承的是MongoRepository類型，現(xiàn)在需要改為自定義的類型BaseRepository

public interface UserRepository extends BaseRepository<User,String> {
}

5.2.2 修改UserService

添加一個page方法

public interface IUserService extends IBaseService<User,String> {

    List<User> list();

    List<User> page(User user);
}

@Service
public class UserServiceImpl extends BaseServiceImpl<User,String> implements IUserService {

    @Autowired
    private UserRepository userRepository;

    @Override
    public List<User> list() {
        return userRepository.findAll();
    }

    @Override
    public List<User> page(User user) {
        return userRepository.findPage(user);
    }
}

5.2.3 修改UserController

添加page方法

@RestController
@RequestMapping("/user")
public class UserController {

    @Autowired
    private IUserService userService;

    @PostMapping("/save")
    public void save(@RequestBody  User user) {
        userService.save(user);
    }

    @GetMapping("/list")
    public List<User> list() {
        return userService.list();
    }

    @PostMapping("/page")
    public List<User> page(@RequestBody User user) {
        return userService.page(user);
    }
}

5.2.4 測試

訪問 POST http://localhost:9999/user/page

{
    "pageSize":2,
    "page":1
}

1586766653846.png

可以看到成功訪問垦藏。

6.擴展特定Repository接口

上面已經(jīng)對公用的Repository做了擴展梆暖，但有時候需要對特定的Repository進行擴展，這里以UserRepository為例掂骏。

6.1 創(chuàng)建UserRepositoryEnhance接口

UserRepositoryEnhance為中間接口轰驳，所有擴展方法都寫在這里。

public interface UserRepositoryEnhance {
    /**
     * 根據(jù)性別類型獲取列表
     * @param sexType
     * @return
     */
    List<User> findListBySexType(String sexType);
}

6.2 修改 UserRepository

繼承UserRepositoryEnhance。

public interface UserRepository extends BaseRepository<User,String>,UserRepositoryEnhance{

}

6.3 創(chuàng)建UserRepositoryImpl

實現(xiàn)UserRepositoryEnhance

public class UserRepositoryImpl implements UserRepositoryEnhance {

    @Autowired
    private MongoOperations mongoOperations;
    @Override
    public List<User> findListBySexType(String sexType) {
        Integer sex = "M".equals(sexType) ? 1 : 2;
        Criteria where = Criteria.where("sex").is(sex);
        Query query = Query.query(where);
        return mongoOperations.find(query, User.class);
    }
}

6.4 修改UserService

增加findListBySexType方法

public interface IUserService extends IBaseService<User,String> {

    List<User> list();

    List<User> page(User user);

    List<User> findListBySexType(String sexType);
}

@Service
public class UserServiceImpl extends BaseServiceImpl<User,String> implements IUserService {

    @Autowired
    private UserRepository userRepository;

    @Override
    public List<User> list() {
        return userRepository.findAll();
    }

    @Override
    public List<User> page(User user) {
        return userRepository.findPage(user);
    }

    @Override
    public List<User> findListBySexType(String sexType) {
        return userRepository.findListBySexType(sexType);
    }
}

6.5 修改UserController