org.everit.persistence.querydsl.dtoquery.PropertyQuery Maven / Gradle / Ivy

Go to download
/*
 * Copyright © 2011 Everit Kft. (http://www.everit.biz)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.everit.persistence.querydsl.dtoquery;

import java.util.Collection;
import java.util.function.BiConsumer;
import java.util.function.Function;

/**
 * The programmer can use this class to select a DTO property within a DTO.
 *
 * @param 
 *          The type of the source DTO.
 * @param 
 *          The foreign key type that connects the source DTO and its property.
 * @param 
 *          The type of the property DTO.
 */
public class PropertyQuery {

  protected DTOQuery dtoQuery;

  protected Function keyInPropertyResolver;

  protected Function keyInSourceDTOResolver;

  protected BiConsumer> oneToManySetter;

  protected BiConsumer setter;

  public PropertyQuery dtoQuery(DTOQuery dtoQuery) {
    this.dtoQuery = dtoQuery;
    return this;
  }

  public PropertyQuery keyInPropertyDTOResolver(Function keyInPropertyResolver) {
    this.keyInPropertyResolver = keyInPropertyResolver;
    return this;
  }

  public PropertyQuery keyInSourceDTOResolver(Function keyInSourceResolver) {
    this.keyInSourceDTOResolver = keyInSourceResolver;
    return this;
  }

  /**
   * Specifies the setter for one-to-many relationships as a lambda expression (functional
   * interface) that sets the property in the DTO. Either {@link #setter(BiConsumer)} or
   * {@link #oneToManySetter(BiConsumer)} should be used.
   *
   * @param oneToManySetter
   *          The setter function to set the collection property in the DTO.
   * @return This object instance.
   */
  public PropertyQuery oneToManySetter(BiConsumer> oneToManySetter) {
    if (this.setter != null) {
      throw new IllegalArgumentException(
          "Setter and oneToManySetter cannot be specified at the same time");
    }
    this.oneToManySetter = oneToManySetter;
    return this;

  }

  /**
   * Specifies the setter as a lambda expression (functional interface) that sets the property in
   * the DTO. Either {@link #setter(BiConsumer)} or {@link #oneToManySetter(BiConsumer)} should be
   * used.
   *
   * @param setter
   *          The setter function to set the property in the DTO.
   * @return This object instance.
   */
  public PropertyQuery setter(BiConsumer setter) {
    if (this.oneToManySetter != null) {
      throw new IllegalArgumentException(
          "Setter and oneToManySetter cannot be specified at the same time");
    }
    this.setter = setter;
    return this;
  }
}