// VisibleForTesting
    protected boolean runGenerator(RuleGenerateWith generatorRule, Set<String> reboundTypeNames)
        throws UnableToCompleteException {
      boolean fixedPoint = true;
      StandardGeneratorContext generatorContext = getGeneratorContext();
      removePreviouslyReboundCombinations(generatorRule.getName(), reboundTypeNames);
      reboundTypeNames.removeAll(previouslyReboundTypeNames);

      for (String reboundTypeName : reboundTypeNames) {
        if (badRebindCombinations.contains(generatorRule.getName() + "-" + reboundTypeName)) {
          continue;
        }
        generatorNamesByPreviouslyReboundTypeName.put(reboundTypeName, generatorRule.getName());
        reboundTypeName = reboundTypeName.replace("$", ".");
        generatorRule.generate(logger, module.getProperties(), generatorContext, reboundTypeName);

        if (generatorContext.isDirty()) {
          fixedPoint = false;
          previouslyReboundTypeNames.add(reboundTypeName);
          // Ensure that cascading generations rerun properly.
          for (String generatedTypeName : generatorContext.getGeneratedUnitMap().keySet()) {
            generatorNamesByPreviouslyReboundTypeName.removeAll(generatedTypeName);
          }
          generatorContext.finish(logger);
        } else {
          badRebindCombinations.add(generatorRule.getName() + "-" + reboundTypeName);
        }
      }

      return fixedPoint;
    }
예제 #2
0
  public JExpression doRebind(String clsName, ReflectionGeneratorContext params)
      throws UnableToCompleteException {
    // generate
    params.getLogger().log(Type.INFO, "Binding magic class for " + clsName);
    // JType type = params.getClazz().getRefType();
    JDeclaredType type =
        params.getAst().searchForTypeBySource(params.getClazz().getRefType().getName());

    StandardGeneratorContext ctx = params.getGeneratorContext();
    Class<? extends Generator> generator = MagicClassGenerator.class;

    String result =
        ctx.runGenerator(params.getLogger(), generator, SourceUtil.toSourceName(type.getName()));
    ctx.finish(params.getLogger());

    params.getLogger().log(Type.INFO, "Generated Class Enhancer: " + result);
    JDeclaredType success = params.getAst().searchForTypeBySource(result);

    // Okay, we've generated the correct magic class subtype;
    // Now pull off its static accessor method to grab our generated class.

    for (JMethod method : success.getMethods()) {
      if (method.isStatic() && method.getName().equals("enhanceClass")) {
        JMethodCall call = new JMethodCall(method.getSourceInfo(), null, method);
        call.addArg(params.getClazz().makeStatement().getExpr());
        return call;
      }
    }
    params.getLogger().log(Type.ERROR, "Unable to load " + result + ".enhanceClass()");
    throw new UnableToCompleteException();
  }
 // VisibleForTesting
 void buildRuntimeRebindRegistrator(Set<String> allRootTypes) throws UnableToCompleteException {
   RuntimeRebindRegistratorGenerator runtimeRebindRegistratorGenerator =
       new RuntimeRebindRegistratorGenerator();
   StandardGeneratorContext generatorContext = getGeneratorContext();
   String runtimeRebindRegistratorTypeName =
       runtimeRebindRegistratorGenerator.generate(logger, generatorContext, module.getName());
   // Ensures that unification traverses and keeps the class.
   allRootTypes.add(runtimeRebindRegistratorTypeName);
   // Ensures that JProgram knows to index this class's methods so that later bootstrap
   // construction code is able to locate the FooRuntimeRebindRegistrator.register() function.
   jprogram.addIndexedTypeName(runtimeRebindRegistratorTypeName);
   jprogram.setRuntimeRebindRegistratorTypeName(runtimeRebindRegistratorTypeName);
   generatorContext.finish(logger);
 }
 // VisibleForTesting
 void buildPropertyProviderRegistrator(
     Set<String> allRootTypes,
     SortedSet<BindingProperty> bindingProperties,
     SortedSet<ConfigurationProperty> configurationProperties)
     throws UnableToCompleteException {
   PropertyProviderRegistratorGenerator propertyProviderRegistratorGenerator =
       new PropertyProviderRegistratorGenerator(bindingProperties, configurationProperties);
   StandardGeneratorContext generatorContext = getGeneratorContext();
   String propertyProviderRegistratorTypeName =
       propertyProviderRegistratorGenerator.generate(logger, generatorContext, module.getName());
   // Ensures that unification traverses and keeps the class.
   allRootTypes.add(propertyProviderRegistratorTypeName);
   // Ensures that JProgram knows to index this class's methods so that later bootstrap
   // construction code is able to locate the FooPropertyProviderRegistrator.register() function.
   jprogram.addIndexedTypeName(propertyProviderRegistratorTypeName);
   jprogram.setPropertyProviderRegistratorTypeSourceName(propertyProviderRegistratorTypeName);
   generatorContext.finish(logger);
 }