ConfigPackageAliasResolver.java

package com.seventh7.mybatis.alias;

import com.google.common.collect.Lists;

import com.intellij.openapi.project.Project;
import com.intellij.psi.PsiElement;
import com.intellij.util.Processor;
import com.seventh7.mybatis.dom.model.Package;
import com.seventh7.mybatis.util.MapperUtils;

import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import java.util.ArrayList;
import java.util.Collection;

/**
 * @author yanglin
 */
public class ConfigPackageAliasResolver extends PackageAliasResolver{

  public ConfigPackageAliasResolver(Project project) {
    super(project);
  }

  @NotNull @Override
  public Collection<String> getPackages(@Nullable PsiElement element) {
    final ArrayList<String> result = Lists.newArrayList();
    MapperUtils.processConfiguredPackage(project, new Processor<com.seventh7.mybatis.dom.model.Package>() {
      @Override
      public boolean process(Package pkg) {
        result.add(pkg.getName().getStringValue());
        return true;
      }
    });
    return result;
  }

}