Package com.facebook.presto.type

Examples of com.facebook.presto.type.TypeRegistry

77
78
79
80
81
82
83
84
85
86
        Signature signature = getMagicLiteralFunctionSignature(TIMESTAMP_WITH_TIME_ZONE);
        assertEquals(signature.getName(), "$literal$timestamp with time zone");
        assertEquals(signature.getArgumentTypes(), ImmutableList.of(StandardTypes.BIGINT));
        assertEquals(signature.getReturnType(), StandardTypes.TIMESTAMP_WITH_TIME_ZONE);

        FunctionRegistry registry = new FunctionRegistry(new TypeRegistry(), true);
        FunctionInfo function = registry.resolveFunction(QualifiedName.of(signature.getName()), signature.getArgumentTypes(), false);
        assertEquals(function.getArgumentTypes(), ImmutableList.of(StandardTypes.BIGINT));
        assertEquals(signature.getReturnType(), StandardTypes.TIMESTAMP_WITH_TIME_ZONE);
    }
View Full Code Here
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
    }

    @Test(expectedExceptions = IllegalArgumentException.class, expectedExceptionsMessageRegExp = "\\QFunction already registered: custom_add(bigint,bigint):bigint\\E")
    public void testDuplicateFunctions()
    {
        List<ParametricFunction> functions = new FunctionListBuilder(new TypeRegistry())
                .scalar(CustomFunctions.class)
                .getFunctions();

        functions = FluentIterable.from(functions).filter(new Predicate<ParametricFunction>()
        {
            @Override
            public boolean apply(ParametricFunction input)
            {
                return input.getSignature().getName().equals("custom_add");
            }
        }).toList();

        FunctionRegistry registry = new FunctionRegistry(new TypeRegistry(), true);
        registry.addFunctions(functions);
        registry.addFunctions(functions);
    }
View Full Code Here
108
109
110
111
112
113
114
115
116
117
118
119
    @Test(expectedExceptions = IllegalStateException.class, expectedExceptionsMessageRegExp = "'sum' is both an aggregation and a scalar function")
    public void testConflictingScalarAggregation()
            throws Exception
    {
        List<ParametricFunction> functions = new FunctionListBuilder(new TypeRegistry())
                .scalar(ScalarSum.class)
                .getFunctions();

        FunctionRegistry registry = new FunctionRegistry(new TypeRegistry(), true);
        registry.addFunctions(functions);
    }
View Full Code Here
120
121
122
123
124
125
126
127
128
129
130
    @Test
    public void testListingHiddenFunctions()
            throws Exception
    {
        FunctionRegistry registry = new FunctionRegistry(new TypeRegistry(), true);
        List<ParametricFunction> functions = registry.list();
        List<String> names = transform(functions, nameGetter());

        assertTrue(names.contains("length"), "Expected function names " + names + " to contain 'length'");
        assertTrue(names.contains("stddev"), "Expected function names " + names + " to contain 'stddev'");
View Full Code Here
42
43
44
45
46
47
48
49
    }

    @Test
    public void testNonexistentType()
    {
        TypeManager typeManager = new TypeRegistry();
        assertNull(typeManager.getType("not a real type"));
    }
View Full Code Here
33
34
35
36
37
38
39
{
    @BeforeClass
    public void setup()
    {
        InternalAggregationFunction function = new AggregationCompiler().generateAggregationFunction(CountNull.class);
        functionRegistry.addFunctions(new FunctionListBuilder(new TypeRegistry()).aggregate(function).getFunctions());
    }
View Full Code Here
70
71
72
73
74
75
76
77
78
79
80
        HiveConnectorFactory hiveConnectorFactory = new HiveConnectorFactory(
                "hive",
                ImmutableMap.of("node.environment", "test"),
                HiveBenchmarkQueryRunner.class.getClassLoader(),
                metastore,
                new TypeRegistry());

        Map<String, String> hiveCatalogConfig = ImmutableMap.<String, String>builder()
                .put("hive.metastore.uri", "thrift://none.invalid")
                .put("hive.max-split-size", "10GB")
                .build();
View Full Code Here
73
74
75
76
77
78
79
80
81
82
83
                    splitProperties,
                    new ArrayList<HivePartitionKey>(),
                    getColumns(),
                    DateTimeZone.getDefault(),
                    DateTimeZone.getDefault(),
                    new TypeRegistry());

            checkCursor(cursor);
        }
        finally {
            //noinspection ResultOfMethodCallIgnored
View Full Code Here
108
109
110
111
112
113
114
115
116
117
118
                    split.getLength(),
                    splitProperties,
                    new ArrayList<HivePartitionKey>(),
                    getColumns(),
                    DateTimeZone.getDefault(),
                    new TypeRegistry());

            checkCursor(cursor);
        }
        finally {
            //noinspection ResultOfMethodCallIgnored
View Full Code Here
245
246
247
248
249
250
251
252
253
254
255
                hiveClientConfig.getMaxInitialSplits(),
                false,
                true,
                hiveClientConfig.getHiveStorageFormat(),
                false,
                new TypeRegistry());

        metadata = client;
        splitManager = client;
        recordSetProvider = client;
        recordSinkProvider = client;
View Full Code Here
TOP

Related Classes of com.facebook.presto.type.TypeRegistry

  • com.facebook.presto.AbstractTestQueryFramework
  • com.facebook.presto.benchmark.BenchmarkQueryRunner
  • com.facebook.presto.execution.TaskTestUtils
  • com.facebook.presto.execution.TestSqlStageExecution
  • com.facebook.presto.execution.TestSqlTaskExecution
  • com.facebook.presto.execution.TestSqlTaskManager
  • com.facebook.presto.hive.AbstractTestHiveClient
  • com.facebook.presto.hive.AbstractTestHiveClientS3
  • com.facebook.presto.hive.HiveBenchmarkQueryRunner
  • com.facebook.presto.hive.TestHiveConnectorFactory
    • 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.