All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.querydsl.sql.ForeignKey Maven / Gradle / Ivy

There is a newer version: 6.9
Show newest version
/*
 * Copyright 2015, The Querydsl Team (http://www.querydsl.com/team)
 *
 * 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 com.querydsl.sql;

import com.querydsl.core.BooleanBuilder;
import com.querydsl.core.Tuple;
import com.querydsl.core.annotations.Immutable;
import com.querydsl.core.types.*;
import com.querydsl.core.types.dsl.BooleanExpression;
import com.querydsl.core.types.dsl.Expressions;
import com.querydsl.core.util.CollectionUtils;
import java.io.Serializable;
import java.util.Collections;
import java.util.List;
import org.jetbrains.annotations.Nullable;

/**
 * {@code ForeignKey} defines a foreign key on a table to another table
 *
 * @author tiwe
 * @param 
 */
@Immutable
public final class ForeignKey implements Serializable, ProjectionRole {

  private static final long serialVersionUID = 2260578033772289023L;

  private final RelationalPath entity;

  private final List> localColumns;

  private final List foreignColumns;

  @Nullable private transient volatile Expression mixin;

  public ForeignKey(RelationalPath entity, Path localColumn, String foreignColumn) {
    this(entity, Collections.singletonList(localColumn), Collections.singletonList(foreignColumn));
  }

  public ForeignKey(
      RelationalPath entity, List> localColumns, List foreignColumns) {
    this.entity = entity;
    this.localColumns = CollectionUtils.unmodifiableList(localColumns);
    this.foreignColumns = CollectionUtils.unmodifiableList(foreignColumns);
  }

  public RelationalPath getEntity() {
    return entity;
  }

  public List> getLocalColumns() {
    return localColumns;
  }

  public List getForeignColumns() {
    return foreignColumns;
  }

  @SuppressWarnings("unchecked")
  public Predicate on(RelationalPath entity) {
    BooleanBuilder builder = new BooleanBuilder();
    for (int i = 0; i < localColumns.size(); i++) {
      Expression local = (Expression) localColumns.get(i);
      Expression foreign = ExpressionUtils.path(local.getType(), entity, foreignColumns.get(i));
      builder.and(ExpressionUtils.eq(local, foreign));
    }
    return builder.getValue();
  }

  public BooleanExpression in(SubQueryExpression coll) {
    return Expressions.booleanOperation(Ops.IN, getProjection(), coll);
  }

  @Override
  public Expression getProjection() {
    if (mixin == null) {
      mixin = ExpressionUtils.list(Tuple.class, localColumns);
    }
    return mixin;
  }
}