コード例 #1
0
  protected void initKsession(
      Integer sessionId,
      KnowledgeBase kbase,
      KnowledgeSessionConfiguration conf,
      PersistenceContext persistenceContext) {
    if (!doRollback && this.ksession != null) {
      return;
      // nothing to initialise
    }

    this.doRollback = false;
    try {
      this.sessionInfo = persistenceContext.findSessionInfo(sessionId);
    } catch (Exception e) {
      throw new RuntimeException("Could not find session data for id " + sessionId, e);
    }

    if (sessionInfo == null) {
      throw new RuntimeException("Could not find session data for id " + sessionId);
    }

    if (this.marshallingHelper == null) {
      // this should only happen when this class is first constructed
      this.marshallingHelper = new SessionMarshallingHelper(kbase, conf, env);
      MarshallingConfigurationImpl config =
          (MarshallingConfigurationImpl)
              ((DefaultMarshaller) this.marshallingHelper.getMarshaller())
                  .getMarshallingConfiguration();
      config.setMarshallProcessInstances(false);
      config.setMarshallWorkItems(false);
    }

    this.sessionInfo.setJPASessionMashallingHelper(this.marshallingHelper);

    // if this.ksession is null, it'll create a new one, else it'll use the existing one
    this.ksession = this.marshallingHelper.loadSnapshot(this.sessionInfo.getData(), this.ksession);

    // update the session id to be the same as the session info id
    ((InternalKnowledgeRuntime) ksession).setId(this.sessionInfo.getId());

    ((InternalKnowledgeRuntime) this.ksession)
        .setEndOperationListener(new EndOperationListenerImpl(this.sessionInfo));

    if (this.kContext == null) {
      // this should only happen when this class is first constructed
      this.kContext = new FixedKnowledgeCommandContext(null, null, null, this.ksession, null);
    }

    this.commandService = new DefaultCommandService(kContext);

    ((AcceptsTimerJobFactoryManager) ((InternalKnowledgeRuntime) ksession).getTimerService())
        .getTimerJobFactoryManager()
        .setCommandService(this);
  }
コード例 #2
0
  public SingleSessionCommandService(
      KnowledgeBase kbase, KnowledgeSessionConfiguration conf, Environment env) {
    if (conf == null) {
      conf = new SessionConfiguration();
    }
    this.env = env;

    checkEnvironment(this.env);

    this.sessionInfo = new SessionInfo();

    initTransactionManager(this.env);

    // create session but bypass command service
    this.ksession = kbase.newStatefulKnowledgeSession(conf, this.env);

    this.kContext = new FixedKnowledgeCommandContext(null, null, null, this.ksession, null);

    this.commandService = new DefaultCommandService(kContext);

    ((AcceptsTimerJobFactoryManager) ((InternalKnowledgeRuntime) ksession).getTimerService())
        .getTimerJobFactoryManager()
        .setCommandService(this);

    this.marshallingHelper = new SessionMarshallingHelper(this.ksession, conf);
    MarshallingConfigurationImpl config =
        (MarshallingConfigurationImpl)
            ((DefaultMarshaller) this.marshallingHelper.getMarshaller())
                .getMarshallingConfiguration();
    config.setMarshallProcessInstances(false);
    config.setMarshallWorkItems(false);

    this.sessionInfo.setJPASessionMashallingHelper(this.marshallingHelper);
    ((InternalKnowledgeRuntime) this.ksession)
        .setEndOperationListener(new EndOperationListenerImpl(this.sessionInfo));

    // Use the App scoped EntityManager if the user has provided it, and it is open.
    // - open the entity manager before the transaction begins.
    PersistenceContext persistenceContext = jpm.getApplicationScopedPersistenceContext();

    boolean transactionOwner = false;
    try {
      transactionOwner = txm.begin();

      registerRollbackSync();

      persistenceContext.joinTransaction();
      persistenceContext.persist(this.sessionInfo);

      txm.commit(transactionOwner);
    } catch (RuntimeException re) {
      rollbackTransaction(re, transactionOwner);
      throw re;
    } catch (Exception t1) {
      rollbackTransaction(t1, transactionOwner);
      throw new RuntimeException("Wrapped exception see cause", t1);
    }

    // update the session id to be the same as the session info id
    ((InternalKnowledgeRuntime) ksession).setId(this.sessionInfo.getId());
  }