org.apache.calcite.test.schemata.countries.CountriesTableFunction Maven / Gradle / Ivy
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to you under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.calcite.test.schemata.countries;
import org.apache.calcite.DataContext;
import org.apache.calcite.config.CalciteConnectionConfig;
import org.apache.calcite.linq4j.Enumerable;
import org.apache.calcite.linq4j.Linq4j;
import org.apache.calcite.rel.type.RelDataType;
import org.apache.calcite.rel.type.RelDataTypeFactory;
import org.apache.calcite.schema.ScannableTable;
import org.apache.calcite.schema.Schema;
import org.apache.calcite.schema.Statistic;
import org.apache.calcite.schema.Statistics;
import org.apache.calcite.sql.SqlCall;
import org.apache.calcite.sql.SqlNode;
import org.apache.calcite.sql.type.SqlTypeName;
import org.apache.calcite.util.ImmutableBitSet;
import com.google.common.collect.ImmutableList;
import org.checkerframework.checker.nullness.qual.Nullable;
/** A table function that returns all countries in the world.
*
* Has same content as
* file/src/test/resources/geo/countries.csv. */
public class CountriesTableFunction {
private CountriesTableFunction() {}
private static final Object[][] ROWS = {
{"AD", 42.546245, 1.601554, "Andorra"},
{"AE", 23.424076, 53.847818, "United Arab Emirates"},
{"AF", 33.93911, 67.709953, "Afghanistan"},
{"AG", 17.060816, -61.796428, "Antigua and Barbuda"},
{"AI", 18.220554, -63.068615, "Anguilla"},
{"AL", 41.153332, 20.168331, "Albania"},
{"AM", 40.069099, 45.038189, "Armenia"},
{"AN", 12.226079, -69.060087, "Netherlands Antilles"},
{"AO", -11.202692, 17.873887, "Angola"},
{"AQ", -75.250973, -0.071389, "Antarctica"},
{"AR", -38.416097, -63.616672, "Argentina"},
{"AS", -14.270972, -170.132217, "American Samoa"},
{"AT", 47.516231, 14.550072, "Austria"},
{"AU", -25.274398, 133.775136, "Australia"},
{"AW", 12.52111, -69.968338, "Aruba"},
{"AZ", 40.143105, 47.576927, "Azerbaijan"},
{"BA", 43.915886, 17.679076, "Bosnia and Herzegovina"},
{"BB", 13.193887, -59.543198, "Barbados"},
{"BD", 23.684994, 90.356331, "Bangladesh"},
{"BE", 50.503887, 4.469936, "Belgium"},
{"BF", 12.238333, -1.561593, "Burkina Faso"},
{"BG", 42.733883, 25.48583, "Bulgaria"},
{"BH", 25.930414, 50.637772, "Bahrain"},
{"BI", -3.373056, 29.918886, "Burundi"},
{"BJ", 9.30769, 2.315834, "Benin"},
{"BM", 32.321384, -64.75737, "Bermuda"},
{"BN", 4.535277, 114.727669, "Brunei"},
{"BO", -16.290154, -63.588653, "Bolivia"},
{"BR", -14.235004, -51.92528, "Brazil"},
{"BS", 25.03428, -77.39628, "Bahamas"},
{"BT", 27.514162, 90.433601, "Bhutan"},
{"BV", -54.423199, 3.413194, "Bouvet Island"},
{"BW", -22.328474, 24.684866, "Botswana"},
{"BY", 53.709807, 27.953389, "Belarus"},
{"BZ", 17.189877, -88.49765, "Belize"},
{"CA", 56.130366, -106.346771, "Canada"},
{"CC", -12.164165, 96.870956, "Cocos [Keeling] Islands"},
{"CD", -4.038333, 21.758664, "Congo [DRC]"},
{"CF", 6.611111, 20.939444, "Central African Republic"},
{"CG", -0.228021, 15.827659, "Congo [Republic]"},
{"CH", 46.818188, 8.227512, "Switzerland"},
{"CI", 7.539989, -5.54708, "Côte d'Ivoire"},
{"CK", -21.236736, -159.777671, "Cook Islands"},
{"CL", -35.675147, -71.542969, "Chile"},
{"CM", 7.369722, 12.354722, "Cameroon"},
{"CN", 35.86166, 104.195397, "China"},
{"CO", 4.570868, -74.297333, "Colombia"},
{"CR", 9.748917, -83.753428, "Costa Rica"},
{"CU", 21.521757, -77.781167, "Cuba"},
{"CV", 16.002082, -24.013197, "Cape Verde"},
{"CX", -10.447525, 105.690449, "Christmas Island"},
{"CY", 35.126413, 33.429859, "Cyprus"},
{"CZ", 49.817492, 15.472962, "Czech Republic"},
{"DE", 51.165691, 10.451526, "Germany"},
{"DJ", 11.825138, 42.590275, "Djibouti"},
{"DK", 56.26392, 9.501785, "Denmark"},
{"DM", 15.414999, -61.370976, "Dominica"},
{"DO", 18.735693, -70.162651, "Dominican Republic"},
{"DZ", 28.033886, 1.659626, "Algeria"},
{"EC", -1.831239, -78.183406, "Ecuador"},
{"EE", 58.595272, 25.013607, "Estonia"},
{"EG", 26.820553, 30.802498, "Egypt"},
{"EH", 24.215527, -12.885834, "Western Sahara"},
{"ER", 15.179384, 39.782334, "Eritrea"},
{"ES", 40.463667, -3.74922, "Spain"},
{"ET", 9.145, 40.489673, "Ethiopia"},
{"FI", 61.92411, 25.748151, "Finland"},
{"FJ", -16.578193, 179.414413, "Fiji"},
{"FK", -51.796253, -59.523613, "Falkland Islands [Islas Malvinas]"},
{"FM", 7.425554, 150.550812, "Micronesia"},
{"FO", 61.892635, -6.911806, "Faroe Islands"},
{"FR", 46.227638, 2.213749, "France"},
{"GA", -0.803689, 11.609444, "Gabon"},
{"GB", 55.378051, -3.435973, "United Kingdom"},
{"GD", 12.262776, -61.604171, "Grenada"},
{"GE", 42.315407, 43.356892, "Georgia"},
{"GF", 3.933889, -53.125782, "French Guiana"},
{"GG", 49.465691, -2.585278, "Guernsey"},
{"GH", 7.946527, -1.023194, "Ghana"},
{"GI", 36.137741, -5.345374, "Gibraltar"},
{"GL", 71.706936, -42.604303, "Greenland"},
{"GM", 13.443182, -15.310139, "Gambia"},
{"GN", 9.945587, -9.696645, "Guinea"},
{"GP", 16.995971, -62.067641, "Guadeloupe"},
{"GQ", 1.650801, 10.267895, "Equatorial Guinea"},
{"GR", 39.074208, 21.824312, "Greece"},
{"GS", -54.429579, -36.587909, "South Georgia and the South Sandwich Islands"},
{"GT", 15.783471, -90.230759, "Guatemala"},
{"GU", 13.444304, 144.793731, "Guam"},
{"GW", 11.803749, -15.180413, "Guinea-Bissau"},
{"GY", 4.860416, -58.93018, "Guyana"},
{"GZ", 31.354676, 34.308825, "Gaza Strip"},
{"HK", 22.396428, 114.109497, "Hong Kong"},
{"HM", -53.08181, 73.504158, "Heard Island and McDonald Islands"},
{"HN", 15.199999, -86.241905, "Honduras"},
{"HR", 45.1, 15.2, "Croatia"},
{"HT", 18.971187, -72.285215, "Haiti"},
{"HU", 47.162494, 19.503304, "Hungary"},
{"ID", -0.789275, 113.921327, "Indonesia"},
{"IE", 53.41291, -8.24389, "Ireland"},
{"IL", 31.046051, 34.851612, "Israel"},
{"IM", 54.236107, -4.548056, "Isle of Man"},
{"IN", 20.593684, 78.96288, "India"},
{"IO", -6.343194, 71.876519, "British Indian Ocean Territory"},
{"IQ", 33.223191, 43.679291, "Iraq"},
{"IR", 32.427908, 53.688046, "Iran"},
{"IS", 64.963051, -19.020835, "Iceland"},
{"IT", 41.87194, 12.56738, "Italy"},
{"JE", 49.214439, -2.13125, "Jersey"},
{"JM", 18.109581, -77.297508, "Jamaica"},
{"JO", 30.585164, 36.238414, "Jordan"},
{"JP", 36.204824, 138.252924, "Japan"},
{"KE", -0.023559, 37.906193, "Kenya"},
{"KG", 41.20438, 74.766098, "Kyrgyzstan"},
{"KH", 12.565679, 104.990963, "Cambodia"},
{"KI", -3.370417, -168.734039, "Kiribati"},
{"KM", -11.875001, 43.872219, "Comoros"},
{"KN", 17.357822, -62.782998, "Saint Kitts and Nevis"},
{"KP", 40.339852, 127.510093, "North Korea"},
{"KR", 35.907757, 127.766922, "South Korea"},
{"KW", 29.31166, 47.481766, "Kuwait"},
{"KY", 19.513469, -80.566956, "Cayman Islands"},
{"KZ", 48.019573, 66.923684, "Kazakhstan"},
{"LA", 19.85627, 102.495496, "Laos"},
{"LB", 33.854721, 35.862285, "Lebanon"},
{"LC", 13.909444, -60.978893, "Saint Lucia"},
{"LI", 47.166, 9.555373, "Liechtenstein"},
{"LK", 7.873054, 80.771797, "Sri Lanka"},
{"LR", 6.428055, -9.429499, "Liberia"},
{"LS", -29.609988, 28.233608, "Lesotho"},
{"LT", 55.169438, 23.881275, "Lithuania"},
{"LU", 49.815273, 6.129583, "Luxembourg"},
{"LV", 56.879635, 24.603189, "Latvia"},
{"LY", 26.3351, 17.228331, "Libya"},
{"MA", 31.791702, -7.09262, "Morocco"},
{"MC", 43.750298, 7.412841, "Monaco"},
{"MD", 47.411631, 28.369885, "Moldova"},
{"ME", 42.708678, 19.37439, "Montenegro"},
{"MG", -18.766947, 46.869107, "Madagascar"},
{"MH", 7.131474, 171.184478, "Marshall Islands"},
{"MK", 41.608635, 21.745275, "Macedonia [FYROM]"},
{"ML", 17.570692, -3.996166, "Mali"},
{"MM", 21.913965, 95.956223, "Myanmar [Burma]"},
{"MN", 46.862496, 103.846656, "Mongolia"},
{"MO", 22.198745, 113.543873, "Macau"},
{"MP", 17.33083, 145.38469, "Northern Mariana Islands"},
{"MQ", 14.641528, -61.024174, "Martinique"},
{"MR", 21.00789, -10.940835, "Mauritania"},
{"MS", 16.742498, -62.187366, "Montserrat"},
{"MT", 35.937496, 14.375416, "Malta"},
{"MU", -20.348404, 57.552152, "Mauritius"},
{"MV", 3.202778, 73.22068, "Maldives"},
{"MW", -13.254308, 34.301525, "Malawi"},
{"MX", 23.634501, -102.552784, "Mexico"},
{"MY", 4.210484, 101.975766, "Malaysia"},
{"MZ", -18.665695, 35.529562, "Mozambique"},
{"NA", -22.95764, 18.49041, "Namibia"},
{"NC", -20.904305, 165.618042, "New Caledonia"},
{"NE", 17.607789, 8.081666, "Niger"},
{"NF", -29.040835, 167.954712, "Norfolk Island"},
{"NG", 9.081999, 8.675277, "Nigeria"},
{"NI", 12.865416, -85.207229, "Nicaragua"},
{"NL", 52.132633, 5.291266, "Netherlands"},
{"NO", 60.472024, 8.468946, "Norway"},
{"NP", 28.394857, 84.124008, "Nepal"},
{"NR", -0.522778, 166.931503, "Nauru"},
{"NU", -19.054445, -169.867233, "Niue"},
{"NZ", -40.900557, 174.885971, "New Zealand"},
{"OM", 21.512583, 55.923255, "Oman"},
{"PA", 8.537981, -80.782127, "Panama"},
{"PE", -9.189967, -75.015152, "Peru"},
{"PF", -17.679742, -149.406843, "French Polynesia"},
{"PG", -6.314993, 143.95555, "Papua New Guinea"},
{"PH", 12.879721, 121.774017, "Philippines"},
{"PK", 30.375321, 69.345116, "Pakistan"},
{"PL", 51.919438, 19.145136, "Poland"},
{"PM", 46.941936, -56.27111, "Saint Pierre and Miquelon"},
{"PN", -24.703615, -127.439308, "Pitcairn Islands"},
{"PR", 18.220833, -66.590149, "Puerto Rico"},
{"PS", 31.952162, 35.233154, "Palestinian Territories"},
{"PT", 39.399872, -8.224454, "Portugal"},
{"PW", 7.51498, 134.58252, "Palau"},
{"PY", -23.442503, -58.443832, "Paraguay"},
{"QA", 25.354826, 51.183884, "Qatar"},
{"RE", -21.115141, 55.536384, "Réunion"},
{"RO", 45.943161, 24.96676, "Romania"},
{"RS", 44.016521, 21.005859, "Serbia"},
{"RU", 61.52401, 105.318756, "Russia"},
{"RW", -1.940278, 29.873888, "Rwanda"},
{"SA", 23.885942, 45.079162, "Saudi Arabia"},
{"SB", -9.64571, 160.156194, "Solomon Islands"},
{"SC", -4.679574, 55.491977, "Seychelles"},
{"SD", 12.862807, 30.217636, "Sudan"},
{"SE", 60.128161, 18.643501, "Sweden"},
{"SG", 1.352083, 103.819836, "Singapore"},
{"SH", -24.143474, -10.030696, "Saint Helena"},
{"SI", 46.151241, 14.995463, "Slovenia"},
{"SJ", 77.553604, 23.670272, "Svalbard and Jan Mayen"},
{"SK", 48.669026, 19.699024, "Slovakia"},
{"SL", 8.460555, -11.779889, "Sierra Leone"},
{"SM", 43.94236, 12.457777, "San Marino"},
{"SN", 14.497401, -14.452362, "Senegal"},
{"SO", 5.152149, 46.199616, "Somalia"},
{"SR", 3.919305, -56.027783, "Suriname"},
{"ST", 0.18636, 6.613081, "São Tomé and Príncipe"},
{"SV", 13.794185, -88.89653, "El Salvador"},
{"SY", 34.802075, 38.996815, "Syria"},
{"SZ", -26.522503, 31.465866, "Swaziland"},
{"TC", 21.694025, -71.797928, "Turks and Caicos Islands"},
{"TD", 15.454166, 18.732207, "Chad"},
{"TF", -49.280366, 69.348557, "French Southern Territories"},
{"TG", 8.619543, 0.824782, "Togo"},
{"TH", 15.870032, 100.992541, "Thailand"},
{"TJ", 38.861034, 71.276093, "Tajikistan"},
{"TK", -8.967363, -171.855881, "Tokelau"},
{"TL", -8.874217, 125.727539, "Timor-Leste"},
{"TM", 38.969719, 59.556278, "Turkmenistan"},
{"TN", 33.886917, 9.537499, "Tunisia"},
{"TO", -21.178986, -175.198242, "Tonga"},
{"TR", 38.963745, 35.243322, "Turkey"},
{"TT", 10.691803, -61.222503, "Trinidad and Tobago"},
{"TV", -7.109535, 177.64933, "Tuvalu"},
{"TW", 23.69781, 120.960515, "Taiwan"},
{"TZ", -6.369028, 34.888822, "Tanzania"},
{"UA", 48.379433, 31.16558, "Ukraine"},
{"UG", 1.373333, 32.290275, "Uganda"},
{"UM", null, null, "U.S.Minor Outlying Islands"},
{"US", 37.09024, -95.712891, "United States"},
{"UY", -32.522779, -55.765835, "Uruguay"},
{"UZ", 41.377491, 64.585262, "Uzbekistan"},
{"VA", 41.902916, 12.453389, "Vatican City"},
{"VC", 12.984305, -61.287228, "Saint Vincent and the Grenadines"},
{"VE", 6.42375, -66.58973, "Venezuela"},
{"VG", 18.420695, -64.639968, "British Virgin Islands"},
{"VI", 18.335765, -64.896335, "U.S. Virgin Islands"},
{"VN", 14.058324, 108.277199, "Vietnam"},
{"VU", -15.376706, 166.959158, "Vanuatu"},
{"WF", -13.768752, -177.156097, "Wallis and Futuna"},
{"WS", -13.759029, -172.104629, "Samoa"},
{"XK", 42.602636, 20.902977, "Kosovo"},
{"YE", 15.552727, 48.516388, "Yemen"},
{"YT", -12.8275, 45.166244, "Mayotte"},
{"ZA", -30.559482, 22.937506, "South Africa"},
{"ZM", -13.133897, 27.849332, "Zambia"},
{"ZW", -19.015438, 29.154857, "Zimbabwe"},
};
public static ScannableTable eval(boolean b) {
return new ScannableTable() {
@Override public Enumerable<@Nullable Object[]> scan(DataContext root) {
return Linq4j.asEnumerable(ROWS);
};
@Override public RelDataType getRowType(RelDataTypeFactory typeFactory) {
return typeFactory.builder()
.add("country", SqlTypeName.VARCHAR)
.add("latitude", SqlTypeName.DECIMAL).nullable(true)
.add("longitude", SqlTypeName.DECIMAL).nullable(true)
.add("name", SqlTypeName.VARCHAR)
.build();
}
@Override public Statistic getStatistic() {
return Statistics.of(246D,
ImmutableList.of(ImmutableBitSet.of(0), ImmutableBitSet.of(3)));
}
@Override public Schema.TableType getJdbcTableType() {
return Schema.TableType.TABLE;
}
@Override public boolean isRolledUp(String column) {
return false;
}
@Override public boolean rolledUpColumnValidInsideAgg(String column, SqlCall call,
@Nullable SqlNode parent, @Nullable CalciteConnectionConfig config) {
return false;
}
};
}
}