infra.beans.factory.aot.BeanRegistrationsAotProcessor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of today-beans Show documentation
Infra Beans
The newest version!
/*
 * Copyright 2017 - 2024 the original author or authors.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see [https://www.gnu.org/licenses/]
 */

package infra.beans.factory.aot;

import java.util.ArrayList;

import infra.beans.factory.config.ConfigurableBeanFactory;
import infra.beans.factory.support.RegisteredBean;
import infra.lang.Nullable;

/**
 * {@link BeanFactoryInitializationAotProcessor} that contributes code to
 * register beans.
 *
 * @author Phillip Webb
 * @author Sebastien Deleuze
 * @author Stephane Nicoll
 * @author Brian Clozel
 * @author Harry Yang
 * @since 4.0
 */
class BeanRegistrationsAotProcessor implements BeanFactoryInitializationAotProcessor {

  @Override
  @Nullable
  public BeanRegistrationsAotContribution processAheadOfTime(ConfigurableBeanFactory beanFactory) {
    var beanDefinitionMethodGeneratorFactory = new BeanDefinitionMethodGeneratorFactory(beanFactory);
    var registrations = new ArrayList();

    for (String beanName : beanFactory.getBeanDefinitionNames()) {
      RegisteredBean registeredBean = RegisteredBean.of(beanFactory, beanName);
      BeanDefinitionMethodGenerator beanDefinitionMethodGenerator =
              beanDefinitionMethodGeneratorFactory.getBeanDefinitionMethodGenerator(registeredBean);
      if (beanDefinitionMethodGenerator != null) {
        registrations.add(new BeanRegistrationsAotContribution.Registration(registeredBean, beanDefinitionMethodGenerator, beanFactory.getAliases(beanName)));
      }
    }

    if (registrations.isEmpty()) {
      return null;
    }
    return new BeanRegistrationsAotContribution(registrations);
  }

}