Examples with INCOMING_PROP_PREFIX io.automatiko.engine.codegen.CodeGenConstants.INCOMING_PROP_PREFIX used on opensource projects

Example 1 with INCOMING_PROP_PREFIX

use of io.automatiko.engine.codegen.CodeGenConstants.INCOMING_PROP_PREFIX in project automatiko-engine by automatiko-io.

the class MessageConsumerGenerator method generate.

public String generate() {
    String sanitizedName = CodegenUtils.triggerSanitizedName(trigger, process.getVersion());
    String connector = CodegenUtils.getConnector(INCOMING_PROP_PREFIX + sanitizedName + ".connector", context, (String) trigger.getContext("connector"));
    if (connector != null && !OPERATOR_CONNECTOR.equals(connector)) {
        context.setApplicationProperty(INCOMING_PROP_PREFIX + sanitizedName + ".connector", connector);
        appendConnectorSpecificProperties(connector);
    }
    // operator consumer only supports starting message endpoints
    if (OPERATOR_CONNECTOR.equals(connector) && !trigger.isStart()) {
        return null;
    }
    CompilationUnit clazz = parse(this.getClass().getResourceAsStream(consumerTemplate(connector)));
    clazz.setPackageDeclaration(process.getPackageName());
    clazz.addImport(modelfqcn);
    // add functions so they can be easily accessed in message consumer classes
    clazz.addImport(new ImportDeclaration(BaseFunctions.class.getCanonicalName(), true, true));
    context.getBuildContext().classThatImplement(Functions.class.getCanonicalName()).forEach(c -> clazz.addImport(new ImportDeclaration(c, true, true)));
    ClassOrInterfaceDeclaration template = clazz.findFirst(ClassOrInterfaceDeclaration.class).get();
    template.setName(resourceClazzName);
    template.findAll(ClassOrInterfaceType.class).forEach(cls -> interpolateTypes(cls, dataClazzName));
    template.findAll(MethodDeclaration.class).stream().filter(md -> md.getNameAsString().equals("configure")).forEach(md -> md.addAnnotation("javax.annotation.PostConstruct"));
    template.findAll(MethodDeclaration.class).stream().filter(md -> md.getNameAsString().equals("consume") || md.getNameAsString().equals("cleanup") || md.getNameAsString().equals("reconcile")).forEach(md -> {
        md.findAll(StringLiteralExpr.class).forEach(str -> str.setString(str.asString().replace("$Trigger$", trigger.getName())));
        md.findAll(ClassOrInterfaceType.class).forEach(t -> t.setName(t.getNameAsString().replace("$DataEventType$", messageDataEventClassName)));
        md.findAll(ClassOrInterfaceType.class).forEach(t -> t.setName(t.getNameAsString().replace("$DataType$", trigger.getDataType())));
        md.setType(md.getTypeAsString().replace("$DataType$", trigger.getDataType()));
    });
    template.findAll(MethodDeclaration.class).stream().filter(md -> md.getNameAsString().equals("convert")).forEach(md -> {
        md.setType(md.getTypeAsString().replace("$DataType$", trigger.getDataType()));
        md.findAll(CastExpr.class).forEach(c -> c.setType(c.getTypeAsString().replace("$DataType$", trigger.getDataType())));
        md.findAll(ClassOrInterfaceType.class).forEach(t -> t.setName(t.getNameAsString().replace("$DataType$", trigger.getDataType())));
    });
    if (trigger.getModelRef().startsWith("#")) {
        template.findAll(MethodCallExpr.class).stream().filter(m -> m.getNameAsString().endsWith("$ModelRef$")).forEach(m -> {
            m.getParentNode().ifPresent(p -> p.removeForced());
        });
    } else {
        template.findAll(MethodCallExpr.class).forEach(this::interpolateStrings);
    }
    if (useInjection()) {
        if (!OPERATOR_CONNECTOR.equals(connector)) {
            annotator.withApplicationComponent(template);
        }
        template.findAll(FieldDeclaration.class, fd -> isProcessField(fd)).forEach(fd -> annotator.withNamedInjection(fd, processId + version));
        template.findAll(FieldDeclaration.class, fd -> isApplicationField(fd)).forEach(fd -> annotator.withInjection(fd));
        template.findAll(FieldDeclaration.class, fd -> fd.getVariables().get(0).getNameAsString().equals("converter")).forEach(fd -> {
            annotator.withInjection(fd);
            fd.getVariable(0).setType(fd.getVariable(0).getTypeAsString().replace("$DataType$", trigger.getDataType()));
        });
        template.findAll(FieldDeclaration.class, fd -> fd.getVariable(0).getNameAsString().equals("useCloudEvents")).forEach(fd -> annotator.withConfigInjection(fd, "quarkus.automatiko.messaging.as-cloudevents"));
        template.findAll(MethodDeclaration.class).stream().filter(md -> md.getNameAsString().equals("consume")).forEach(md -> {
            if (persistence) {
                annotator.withBlocking(md);
            }
            annotator.withIncomingMessage(md, sanitizedName);
            if (context.getBuildContext().hasClassAvailable("org.eclipse.microprofile.opentracing.Traced")) {
                md.addAnnotation("org.eclipse.microprofile.opentracing.Traced");
            }
        });
    } else {
        template.findAll(FieldDeclaration.class, fd -> isProcessField(fd)).forEach(fd -> initializeProcessField(fd, template));
        template.findAll(FieldDeclaration.class, fd -> isApplicationField(fd)).forEach(fd -> initializeApplicationField(fd, template));
        template.findAll(FieldDeclaration.class, fd -> fd.getVariables().get(0).getNameAsString().equals("converter")).forEach(fd -> {
            fd.getVariable(0).setType(fd.getVariable(0).getTypeAsString().replace("$DataType$", trigger.getDataType()));
        });
    }
    BlockStmt filterExpressionBody = new BlockStmt();
    if (trigger.getContext("filterExpression") != null) {
        filterExpressionBody.addStatement(new ReturnStmt(new NameExpr((String) trigger.getContext("filterExpression"))));
    } else {
        filterExpressionBody.addStatement(new ReturnStmt(new BooleanLiteralExpr(true)));
    }
    BlockStmt body = new BlockStmt();
    if (trigger.getCorrelation() != null) {
        body.addStatement(new ReturnStmt(new StringLiteralExpr(trigger.getCorrelation())));
    } else if (trigger.getCorrelationExpression() != null) {
        body.addStatement(new ReturnStmt(new NameExpr(trigger.getCorrelationExpression())));
    } else {
        body.addStatement(new ReturnStmt(new NullLiteralExpr()));
    }
    boolean cloudEvents = context.getBuildContext().config().messaging().asCloudevents() || Boolean.parseBoolean(context.getApplicationProperty("quarkus.automatiko.messaging.as-cloudevents").orElse("false"));
    if (cloudEvents) {
        template.findAll(MethodDeclaration.class).stream().filter(md -> md.getNameAsString().equals("correlationEvent")).forEach(md -> {
            md.setBody(body);
            md.getParameters().get(0).setType(messageDataEventClassName);
        });
        template.findAll(MethodDeclaration.class).stream().filter(md -> md.getNameAsString().equals("acceptedEvent")).forEach(md -> {
            md.setBody(filterExpressionBody);
            md.getParameters().get(0).setType(messageDataEventClassName);
        });
    } else {
        template.findAll(MethodDeclaration.class).stream().filter(md -> md.getNameAsString().equals("correlationPayload")).forEach(md -> {
            md.setBody(body);
            md.getParameters().get(0).setType(trigger.getDataType());
        });
        template.findAll(MethodDeclaration.class).stream().filter(md -> md.getNameAsString().equals("acceptedPayload")).forEach(md -> {
            md.setBody(filterExpressionBody);
            md.getParameters().get(0).setType(trigger.getDataType());
        });
    }
    template.addMember(new MethodDeclaration().setName("canStartInstance").setType(Boolean.class).setModifiers(Keyword.PROTECTED).setBody(new BlockStmt().addStatement(new ReturnStmt(new BooleanLiteralExpr(trigger.isStart())))));
    // add connector and message name as static fields of the class
    FieldDeclaration connectorField = new FieldDeclaration().setStatic(true).setFinal(true).addVariable(new VariableDeclarator(new ClassOrInterfaceType(null, "String"), "CONNECTOR", new StringLiteralExpr(connector)));
    template.addMember(connectorField);
    FieldDeclaration messageNameField = new FieldDeclaration().setStatic(true).setFinal(true).addVariable(new VariableDeclarator(new ClassOrInterfaceType(null, "String"), "MESSAGE", new StringLiteralExpr(trigger.getName())));
    template.addMember(messageNameField);
    if (OPERATOR_CONNECTOR.equals(connector)) {
        // add event filter
        String eventFilterClass = (String) trigger.getContext("eventFilter");
        if (eventFilterClass != null && !eventFilterClass.isEmpty()) {
            AnnotationExpr controllerAnnotation = template.getAnnotationByName("Controller").get();
            ((NormalAnnotationExpr) controllerAnnotation).addPair("eventFilters", new NameExpr(eventFilterClass + ".class"));
        }
    }
    template.getMembers().sort(new BodyDeclarationComparator());
    ImportsOrganizer.organize(clazz);
    return clazz.toString().replaceAll("\\$DataType\\$", trigger.getDataType()).replaceAll("\\$DataEventType\\$", messageDataEventClassName).replaceAll("\\$ProcessId\\$", processId + version).replaceAll("\\$GenControllerParam\\$", (String) trigger.getContext("generationAware", "true")).replaceAll("\\$ControllerParam\\$", "{" + Stream.of(namespaces.split(",")).filter(s -> !s.trim().isEmpty()).map(s -> "\"" + s.trim() + "\"").collect(Collectors.joining(",")) + "}");
}