com.querydsl.codegen.PropertyHandling Maven / Gradle / Ivy

Go to download
/*
 * 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.codegen;

import com.querydsl.core.annotations.QueryInit;
import com.querydsl.core.annotations.QueryType;
import java.lang.annotation.Annotation;
import java.lang.reflect.AnnotatedElement;
import java.lang.reflect.Field;
import java.lang.reflect.Method;

/** Defines handling of fields and getters in property extractions */
public enum PropertyHandling {
  /** Inspect fields and methods */
  ALL {
    @Override
    public Config getConfig(Class type) {
      return Config.ALL;
    }
  },
  /** Inspect fields only */
  FIELDS {
    @Override
    public Config getConfig(Class type) {
      return Config.FIELDS;
    }
  },
  /** Inspect methods only */
  METHODS {
    @Override
    public Config getConfig(Class type) {
      return Config.METHODS;
    }
  },
  /** No member inspection */
  NONE {
    @Override
    public Config getConfig(Class type) {
      return Config.NONE;
    }
  },
  /** JDO compatibility */
  JDO {
    @Override
    public Config getConfig(Class type) {
      var fields = false;
      var methods = false;
      for (Field field : type.getDeclaredFields()) {
        fields |= hasAnnotations(field, "javax.jdo.annotations.");
      }
      for (Method method : type.getDeclaredMethods()) {
        methods |= hasAnnotations(method, "javax.jdo.annotations.");
      }
      return Config.of(fields, methods, Config.FIELDS);
    }
  },
  /** JPA compatibility */
  JPA {
    @Override
    public Config getConfig(Class type) {
      var fields = false;
      var methods = false;
      for (Field field : type.getDeclaredFields()) {
        fields |= hasAnnotations(field, "jakarta.persistence.");
      }
      for (Method method : type.getDeclaredMethods()) {
        methods |= hasAnnotations(method, "jakarta.persistence.");
      }
      return Config.of(fields, methods, Config.ALL);
    }
  };

  private static boolean hasAnnotations(AnnotatedElement element, String packagePrefix) {
    for (Annotation ann : element.getAnnotations()) {
      if (ann.annotationType().getName().startsWith(packagePrefix)) {
        return true;
      }
    }
    return element.isAnnotationPresent(QueryType.class)
        || element.isAnnotationPresent(QueryInit.class);
  }

  public abstract Config getConfig(Class type);

  /** Property handling options */
  public enum Config {
    ALL(true, true),
    FIELDS(true, false),
    METHODS(false, true),
    NONE(false, false);
    private final boolean fields, methods;

    Config(boolean fields, boolean methods) {
      this.fields = fields;
      this.methods = methods;
    }

    public boolean isFields() {
      return fields;
    }

    public boolean isMethods() {
      return methods;
    }

    public static Config of(boolean fields, boolean methods, Config defaultConfig) {
      if (fields && methods) {
        return ALL;
      } else if (fields) {
        return FIELDS;
      } else if (methods) {
        return METHODS;
      } else {
        return defaultConfig;
      }
    }
  }
}