Examples with KieContainerInstance org.kie.server.services.api.KieContainerInstance used on opensource projects

Example 6 with KieContainerInstance

use of org.kie.server.services.api.KieContainerInstance in project droolsjbpm-integration by kiegroup.

the class MarshallerHelper method marshal.

public String marshal(String containerId, String marshallingFormat, Object entity, ContainerLocator locator) {
    MarshallingFormat format = getFormat(marshallingFormat);
    if (format == null) {
        throw new IllegalArgumentException("Unknown marshalling format " + marshallingFormat);
    }
    KieContainerInstance containerInstance = registry.getContainer(containerId, locator);
    if (containerInstance == null) {
        throw new IllegalArgumentException("No container found for id " + containerId + " .");
    }
    Marshaller marshaller = containerInstance.getMarshaller(format);
    if (marshaller == null) {
        throw new IllegalArgumentException("No marshaller found for format " + format);
    }
    return marshaller.marshall(entity, MarshallingFormat.buildParameters(marshallingFormat));
}

Example 7 with KieContainerInstance

use of org.kie.server.services.api.KieContainerInstance in project droolsjbpm-integration by kiegroup.

the class KieServerMDB method onMessage.

public void onMessage(Message message) {
    JMSConnection connect = null;
    try {
        String username = null;
        String password = null;
        try {
            username = message.getStringProperty(USER_PROPERTY_NAME);
            password = message.getStringProperty(PASSWRD_PROPERTY_NAME);
        } catch (JMSException jmse) {
            logger.warn("Unable to retrieve user name and/or password, from message");
        }
        if (username != null && password != null) {
            JMSSecurityAdapter.login(username, password);
        } else {
            logger.warn("Unable to login to JMSSecurityAdapter, user name and/or password missing");
        }
        KieContainerCommandService executor = null;
        // 0. Get msg correlation id (for response)
        String msgCorrId = null;
        try {
            msgCorrId = message.getJMSCorrelationID();
        } catch (JMSException jmse) {
            String errMsg = "Unable to retrieve JMS correlation id from message! " + ID_NECESSARY;
            throw new JMSRuntimeException(errMsg, jmse);
        }
        // for backward compatibility default to KieServer
        String targetCapability = getStringProperty(message, TARGET_CAPABILITY_PROPERTY_NAME, "KieServer");
        String containerId = getStringProperty(message, CONTAINER_ID_PROPERTY_NAME, null);
        String conversationId = getStringProperty(message, CONVERSATION_ID_PROPERTY_NAME, null);
        int interactionPattern = getIntProperty(message, INTERACTION_PATTERN_PROPERTY_NAME, REQUEST_REPLY_PATTERN);
        // 1. get marshalling info
        MarshallingFormat format = null;
        String classType = null;
        try {
            classType = message.getStringProperty(CLASS_TYPE_PROPERTY_NAME);
            if (!message.propertyExists(SERIALIZATION_FORMAT_PROPERTY_NAME)) {
                format = MarshallingFormat.JAXB;
            } else {
                int intFormat = message.getIntProperty(SERIALIZATION_FORMAT_PROPERTY_NAME);
                logger.debug("Serialization format (int) is {}", intFormat);
                format = MarshallingFormat.fromId(intFormat);
                logger.debug("Serialization format is {}", format);
                if (format == null) {
                    String errMsg = "Unsupported marshalling format '" + intFormat + "' from message " + msgCorrId + ".";
                    throw new JMSRuntimeException(errMsg);
                }
            }
        } catch (JMSException jmse) {
            String errMsg = "Unable to retrieve property '" + SERIALIZATION_FORMAT_PROPERTY_NAME + "' from message " + msgCorrId + ".";
            throw new JMSRuntimeException(errMsg, jmse);
        }
        // 2. get marshaller
        Marshaller marshaller = getMarshaller(containerId, format);
        logger.debug("Selected marshaller is {}", marshaller);
        // 3. deserialize request
        CommandScript script = unmarshallRequest(message, msgCorrId, marshaller, format);
        logger.debug("Target capability is {}", targetCapability);
        for (KieServerExtension extension : kieServer.getServerExtensions()) {
            KieContainerCommandService tmp = extension.getAppComponents(KieContainerCommandService.class);
            if (tmp != null && extension.getImplementedCapability().equalsIgnoreCase(targetCapability)) {
                executor = tmp;
                logger.debug("Extension {} returned command executor {} with capability {}", extension, executor, extension.getImplementedCapability());
                break;
            }
        }
        if (executor == null) {
            throw new IllegalStateException("No executor found for script execution");
        }
        // 4. process request
        ServiceResponsesList response = executor.executeScript(script, format, classType);
        if (interactionPattern < UPPER_LIMIT_REPLY_INTERACTION_PATTERNS) {
            connect = startConnectionAndSession();
            logger.debug("Response message is about to be sent according to selected interaction pattern {}", interactionPattern);
            // 5. serialize response
            Message msg = marshallResponse(connect.getSession(), msgCorrId, format, marshaller, response);
            // set conversation id for routing
            if (containerId != null && (conversationId == null || conversationId.trim().isEmpty())) {
                try {
                    KieContainerInstance containerInstance = kieServer.getServerRegistry().getContainer(containerId);
                    if (containerInstance != null) {
                        ReleaseId releaseId = containerInstance.getResource().getResolvedReleaseId();
                        if (releaseId == null) {
                            releaseId = containerInstance.getResource().getReleaseId();
                        }
                        conversationId = ConversationId.from(KieServerEnvironment.getServerId(), containerId, releaseId).toString();
                    }
                } catch (Exception e) {
                    logger.warn("Unable to build conversation id due to {}", e.getMessage(), e);
                }
            }
            try {
                if (conversationId != null) {
                    msg.setStringProperty(CONVERSATION_ID_PROPERTY_NAME, conversationId);
                }
            } catch (JMSException e) {
                logger.debug("Unable to set conversation id on response message due to {}", e.getMessage());
            }
            // 6. send response
            sendResponse(connect.getSession(), msgCorrId, format, msg);
        } else {
            logger.debug("Response message is skipped according to selected interaction pattern {}", FIRE_AND_FORGET_PATTERN);
        }
    } finally {
        if (connect != null) {
            // Only attempt to close the connection/session if they were actually created
            try {
                closeConnectionAndSession(connect);
            } catch (JMSRuntimeException runtimeException) {
                logger.error("Error while attempting to close connection/session", runtimeException);
            } finally {
                JMSSecurityAdapter.logout();
            }
        } else {
            JMSSecurityAdapter.logout();
        }
    }
}

Example 8 with KieContainerInstance

use of org.kie.server.services.api.KieContainerInstance in project droolsjbpm-integration by kiegroup.

the class CustomResource method insertFireReturn.

@POST
@Path("/{ksessionId}")
@Consumes({ MediaType.APPLICATION_XML, MediaType.APPLICATION_JSON })
@Produces({ MediaType.APPLICATION_XML, MediaType.APPLICATION_JSON })
public Response insertFireReturn(@Context HttpHeaders headers, @PathParam("id") String id, @PathParam("ksessionId") String ksessionId, String cmdPayload) {
    Variant v = RestUtils.getVariant(headers);
    String contentType = RestUtils.getContentType(headers);
    MarshallingFormat format = MarshallingFormat.fromType(contentType);
    try {
        KieContainerInstance kci = registry.getContainer(id, ContainerLocatorProvider.get().getLocator());
        Marshaller marshaller = kci.getMarshaller(format);
        List<?> listOfFacts = marshaller.unmarshall(cmdPayload, List.class);
        List<Command<?>> commands = new ArrayList<Command<?>>();
        BatchExecutionCommand executionCommand = commandsFactory.newBatchExecution(commands, ksessionId);
        for (Object fact : listOfFacts) {
            commands.add(commandsFactory.newInsert(fact, fact.toString()));
        }
        commands.add(commandsFactory.newFireAllRules());
        commands.add(commandsFactory.newGetObjects());
        ExecutionResults results = rulesExecutionService.call(kci, executionCommand);
        String result = marshaller.marshall(results);
        logger.debug("Returning OK response with content '{}'", result);
        return RestUtils.createResponse(result, v, Response.Status.OK);
    } catch (Exception e) {
        // in case marshalling failed return the call container response to keep backward compatibility
        String response = "Execution failed with error : " + e.getMessage();
        logger.debug("Returning Failure response with content '{}'", response);
        return RestUtils.createResponse(response, v, Response.Status.INTERNAL_SERVER_ERROR);
    }
}

Example 9 with KieContainerInstance

use of org.kie.server.services.api.KieContainerInstance in project kiegroup-examples by tkobayas.

the class MyKieServerWarmupEventListenerVariant method afterContainerStarted.

@Override
public void afterContainerStarted(KieServer kieServer, KieContainerInstance containerInstance) {
    KieContainerResource resource = containerInstance.getResource();
    Optional<KieServerConfigItem> optItem = resource.getConfigItems().stream().filter(item -> item.getName().equals(KIE_WARMUP_LISTENER_STRATEGY)).findFirst();
    if (!optItem.isPresent()) {
        LOGGER.info("{} is not enabled : ", KIE_WARMUP_LISTENER_STRATEGY);
        return;
    }
    String strategy = optItem.get().getValue();
    LOGGER.info("start warmup : " + strategy);
    long start = System.currentTimeMillis();
    switch(strategy) {
        case "rest-full":
            restWarmupFull(containerInstance);
            break;
        case "rest-fire-only":
            restWarmupFireOnly(containerInstance);
            break;
        case "internal-full":
            internalWarmupFull(containerInstance);
            break;
        case "internal-fire-only":
            internalWarmupFireOnly(containerInstance);
            break;
        default:
            break;
    }
    LOGGER.info("afterContainerStarted : elapsed time = {}ms", System.currentTimeMillis() - start);
}