Search code examples
jpaentitycriteriaentitymanagerhibernate-criteria

fetch just one child in a JPA CriteriaBuilder Join Query


I have 2 entities:

public class Tyre {

 @Id
 @Column(name = "tyreid")
 @GeneratedValue
 private int tyreid;
 @Column(name = "brand")
 private String brand;
 @Column(name = "tyretype")
 private String tyretype;
 @OneToMany(targetEntity = TyreAuto.class, cascade = CascadeType.ALL)
 @JoinColumn(name = "tyreid",referencedColumnName = "tyreid")
 private List<TyreAuto> tyreAutos;
}

and:

public class TyreAuto {
 @Id
 @Column(name = "tyreautoid")
 @GeneratedValue
 private int tyreautoid;
 @Column(name = "serie")
 private String serie;
 @Column(name = "tyreid")
 private int tyreid;
}

And I have the following code that returns a json response:

    CriteriaBuilder criteriaBuilder = entityManager.getCriteriaBuilder();
    CriteriaQuery<Tyre> criteriaQuery = criteriaBuilder.createQuery(Tyre.class);
    Metamodel m = entityManager.getMetamodel();
    EntityType<Tyre> Tyre_ = m.entity(Tyre.class);
    Root<Tyre> tyreRoot = criteriaQuery.from(Tyre_);
    Join<Tyre,TyreAuto> tyreAutos = tyreRoot.join("tyreAutos",JoinType.INNER);

    Predicate predicateForBrand
            = criteriaBuilder.equal(tyreRoot.get("brand"), "Firestone");
    Predicate predicateForSerie
            = criteriaBuilder.equal(tyreAutos.get("serie"),"XYZ");
    Predicate predicateForBrandAndSerie
            = criteriaBuilder.and(predicateForSerie,predicateForBrand);
    criteriaQuery.where(predicateForBrandAndSerie);
    List<Tyre> items = entityManager.createQuery(criteriaQuery).getResultList();

    if (!items.equals(null)) {
        return new ResponseEntity<>(items, HttpStatus.OK);
    } else {
        return new ResponseEntity(ResponseEntity.notFound().build(),HttpStatus.NOT_FOUND);
    }

this returns the following Json:

[
   {
    "tyreid": 1,
    "brand": "Firestone",
    "tyretype": "Big",
    "tyreAutos": [
        {
            "tyreautoid": 38,
            "serie": "XYZ",
            "tyreid": 13
        },
        {
            "tyreautoid": 39,
            "serie": "ABC",
            "tyreid": 13
        },
        {
            "tyreautoid": 40,
            "serie": "JKL",
            "tyreid": 13
        }
      ]
    }
 ]

My Question is:

  • Why I am getting more than one child register if I am filtering for Serie: "XYZ"?
  • How can I get just one child element instead of all 3?

How can I filter in a way that the response is something like this:

[
   {
    "tyreid": 1,
    "brand": "Firestone",
    "tyretype": "Big",
    "tyreAutos": [
        {
            "tyreautoid": 38,
            "serie": "XYZ",
            "tyreid": 13
        }
      ]
    }
 ]

Solution

  • Why I am getting more than one child register if I am filtering for Serie: "XYZ"?

    You get whole Tyre object as declared criteriaBuilder.createQuery(Tyre.class) where brand = "Firestone" and tyreAutos contains tyreAuto with serie = "XYZ".

    How can I get just one child element instead of all 3?

    To get only one TyreAuto you can use inversed query like this criteriaBuilder.createQuery(TyreAuto.class) changing you entity this way

    public class TyreAuto {
        @Id
        @Column(name = "tyreautoid")
        @GeneratedValue
        private int tyreautoid;
    
        @Column(name = "serie")
        private String serie;
    
        @ManyToOne
        @JoinColumn(name = "tyreid")
        private Tyre tyre;
    }
    

    Now you have access to the tyre from TyreAuto

    CriteriaBuilder criteriaBuilder = entityManager.getCriteriaBuilder();
    CriteriaQuery<TyreAuto> criteriaQuery = criteriaBuilder.createQuery(TyreAuto.class);
    Metamodel m = entityManager.getMetamodel();
    EntityType<TyreAuto> TyreAuto_ = m.entity(TyreAuto.class);
    Root<TyreAuto> tyreAutoRoot = criteriaQuery.from(TyreAuto_);
    Join<TyreAuto, Tyre> tyre = tyreAutoRoot.join("tyre",JoinType.INNER);
    
    Predicate predicateForBrand
            = criteriaBuilder.equal(tyre.get("brand"), "Firestone");
    Predicate predicateForSerie
            = criteriaBuilder.equal(tyreAutoRoot.get("serie"),"XYZ");
    
    criteriaQuery.select(tyreAutoRoot).where(predicateForSerie, predicateForBrand);
    List<TyreAuto> items = entityManager.createQuery(criteriaQuery).getResultList();
    

    Another option is to use dto class as query result. You can configure it you need. Here you can get more info. Read the chapter DTO projections in criteria queries