com.github.cloudyrock.mongock.MongockAnnotationProcessor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mongock-api Show documentation

There is a newer version: 5.0.1.BETA

package com.github.cloudyrock.mongock;

import io.changock.migration.api.ChangeSetItem;
import io.changock.migration.api.ChangockAnnotationProcessor;

import java.lang.annotation.Annotation;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;

public class MongockAnnotationProcessor extends ChangockAnnotationProcessor {
  @Override
  public Collection> getChangeLogAnnotationClass() {
    Collection> changeLogClasses = new ArrayList<>(Collections.singletonList(ChangeLog.class));
    changeLogClasses.addAll(super.getChangeLogAnnotationClass());
    return changeLogClasses;
  }

  @Override
  public boolean isChangeSetAnnotated(Method method) {
    return super.isChangeSetAnnotated(method) || method.isAnnotationPresent(ChangeSet.class);
  }

  @Override
  public String getChangeLogOrder(Class type) {
    try {
      return super.getChangeLogOrder(type);
    } catch (Exception ex) {
      return type.getAnnotation(ChangeLog.class).order();
    }
  }

  public ChangeSetItem getChangeSet(Method method) {
    try {
      return super.getChangeSet(method);
    } catch (Exception ex) {
      ChangeSet ann = method.getAnnotation(ChangeSet.class);
      return new ChangeSetItem(ann.id(), ann.author(), ann.order(), ann.runAlways(), ann.systemVersion(), ann.failFast(), method);
    }

  }
}