001/* 002 * Copyright 2012-2016 the original author or authors. 003 * 004 * Licensed under the Apache License, Version 2.0 (the "License"); 005 * you may not use this file except in compliance with the License. 006 * You may obtain a copy of the License at 007 * 008 * http://www.apache.org/licenses/LICENSE-2.0 009 * 010 * Unless required by applicable law or agreed to in writing, software 011 * distributed under the License is distributed on an "AS IS" BASIS, 012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 013 * See the License for the specific language governing permissions and 014 * limitations under the License. 015 */ 016 017package org.springframework.boot.cli.compiler.autoconfigure; 018 019import org.codehaus.groovy.ast.ClassNode; 020import org.codehaus.groovy.control.customizers.ImportCustomizer; 021 022import org.springframework.boot.cli.compiler.AstUtils; 023import org.springframework.boot.cli.compiler.CompilerAutoConfiguration; 024import org.springframework.boot.cli.compiler.DependencyCustomizer; 025 026/** 027 * {@link CompilerAutoConfiguration} for Spring Integration. 028 * 029 * @author Dave Syer 030 * @author Artem Bilan 031 */ 032public class SpringIntegrationCompilerAutoConfiguration 033 extends CompilerAutoConfiguration { 034 035 @Override 036 public boolean matches(ClassNode classNode) { 037 return AstUtils.hasAtLeastOneAnnotation(classNode, "EnableIntegration") 038 || AstUtils.hasAtLeastOneAnnotation(classNode, "MessageEndpoint"); 039 } 040 041 @Override 042 public void applyDependencies(DependencyCustomizer dependencies) { 043 dependencies 044 .ifAnyMissingClasses( 045 "org.springframework.integration.config.EnableIntegration") 046 .add("spring-boot-starter-integration"); 047 } 048 049 @Override 050 public void applyImports(ImportCustomizer imports) { 051 imports.addImports("org.springframework.messaging.Message", 052 "org.springframework.messaging.MessageChannel", 053 "org.springframework.messaging.PollableChannel", 054 "org.springframework.messaging.SubscribableChannel", 055 "org.springframework.messaging.MessageHeaders", 056 "org.springframework.integration.support.MessageBuilder", 057 "org.springframework.integration.channel.DirectChannel", 058 "org.springframework.integration.channel.QueueChannel", 059 "org.springframework.integration.channel.ExecutorChannel", 060 "org.springframework.integration.core.MessagingTemplate", 061 "org.springframework.integration.config.EnableIntegration"); 062 imports.addStarImports("org.springframework.integration.annotation"); 063 } 064 065}