Package org.apache.drill.common.config

Examples of org.apache.drill.common.config.DrillConfig

23
24
25
26
27
28
29
30
31
public class Simple {
  static final org.slf4j.Logger logger = org.slf4j.LoggerFactory.getLogger(Simple.class);
 
  public static void main(String[] args) throws Exception {
    DrillConfig c = DrillConfig.create();
    LogicalPlan plan = LogicalPlan.parse(c, FileUtils.getResourceAsString("/jdbc_plan.json"));
    System.out.println(plan.toJsonString(c));
  }
View Full Code Here
36
37
38
39
40
41
42
43
44
45
46
  static final org.slf4j.Logger logger = org.slf4j.LoggerFactory.getLogger(RunSimplePlan.class);
 
 
  @Test
  public void parseSimplePlan() throws Throwable{
    DrillConfig config = DrillConfig.create();
    LogicalPlan plan = LogicalPlan.parse(config, Files.toString(FileUtils.getResourceAsFile("/simple_plan.json"), Charsets.UTF_8));
    IteratorRegistry ir = new IteratorRegistry();
    ReferenceInterpreter i = new ReferenceInterpreter(plan, ir, new BasicEvaluatorFactory(ir), new RSERegistry(config));
    i.setup();
    Collection<RunOutcome> outcomes = i.run();
View Full Code Here
48
49
50
51
52
53
54
55
56
57
58
    assertEquals(outcomes.iterator().next().records, 2);
  }
 
  @Test
  public void joinPlan() throws Throwable{
    DrillConfig config = DrillConfig.create();
    LogicalPlan plan = LogicalPlan.parse(config, Files.toString(FileUtils.getResourceAsFile("/simple_join.json"), Charsets.UTF_8));
    IteratorRegistry ir = new IteratorRegistry();
    ReferenceInterpreter i = new ReferenceInterpreter(plan, ir, new BasicEvaluatorFactory(ir), new RSERegistry(config));
    i.setup();
    Collection<RunOutcome> outcomes = i.run();
View Full Code Here
61
62
63
64
65
66
67
68
69
70
71
    if(out.outcome != OutcomeType.FAILED && out.exception != null) logger.error("Failure while running {}", out.exception);
  }
 
  @Test
  public void flattenPlan() throws Throwable{
    DrillConfig config = DrillConfig.create();
    LogicalPlan plan = LogicalPlan.parse(config, Files.toString(FileUtils.getResourceAsFile("/simple_plan_flattened.json"), Charsets.UTF_8));
    IteratorRegistry ir = new IteratorRegistry();
    ReferenceInterpreter i = new ReferenceInterpreter(plan, ir, new BasicEvaluatorFactory(ir), new RSERegistry(config));
    i.setup();
    Collection<RunOutcome> outcomes = i.run();
View Full Code Here
82
83
84
85
86
87
88
89
90
91
92
    // not sure if we want to keep this as a test and check the results. Now that the internals of the ConstantROP work
    // it might now be worth running the reference intepreter with every build
    @Test
    @Ignore // this plan needs to be updated.
    public void testRefInterp() throws Exception{
            DrillConfig config = DrillConfig.create();
            final String jsonFile = "/constant2.json";
            LogicalPlan plan = LogicalPlan.parse(config, FileUtils.getResourceAsString(jsonFile));
            org.junit.Assert.assertEquals("Constant operator not read in properly or not recognized as a source operator.",
                    plan.getGraph().getLeaves().toString(), "[Constant [memo=null]]");
View Full Code Here
28
29
30
31
32
33
34
35
36
37
38
public class RSERegistryTest {

  @Test
  public void testEnginesWithTheSameNameAreEqual() {
    DrillConfig config = DrillConfig.create();
    RSERegistry rses = new RSERegistry(config);
    StorageEngineConfig hconfig = new ConsoleRSE.ConsoleRSEConfig();
    ConsoleRSE engine = (ConsoleRSE) rses.getEngine(hconfig);
    ConsoleRSE engine2 = (ConsoleRSE) rses.getEngine(hconfig);
    assertSame(engine, engine2);
View Full Code Here
65
66
67
68
69
70
71
72
73
   * @param resourcePath path for json plan file
   * @param recordCount expected record count
   * @throws Exception
   */
  public static void assertProduceCount(String resourcePath, int recordCount) throws Exception {
    DrillConfig config = getConfigWithQueue(0);
    Collection<RunOutcome> outcomes = getOutcome(config, resourcePath);
    assertEquals(recordCount, outcomes.iterator().next().records);
  }
View Full Code Here
86
87
88
89
90
91
92
93
94
95
    i.setup();
    return i.run();
  }
 
  private static DrillConfig getConfigWithQueue(int queueNumber){
    DrillConfig config = DrillConfig.create();
    final BlockingQueue<Object> queue = new ArrayBlockingQueue<Object>(100);
    config.setSinkQueues(queueNumber, queue);
    return config;
  }
View Full Code Here
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
  }
 
 
 
  public static List<UnbackedRecord> getResultAsUnbackedRecords(String resourcePath) throws Exception{
    DrillConfig config = getConfigWithQueue(0);
    Collection<RunOutcome> outcomes = getOutcome(config, resourcePath);
    if(outcomes.size() != 1) throw new Exception("Only supports logical plans that provide a single outcome.");
    RunOutcome out = outcomes.iterator().next();
    switch(out.outcome){
    case CANCELED:
    case FAILED:
      if(out.exception != null) throw (Exception) out.exception;
      throw new Exception("Failure while running query.");
    }
    Object o;
    Queue<Object> q = config.getQueue(0);
    List<UnbackedRecord> records = Lists.newArrayList();
    while( (o = q.poll()) != null){
      if(o instanceof OutcomeType) continue;
      if( !(o instanceof UnbackedRecord) ) throw new Exception(String.format("This method only works when the QueueRSE is configured to use RECORD encoding.  One of the queue objects was of type %s", o.getClass().getCanonicalName()));
      records.add( (UnbackedRecord) o);
View Full Code Here
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
   
  }
 
 
  public static void main(String[] args) throws Exception{
    DrillConfig config = DrillConfig.create();
    int arg = 0;
    final BlockingQueue<Object> queue;
    if (arg < args.length && args[arg].equals("--stdout")) {
      ++arg;
      queue = new ArrayBlockingQueue<>(100);
      config.setSinkQueues(0, queue);
    } else {
      queue = null;
    }
    final String jsonFile = args[arg];
    final String planString;
View Full Code Here
TOP

Related Classes of org.apache.drill.common.config.DrillConfig

  • com.fasterxml.jackson.databind.module.SimpleModule
  • com.fasterxml.jackson.databind.ObjectMapper
  • com.typesafe.config.Config
  • org.apache.drill.common.logical.LogicalPlanBuilderTest
  • org.apache.drill.exec.cache.TestWriteToDisk
  • org.apache.drill.exec.client.QuerySubmitter
  • org.apache.drill.exec.compile.bytecode.ReplaceMethodInvoke
  • org.apache.drill.exec.compile.TestEvaluationVisitor
  • org.apache.drill.exec.fn.impl.TestByteComparisonFunctions
  • org.apache.drill.exec.opt.BasicOptimizerTest
    • Copyright © 2018 www.massapicom. All rights reserved.
    All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.