コード例 #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 static Object unmarshall(byte[] content, Environment env, ClassLoader classloader) {
    MarshallerReaderContext context = null;
    try {
      ByteArrayInputStream stream = new ByteArrayInputStream(content);
      MarshallingConfigurationImpl marshallingConfigurationImpl = null;
      if (env != null) {
        marshallingConfigurationImpl =
            new MarshallingConfigurationImpl(
                (ObjectMarshallingStrategy[])
                    env.get(EnvironmentName.OBJECT_MARSHALLING_STRATEGIES),
                false,
                false);
      } else {
        marshallingConfigurationImpl =
            new MarshallingConfigurationImpl(
                new ObjectMarshallingStrategy[] {
                  new SerializablePlaceholderResolverStrategy(
                      ClassObjectMarshallingStrategyAcceptor.DEFAULT)
                },
                false,
                false);
      }
      ObjectMarshallingStrategyStore objectMarshallingStrategyStore =
          marshallingConfigurationImpl.getObjectMarshallingStrategyStore();
      context =
          new MarshallerReaderContext(
              stream, null, null, objectMarshallingStrategyStore, null, env);
      if (classloader != null) {
        context.classLoader = classloader;
      } else {
        context.classLoader = ContentMarshallerHelper.class.getClassLoader();
      }
      ExtensionRegistry registry = PersisterHelper.buildRegistry(context, null);
      Header _header = PersisterHelper.readFromStreamWithHeader(context, registry);
      Variable parseFrom = JBPMMessages.Variable.parseFrom(_header.getPayload(), registry);
      Object value = ProtobufProcessMarshaller.unmarshallVariableValue(context, parseFrom);

      if (value instanceof Map) {
        Map result = new HashMap();
        Map<String, Variable> variablesMap = (Map<String, Variable>) value;
        for (String key : variablesMap.keySet()) {
          result.put(
              key,
              ProtobufProcessMarshaller.unmarshallVariableValue(context, variablesMap.get(key)));
        }
        return result;
      }
      return value;
    } catch (Exception ex) {
      ex.printStackTrace();
    }
    return null;
  }
コード例 #3
0
  public static ContentData marshal(Object o, Environment env) {
    MarshallerWriteContext context = null;
    ContentData content = null;
    try {
      MarshallingConfigurationImpl marshallingConfigurationImpl = null;
      if (env != null) {
        marshallingConfigurationImpl =
            new MarshallingConfigurationImpl(
                (ObjectMarshallingStrategy[])
                    env.get(EnvironmentName.OBJECT_MARSHALLING_STRATEGIES),
                false,
                false);
      } else {
        marshallingConfigurationImpl =
            new MarshallingConfigurationImpl(
                new ObjectMarshallingStrategy[] {
                  new SerializablePlaceholderResolverStrategy(
                      ClassObjectMarshallingStrategyAcceptor.DEFAULT)
                },
                false,
                false);
      }
      ObjectMarshallingStrategyStore objectMarshallingStrategyStore =
          marshallingConfigurationImpl.getObjectMarshallingStrategyStore();
      ByteArrayOutputStream stream = new ByteArrayOutputStream();

      context =
          new MarshallerWriteContext(stream, null, null, null, objectMarshallingStrategyStore, env);
      Variable marshallVariable = null;
      if (o instanceof Map) {
        marshallVariable =
            ProtobufProcessMarshaller.marshallVariablesMap(context, (Map<String, Object>) o);
      } else {
        marshallVariable = ProtobufProcessMarshaller.marshallVariable(context, "results", o);
      }
      PersisterHelper.writeToStreamWithHeader(context, marshallVariable);

      context.close();

      byte[] toByteArray = stream.toByteArray();

      content = new ContentData();
      content.setContent(toByteArray);
      content.setType(o.getClass().getCanonicalName());
      content.setAccessType(AccessType.Inline);

    } catch (IOException ex) {
      ex.printStackTrace();
    }
    return content;
  }
コード例 #4
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());
  }