Spring数据–向存储库添加自定义方法

最新推荐文章于 2022-06-19 11:43:10 发布

cyan20115

最新推荐文章于 2022-06-19 11:43:10 发布

阅读量233

点赞数

文章标签： java 数据库开发工具

在本文中，我们将向您展示如何向Spring Data JPA CrudRepository和MongoDB MongoRepository添加自定义方法

1. Crud资源库

1.1查看一个CustomerRepository ，我们将向该存储库添加一个自定义方法。

CustomerRepository.java

package com.mkyong.dao;

import com.mkyong.model.Customer;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.CrudRepository;
import org.springframework.data.repository.query.Param;

import java.util.Date;
import java.util.List;
import java.util.stream.Stream;

public interface CustomerRepository extends CrudRepository<Customer, Long> {

    List<Customer> findByEmail(String email);

    @Query("select c from Customer c where c.email = :email")
    Stream<Customer> findByEmailReturnStream(@Param("email") String email);
	
	// I have a complicated logic, no idea how to implement with @Query ...
}

1.2创建接口。

CustomerRepositoryCustomAbc.java

package com.mkyong.dao;

import com.mkyong.model.Customer;

import java.util.List;

public interface CustomerRepositoryCustomAbc {

    List<Customer> findByAVeryComplicatedQuery(Long id, String name, String address);

}

1.3对于实现类，类名称非常严格，您需要遵循“核心存储库接口+ Impl”模式。未能遵循此模式将导致“无法找到Spring属性”错误消息。

CustomerRepositoryImpl.java

package com.mkyong.dao;

import com.mkyong.model.Customer;

import java.util.List;

public class CustomerRepositoryImpl implements CustomerRepositoryCustomAbc {

    @Override
    public List<Customer> findByAVeryComplicatedQuery(Long id, String name, String address) {

        //...

    }
}

1.4更新CustomerRepository以扩展新的CustomerRepositoryCustomAbc接口。

CustomerRepository.java

package com.mkyong.dao;

import com.mkyong.model.Customer;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.CrudRepository;
import org.springframework.data.repository.query.Param;

import java.util.Date;
import java.util.List;
import java.util.stream.Stream;

public interface CustomerRepository extends CrudRepository<Customer, Long>, CustomerRepositoryCustomAbc{

   //...
	
}

做完了

2. Mongo储存库

2.1另一个向MongoRepository添加新的“更新特定字段”方法的MongoRepository

DomainRepository.java

package com.mkyong.domain;

import org.springframework.data.mongodb.repository.MongoRepository;
import org.springframework.data.mongodb.repository.Query;

import java.util.List;

public interface DomainRepository extends MongoRepository<Domain, Long> {

    Domain findByDomainAndIP(String domain, String ip);

    @Query("{domain: { $regex: ?0 } })")
    List<Domain> findCustomByRegExDomain(String domain);

	//how to update a particular field? save() is not what I want.
}

2.2定制界面。

DomainRepositoryCustomAnyName.java

package com.mkyong.domain;

public interface DomainRepositoryCustomAnyName {

    int updateDomainDisplayFlagOnly(String domain, boolean flag);

}

2.3自定义实施。

DomainRepositoryImpl.java

package com.mkyong.domain;

import com.mongodb.WriteResult;
import org.springframework.beans.factory.annotation.Autowired;
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.data.mongodb.core.query.Update;

//Impl postfix of the name on it compared to the core repository interface
public class DomainRepositoryImpl implements DomainRepositoryCustomAnyName {

    @Autowired
    MongoTemplate mongoTemplate;

    @Override
    public int updateDomainDisplayFlagOnly(String domain, boolean flag) {

        Query query = new Query(Criteria.where("domain").is(domain));
        Update update = new Update();
        update.set("display", flag);

        WriteResult result = mongoTemplate.updateFirst(query, update, Domain.class);

        if(result!=null)
            return result.getN();
        else
            return 0;

    }
}

2.4更新DomainRepository以扩展新的自定义接口– DomainRepositoryCustomAnyName

DomainRepository.java

package com.mkyong.domain;

import org.springframework.data.mongodb.repository.MongoRepository;
import org.springframework.data.mongodb.repository.Query;

import java.util.List;

public interface DomainRepository extends MongoRepository<Domain, Long>, DomainRepositoryCustomAnyName {

    //...
	
}

做完了