com.salesforce.functions.jvm.runtime.json.Util Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sf-fx-runtime-java-runtime Show documentation

There is a newer version: 1.1.7

Show newest version

/*
 * Copyright (c) 2021, salesforce.com, inc.
 * All rights reserved.
 * SPDX-License-Identifier: BSD-3-Clause
 * For full license text, see the LICENSE file in the repo root or https://opensource.org/licenses/BSD-3-Clause
 */
package com.salesforce.functions.jvm.runtime.json;

import java.lang.annotation.Annotation;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

class Util {
  public static List getAnnotationsOnClassFieldsAndMethods(Class clazz) {
    ArrayList annotations = new ArrayList<>();

    for (Field field : clazz.getDeclaredFields()) {
      field.setAccessible(true);
      annotations.addAll(Arrays.asList(field.getAnnotations()));
    }

    for (Method method : clazz.getDeclaredMethods()) {
      method.setAccessible(true);
      annotations.addAll(Arrays.asList(method.getAnnotations()));
    }

    annotations.addAll(Arrays.asList(clazz.getAnnotations()));

    return annotations;
  }

  public static boolean typeContainsAnnotationFromPackage(Type type, Package annotationsPackage) {
    List> classesToCheck = new ArrayList<>();
    if (type instanceof Class) {
      classesToCheck.add((Class) type);
    }

    if (type instanceof ParameterizedType) {
      ParameterizedType parameterizedType = (ParameterizedType) type;

      if (parameterizedType.getRawType() instanceof Class) {
        classesToCheck.add((Class) parameterizedType.getRawType());
      }

      for (Type actualTypeArgument : parameterizedType.getActualTypeArguments()) {
        if (actualTypeArgument instanceof Class) {
          classesToCheck.add((Class) actualTypeArgument);
        }
      }
    }

    for (Class clazz : classesToCheck) {
      for (Annotation annotation : Util.getAnnotationsOnClassFieldsAndMethods(clazz)) {
        if (annotation.annotationType().getPackage().equals(annotationsPackage)) {
          return true;
        }
      }
    }

    return false;
  }

  private Util() {}
}