protected KieSession createKnowledgeSession(String processFile) throws Exception {
    KieServices ks = KieServices.Factory.get();
    KieRepository kr = ks.getRepository();
    KieFileSystem kfs = ks.newKieFileSystem();
    if (processFile != null) {
      Resource process = ResourceFactory.newClassPathResource(processFile);
      kfs.write(process);
    }

    KieBuilder kb = ks.newKieBuilder(kfs);
    kb.buildAll();

    if (kb.getResults().hasMessages(Level.ERROR)) {
      throw new RuntimeException("Build Errors:\n" + kb.getResults().toString());
    }

    KieContainer kContainer = ks.newKieContainer(kr.getDefaultReleaseId());
    KieBase kbase = kContainer.getKieBase();

    Environment env = EnvironmentFactory.newEnvironment();

    Properties defaultProps = new Properties();
    defaultProps.setProperty(
        "drools.processSignalManagerFactory", DefaultSignalManagerFactory.class.getName());
    defaultProps.setProperty(
        "drools.processInstanceManagerFactory",
        DefaultProcessInstanceManagerFactory.class.getName());
    SessionConfiguration conf = new SessionConfiguration(defaultProps);

    KieSession ksession = (StatefulKnowledgeSession) kbase.newKieSession(conf, env);
    return ksession;
  }
Exemplo n.º 2
0
  /**
   * Creating a KieBase from the workflow GAV specified in the config.
   *
   * @return KieBase for package SRAMPPackage
   * @throws SrampClientException
   * @throws SrampAtomException
   */
  public KieContainer getKieContainer(ReleaseId releaseId)
      throws SrampClientException, SrampAtomException {
    KieServices ks = KieServices.Factory.get();
    KieRepository repo = ks.getRepository();
    SrampAtomApiClient client = SrampAtomApiClientFactory.createAtomApiClient();

    Governance governance = new Governance();
    QueryResultSet results =
        client
            .buildQuery(SRAMP_KIE_JAR_QUERY)
            .parameter(governance.getGovernanceWorkflowGroup())
            .parameter(governance.getGovernanceWorkflowName())
            .parameter(governance.getGovernanceWorkflowVersion())
            .count(1)
            .query();
    if (results.size() > 0) {
      ArtifactSummary artifactSummery = results.get(0);
      InputStream is = client.getArtifactContent(artifactSummery);
      KieModule kModule = repo.addKieModule(ks.getResources().newInputStreamResource(is));
      logger.info(
          Messages.i18n.format(
              "KieSrampUtil.CreatingKieContainer", artifactSummery)); // $NON-NLS-1$
      KieContainer kContainer = ks.newKieContainer(kModule.getReleaseId());
      // Creating the KieBase for the SRAMPPackage
      logger.info(
          Messages.i18n.format(
              "KieSrampUtil.FindKieBase",
              governance.getGovernanceWorkflowPackage())); // $NON-NLS-1$
      return kContainer;
    } else {
      return null;
    }
  }
Exemplo n.º 3
0
  @Test
  public void testSignalStartDynamic() throws Exception {

    KieBase kbase = createKnowledgeBase("BPMN2-SignalStart.bpmn2");
    ksession = createKnowledgeSession(kbase);
    // create KieContainer after session was created to make sure no runtime data
    // will be used during serialization (deep clone)
    KieServices ks = KieServices.Factory.get();
    KieRepository kr = ks.getRepository();
    KieContainer kContainer = ks.newKieContainer(kr.getDefaultReleaseId());
    kContainer.getKieBase();

    final List<Long> list = new ArrayList<Long>();
    ksession.addEventListener(
        new DefaultProcessEventListener() {
          public void afterProcessStarted(ProcessStartedEvent event) {
            logger.info("{}", event.getProcessInstance().getId());
            list.add(event.getProcessInstance().getId());
          }
        });
    ksession.signalEvent("MySignal", "NewValue");
    Thread.sleep(500);
    assertEquals(1, getNumberOfProcessInstances("Minimal"));
    // now remove the process from kbase to make sure runtime based listeners are removed from
    // signal manager
    kbase.removeProcess("Minimal");
    ksession.signalEvent("MySignal", "NewValue");
    Thread.sleep(500);
    // must be still one as the process was removed
    assertEquals(1, getNumberOfProcessInstances("Minimal"));
  }
  public void go(PrintStream out) {
    KieServices ks = KieServices.Factory.get();
    KieRepository kr = ks.getRepository();

    KieModule kModule =
        kr.addKieModule(ks.getResources().newFileSystemResource(getFile("named-kiesession")));

    KieContainer kContainer = ks.newKieContainer(kModule.getReleaseId());

    KieSession kSession = kContainer.newKieSession("ksession1");
    kSession.setGlobal("out", out);

    Object msg1 = createMessage(kContainer, "Dave", "Hello, HAL. Do you read me, HAL?");
    kSession.insert(msg1);
    kSession.fireAllRules();
  }
Exemplo n.º 5
0
  /**
   * This method loads the rule base associated with the Drools event processor.
   *
   * @return The knowledge base
   */
  private KieBase loadRuleBase() {
    String droolsRuleBase = getRuleName() + ".drl";

    try {
      KieServices ks = KieServices.Factory.get();
      KieRepository kr = ks.getRepository();

      KieModuleModel kmm = ks.newKieModuleModel();
      KieBaseModel kbm =
          kmm.newKieBaseModel(getRuleName())
              .setEqualsBehavior(EqualityBehaviorOption.EQUALITY)
              .setEventProcessingMode(EventProcessingOption.STREAM);
      kbm.setDefault(true);

      KieFileSystem kfs = ks.newKieFileSystem();
      kfs.write(
          "src/main/resources/" + kbm.getName() + "/rule1.drl",
          ks.getResources().newClassPathResource(droolsRuleBase));
      kfs.writeKModuleXML(kmm.toXML());

      KieBuilder kb = ks.newKieBuilder(kfs);
      kb.buildAll();

      KieContainer container = ks.newKieContainer(kr.getDefaultReleaseId());
      KieBase kbase = container.getKieBase();
      // TODO: hack it for now, this attribute should be supported in the following drools6 release.
      System.setProperty("kie.mbean", "enabled");
      return kbase;

    } catch (Throwable e) {
      LOG.log(
          Level.SEVERE,
          MessageFormat.format(
              java.util.PropertyResourceBundle.getBundle("ep-drools.Messages")
                  .getString("EP-DROOLS-1"),
              droolsRuleBase,
              getRuleName()),
          e);
    }

    return (null);
  }
Exemplo n.º 6
0
  public Results updateDependencyToVersion(ReleaseId currentReleaseId, ReleaseId newReleaseId) {
    checkNotClasspathKieProject();

    ReleaseId installedReleaseId = getReleaseId();
    InternalKieModule currentKM;
    if (currentReleaseId.getGroupId().equals(installedReleaseId.getGroupId())
        && currentReleaseId.getArtifactId().equals(installedReleaseId.getArtifactId())) {
      // upgrading the kProject itself: taking the kmodule from there
      currentKM = ((KieModuleKieProject) kProject).getInternalKieModule();
    } else {
      // upgrading a transitive dependency: taking the kmodule from the krepo
      // if the new and the current release are equal (a snapshot) check if there is an older
      // version with the same releaseId
      currentKM =
          currentReleaseId.equals(newReleaseId)
              ? (InternalKieModule) ((KieRepositoryImpl) kr).getOldKieModule(currentReleaseId)
              : (InternalKieModule) kr.getKieModule(currentReleaseId);
    }

    return update(currentKM, newReleaseId);
  }
Exemplo n.º 7
0
  public void discoverKieModules() {
    String[] configFiles = {
      KieModuleModelImpl.KMODULE_JAR_PATH, KieModuleModelImpl.KMODULE_SPRING_JAR_PATH
    };
    for (String configFile : configFiles) {
      final Enumeration<URL> e;
      try {
        e = classLoader.getResources(configFile);
      } catch (IOException exc) {
        log.error("Unable to find and build index of " + configFile + "." + exc.getMessage());
        return;
      }

      // Map of kmodule urls
      while (e.hasMoreElements()) {
        URL url = e.nextElement();
        notifyKieModuleFound(url);
        try {
          InternalKieModule kModule = fetchKModule(url);

          if (kModule != null) {
            ReleaseId releaseId = kModule.getReleaseId();
            kieModules.put(releaseId, kModule);

            log.debug("Discovered classpath module " + releaseId.toExternalForm());

            kieRepository.addKieModule(kModule);
          }

        } catch (Exception exc) {
          log.error(
              "Unable to build index of kmodule.xml url="
                  + url.toExternalForm()
                  + "\n"
                  + exc.getMessage());
        }
      }
    }
  }
Exemplo n.º 8
0
  private Results update(final InternalKieModule currentKM, final ReleaseId newReleaseId) {
    final InternalKieModule newKM = (InternalKieModule) kr.getKieModule(newReleaseId);
    ChangeSetBuilder csb = new ChangeSetBuilder();
    final KieJarChangeSet cs = csb.build(currentKM, newKM);

    final List<String> modifiedClasses = getModifiedClasses(cs);
    final List<String> dslFiles = getUnchangedDslFiles(newKM, cs);

    ((KieModuleKieProject) kProject).updateToModule(newKM);

    final ResultsImpl results = new ResultsImpl();

    List<String> kbasesToRemove = new ArrayList<String>();
    for (Entry<String, KieBase> kBaseEntry : kBases.entrySet()) {
      String kbaseName = kBaseEntry.getKey();
      final KieBaseModel kieBaseModel = kProject.getKieBaseModel(kbaseName);
      // if a kbase no longer exists, just remove it from the cache
      if (kieBaseModel == null) {
        // have to save for later removal to avoid iteration errors
        kbasesToRemove.add(kbaseName);
      } else {
        final InternalKnowledgeBase kBase = (InternalKnowledgeBase) kBaseEntry.getValue();
        boolean locked = kBase.tryLock();
        if (locked) {
          try {
            updateKBase(
                kBase,
                currentKM,
                newReleaseId,
                newKM,
                cs,
                modifiedClasses,
                dslFiles,
                results,
                kieBaseModel);
          } finally {
            kBase.unlock();
          }
        } else {
          kBase.enqueueModification(
              new Runnable() {
                @Override
                public void run() {
                  updateKBase(
                      kBase,
                      currentKM,
                      newReleaseId,
                      newKM,
                      cs,
                      modifiedClasses,
                      dslFiles,
                      results,
                      kieBaseModel);
                }
              });
        }
      }
    }

    for (String kbaseToRemove : kbasesToRemove) {
      kBases.remove(kbaseToRemove);
    }

    for (Iterator<Entry<String, KieSession>> it = this.kSessions.entrySet().iterator();
        it.hasNext(); ) {
      Entry<String, KieSession> ksession = it.next();
      if (kProject.getKieSessionModel(ksession.getKey()) == null) {
        // remove sessions that no longer exist
        it.remove();
      }
    }

    for (Iterator<Entry<String, StatelessKieSession>> it =
            this.statelessKSessions.entrySet().iterator();
        it.hasNext(); ) {
      Entry<String, StatelessKieSession> ksession = it.next();
      if (kProject.getKieSessionModel(ksession.getKey()) == null) {
        // remove sessions that no longer exist
        it.remove();
      }
    }

    return results;
  }