コード例 #1
0
ファイル: QueryBuilderTest.java プロジェクト: adinn/drools
  @Test
  public void testQuery() throws Exception {
    final PackageBuilder builder = new PackageBuilder();

    final PackageDescr packageDescr = new PackageDescr("p1");
    final QueryDescr queryDescr = new QueryDescr("query1");
    queryDescr.addParameter("String", "$type");

    packageDescr.addRule(queryDescr);

    final AndDescr lhs = new AndDescr();
    queryDescr.setLhs(lhs);

    final PatternDescr pattern = new PatternDescr(Cheese.class.getName(), "stilton");
    lhs.addDescr(pattern);

    pattern.addConstraint(new ExprConstraintDescr("type == $type"));

    // Another query, no parameters
    QueryDescr queryDescr2 = new QueryDescr("query2");
    packageDescr.addRule(queryDescr2);
    AndDescr lhs2 = new AndDescr();
    queryDescr2.setLhs(lhs2);
    PatternDescr pattern2 = new PatternDescr(Cheese.class.getName());
    lhs2.addDescr(pattern2);

    builder.addPackage(packageDescr);

    assertLength(0, builder.getErrors().getErrors());

    KnowledgeBase kbase = KnowledgeBaseFactory.newKnowledgeBase();
    kbase.addKnowledgePackages(
        Arrays.asList(new KnowledgePackage[] {new KnowledgePackageImp(builder.getPackage())}));

    final KieSession session = kbase.newStatefulKnowledgeSession();

    session.insert(new Cheese("stilton", 15));

    QueryResults results = session.getQueryResults("query1", "stilton");
    assertEquals(1, results.size());

    Object object = results.iterator().next().get("stilton");
    assertEquals(new Cheese("stilton", 15), object);

    results = session.getQueryResults("query1", new Object[] {"cheddar"});
    assertEquals(0, results.size());

    session.insert(new Cheese("dolcelatte", 20));
    results = session.getQueryResults("query2", new Object[] {});
    assertEquals(2, results.size());
  }
コード例 #2
0
ファイル: FromHandler.java プロジェクト: 2356901696/drools
  public Object end(final String uri, final String localName, final ExtensibleXmlParser parser)
      throws SAXException {
    final Element element = parser.endElementBuilder();

    final FromDescr fromDescr = (FromDescr) parser.getCurrent();

    Object parent = parser.getParent();

    final PatternDescr patternDescr = (PatternDescr) parent;

    final ConditionalElementDescr parentDescr = (ConditionalElementDescr) parser.getParent(1);

    if (element.getElementsByTagName("expression").getLength() > 0) {
      patternDescr.setSource(fromDescr);
    }

    return fromDescr;
  }
コード例 #3
0
  public Object end(final String uri, final String localName, final ExtensibleXmlParser parser)
      throws SAXException {

    final Element element = parser.endElementBuilder();
    final AccumulateDescr accumulateDescr = (AccumulateDescr) parser.getCurrent();

    final Object parent = parser.getParent();

    if (parent.getClass().getName().equals(FromDescr.class.getName())) {
      final PatternDescr result = (PatternDescr) parser.getParent(1);
      result.setSource(accumulateDescr);

    } else if (parent instanceof ConditionalElementDescr) {
      final ConditionalElementDescr parentDescr = (ConditionalElementDescr) parent;
      parentDescr.addDescr(accumulateDescr);
    }

    return accumulateDescr;
  }
コード例 #4
0
  @SuppressWarnings("unchecked")
  public RuleConditionElement build(
      RuleBuildContext context, BaseDescr descr, Pattern prefixPattern, Query query) {
    PatternDescr patternDescr = (PatternDescr) descr;

    Declaration[] params = query.getParameters();

    List<BaseDescr> args = (List<BaseDescr>) patternDescr.getDescrs();
    List<Integer> declrIndexes = new ArrayList<Integer>();
    List<Integer> varIndexes = new ArrayList<Integer>();
    List<Object> arguments = new ArrayList<Object>(params.length);
    for (int i = 0; i < params.length; i++) {
      // as these could be set in any order, initialise first, to allow setting later.
      arguments.add(null);
    }
    List<Declaration> requiredDeclarations = new ArrayList<Declaration>();

    ObjectType argsObjectType = ClassObjectType.ObjectArray_ObjectType;
    InternalReadAccessor arrayReader = new SelfReferenceClassFieldReader(Object[].class, "this");
    Pattern pattern = new Pattern(context.getNextPatternId(), 0, argsObjectType, null);

    // Deal with the constraints, both positional and bindings
    for (int i = 0, length = args.size(); i < length; i++) {
      BaseDescr base = args.get(i);

      String expression = null;
      boolean isPositional = false;
      boolean isBinding = false;
      BindingDescr bind = null;
      ConstraintConnectiveDescr result = null;
      if (base instanceof BindingDescr) {
        bind = (BindingDescr) base;
        expression =
            bind.getVariable() + (bind.isUnification() ? " := " : " : ") + bind.getExpression();
        isBinding = true;
      } else {
        if (base instanceof ExprConstraintDescr) {
          ExprConstraintDescr ecd = (ExprConstraintDescr) base;
          expression = ecd.getExpression();
          isPositional = ecd.getType() == ExprConstraintDescr.Type.POSITIONAL;

        } else {
          expression = base.getText();
        }

        result = parseExpression(context, patternDescr, expression);
        if (result == null) {
          // error, can't parse expression.
          context.addError(
              new DescrBuildError(
                  context.getParentDescr(),
                  descr,
                  null,
                  "Unable to parse constraint: \n" + expression));
          continue;
        }
        isBinding =
            result.getDescrs().size() == 1 && result.getDescrs().get(0) instanceof BindingDescr;
        if (isBinding) {
          bind = (BindingDescr) result.getDescrs().get(0);
        }
      }

      if ((!isPositional) && (!isBinding)) {
        // error, can't have non binding slots.
        context.addError(
            new DescrBuildError(
                context.getParentDescr(),
                descr,
                null,
                "Query's must use positional or bindings, not field constraints:\n" + expression));
        continue;
      } else if (isPositional && isBinding) {
        // error, can't have positional binding slots.
        context.addError(
            new DescrBuildError(
                context.getParentDescr(),
                descr,
                null,
                "Query's can't use positional bindings:\n" + expression));
        continue;
      } else if (isPositional) {
        processPositional(
            context,
            query,
            params,
            declrIndexes,
            varIndexes,
            arguments,
            requiredDeclarations,
            arrayReader,
            pattern,
            base,
            expression,
            result);
      } else {
        // it is binding
        processBinding(
            context,
            descr,
            params,
            declrIndexes,
            varIndexes,
            arguments,
            requiredDeclarations,
            arrayReader,
            pattern,
            bind,
            result);
      }
    }

    Declaration[] declrsArray =
        requiredDeclarations.toArray(new Declaration[requiredDeclarations.size()]);
    int[] declrIndexArray = new int[declrIndexes.size()];
    for (int i = 0; i < declrsArray.length; i++) {
      declrIndexArray[i] = declrIndexes.get(i);
    }
    int[] varIndexesArray = new int[varIndexes.size()];
    for (int i = 0; i < varIndexesArray.length; i++) {
      varIndexesArray[i] = varIndexes.get(i);
    }

    return new QueryElement(
        pattern,
        query.getName(),
        arguments.toArray(new Object[arguments.size()]),
        declrsArray,
        declrIndexArray,
        varIndexesArray,
        !patternDescr.isQuery());
  }
コード例 #5
0
ファイル: XmlDumper.java プロジェクト: XBoom/drools
  public void visitPatternDescr(final PatternDescr descr) {
    this.patternContext = true;
    this.template = new String();
    StringBuilder localString = new StringBuilder();

    if (descr.getDescrs() != Collections.EMPTY_LIST) {
      if (descr.getIdentifier() != null) {
        localString.append(
            "<pattern identifier=\""
                + descr.getIdentifier()
                + "\" object-type=\""
                + descr.getObjectType()
                + "\" >"
                + XmlDumper.eol
                + processDescrList(descr.getDescrs())
                + XmlDumper.eol);
      } else {
        localString.append(
            "<pattern object-type=\""
                + descr.getObjectType()
                + "\" >"
                + XmlDumper.eol
                + processDescrList(descr.getDescrs())
                + XmlDumper.eol);
      }
    } else {
      if (descr.getIdentifier() != null) {
        localString.append(
            "<pattern identifier=\""
                + descr.getIdentifier()
                + "\" object-type=\""
                + descr.getObjectType()
                + "\" >"
                + XmlDumper.eol);
        ;
      } else {
        localString.append(
            "<pattern object-type=\"" + descr.getObjectType() + "\" >" + XmlDumper.eol);
      }
    }

    if (descr.getSource() != null) {
      visit(descr.getSource());
      localString.append(this.template);
    }
    localString.append("</pattern>" + XmlDumper.eol);

    this.template = localString.toString();

    this.patternContext = false;
  }
コード例 #6
0
ファイル: QueryBuilderTest.java プロジェクト: adinn/drools
  @Test
  public void testRuleWithQuery() throws Exception {
    final PackageBuilder builder = new PackageBuilder();

    final PackageDescr packageDescr = new PackageDescr("p1");

    QueryDescr queryDescr = new QueryDescr("query1");
    queryDescr.addParameter("String", "$name");
    queryDescr.addParameter("int", "$age");
    queryDescr.addParameter("String", "$likes");
    packageDescr.addRule(queryDescr);
    AndDescr lhs = new AndDescr();
    queryDescr.setLhs(lhs);
    PatternDescr pattern = new PatternDescr(Person.class.getName());
    lhs.addDescr(pattern);
    pattern.addConstraint(new BindingDescr("$name", "name", true));
    pattern.addConstraint(new BindingDescr("$age", "age", true));
    pattern.addConstraint(new BindingDescr("$likes", "likes", true));

    RuleDescr ruleDescr = new RuleDescr("rule-1");
    packageDescr.addRule(ruleDescr);
    lhs = new AndDescr();
    ruleDescr.setLhs(lhs);

    pattern = new PatternDescr(Cheese.class.getName());
    lhs.addDescr(pattern);
    pattern.addConstraint(new BindingDescr("$type", "type"));

    pattern = new PatternDescr("query1");
    pattern.setQuery(true);
    lhs.addDescr(pattern);
    ExprConstraintDescr expr = new ExprConstraintDescr("'bobba'");
    expr.setPosition(0);
    expr.setType(Type.POSITIONAL);
    pattern.addConstraint(expr);

    expr = new ExprConstraintDescr("$age");
    expr.setPosition(1);
    expr.setType(Type.POSITIONAL);
    pattern.addConstraint(expr);

    expr = new ExprConstraintDescr("$type");
    expr.setPosition(2);
    expr.setType(Type.POSITIONAL);
    pattern.addConstraint(expr);
    ruleDescr.setConsequence("System.out.println(\"age: \" + $age);");

    builder.addPackage(packageDescr);
    assertLength(0, builder.getErrors().getErrors());

    KnowledgeBase kbase = KnowledgeBaseFactory.newKnowledgeBase();
    kbase.addKnowledgePackages(
        Arrays.asList(new KnowledgePackage[] {new KnowledgePackageImp(builder.getPackage())}));
    final KieSession session = kbase.newStatefulKnowledgeSession();

    session.insert(new Person("bobba", "stilton", 90));
    session.insert(new Person("bobba", "brie", 80));
    session.insert(new Person("bobba", "brie", 75));
    session.insert(new Person("darth", "brie", 100));
    session.insert(new Person("luke", "brie", 25));
    session.insert(new Cheese("brie", 25));
    session.fireAllRules();
  }