#

# Copyright 2022 GridGain Systems, Inc. and Contributors.

#

# Licensed under the GridGain Community Edition License (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

#

# https://www.gridgain.com/products/software/community-edition/gridgain-community-edition-license

#

# 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.

#

import asyncio

from helpers.sql_helper import TableNames, Query, TestData

from pygridgain import AioClient

async def main():

# establish connection

client = AioClient()

async with client.connect('127.0.0.1', 10800):

# create tables

for query in [

Query.COUNTRY_CREATE_TABLE,

Query.CITY_CREATE_TABLE,

Query.LANGUAGE_CREATE_TABLE,

]:

await client.sql(query)

# create indices

for query in [Query.CITY_CREATE_INDEX, Query.LANGUAGE_CREATE_INDEX]:

await client.sql(query)

# load data concurrently.

await asyncio.gather(*[

client.sql(Query.COUNTRY_INSERT, query_args=row) for row in TestData.COUNTRY

])

await asyncio.gather(*[

client.sql(Query.CITY_INSERT, query_args=row) for row in TestData.CITY

])

await asyncio.gather(*[

client.sql(Query.LANGUAGE_INSERT, query_args=row) for row in TestData.LANGUAGE

])

# 10 most populated cities (with pagination)

async with client.sql('SELECT name, population FROM City ORDER BY population DESC LIMIT 10') as cursor:

print('Most 10 populated cities:')

async for row in cursor:

print(row)

# Most 10 populated cities:

# ['Mumbai (Bombay)', 10500000]

# ['Shanghai', 9696300]

# ['New York', 8008278]

# ['Peking', 7472000]

# ['Delhi', 7206704]

# ['Chongqing', 6351600]

# ['Tianjin', 5286800]

# ['Calcutta [Kolkata]', 4399819]

# ['Wuhan', 4344600]

# ['Harbin', 4289800]

print('-' * 20)

# 10 most populated cities in 3 countries (with pagination and header row)

most_populated_in_3_countries = '''

SELECT country.name as country_name, city.name as city_name, MAX(city.population) AS max_pop FROM country

JOIN city ON city.countrycode = country.code

WHERE country.code IN ('USA','IND','CHN')

GROUP BY country.name, city.name ORDER BY max_pop DESC LIMIT 10

'''

async with client.sql(most_populated_in_3_countries, include_field_names=True) as cursor:

print('Most 10 populated cities in USA, India and China:')

table_str_pattern = '{:15}\t| {:20}\t| {}'

print(table_str_pattern.format(*await cursor.__anext__()))

print('*' * 50)

async for row in cursor:

print(table_str_pattern.format(*row))

# Most 10 populated cities in USA, India and China:

# COUNTRY_NAME | CITY_NAME | MAX_POP

# **************************************************

# India | Mumbai (Bombay) | 10500000

# China | Shanghai | 9696300

# United States | New York | 8008278

# China | Peking | 7472000

# India | Delhi | 7206704

# China | Chongqing | 6351600

# China | Tianjin | 5286800

# India | Calcutta [Kolkata] | 4399819

# China | Wuhan | 4344600

# China | Harbin | 4289800

print('-' * 20)

# show city info

async with client.sql('SELECT * FROM City WHERE id = ?', query_args=[3802], include_field_names=True) as cursor:

field_names = await cursor.__anext__()

field_data = await cursor.__anext__()

print('City info:')

for field_name, field_value in zip(field_names * len(field_data), field_data):

print('{}: {}'.format(field_name, field_value))

# City info:

# ID: 3802

# NAME: Detroit

# COUNTRYCODE: USA

# DISTRICT: Michigan

# POPULATION: 951270

# clean up concurrently.

await asyncio.gather(*[

client.sql(Query.DROP_TABLE.format(table_name.value)) for table_name in TableNames

])

if __name__ == '__main__':

asyncio.run(main())