def resource_close( conn: 'Connection', cursor: int, query_id: int = None ) -> APIResult: """ Closes a resource, such as query cursor. :param conn: connection to GridGain server, :param cursor: cursor ID, :param query_id: (optional) a value generated by client and returned as-is in response.query_id. When the parameter is omitted, a random value is generated, :return: API result data object. Contains zero status on success, non-zero status and an error description otherwise. """ query_struct = Query( OP_RESOURCE_CLOSE, [ ('cursor', Long), ], query_id=query_id, ) return query_struct.perform( conn, query_params={ 'cursor': cursor, }, )
def sql_cursor_get_page(conn: 'Connection', cursor: int) -> APIResult: """ Retrieves the next SQL query cursor page by cursor ID from `sql`. :param conn: connection to GridGain server, :param cursor: cursor ID, :return: API result data object. Contains zero status and a value of type dict with results on success, non-zero status and an error description otherwise. Value dict is of following format: * `data`: dict, result rows as key-value pairs, * `more`: bool, True if more data is available for subsequent ‘sql_cursor_get_page’ calls. """ query_struct = Query(OP_QUERY_SQL_CURSOR_GET_PAGE, [ ('cursor', Long), ]) result = query_struct.perform( conn, query_params={ 'cursor': cursor, }, response_config=[ ('data', Map), ('more', Bool), ], ) if result.status == 0: result.value = dict(result.value) return result
def cache_destroy( connection: 'Connection', cache: Union[str, int], query_id=None, ) -> 'APIResult': """ Destroys cache with a given name. :param connection: connection to GridGain server, :param cache: name or ID of the cache, :param query_id: (optional) a value generated by client and returned as-is in response.query_id. When the parameter is omitted, a random value is generated, :return: API result data object. """ query_struct = Query( OP_CACHE_DESTROY, [ ('hash_code', Int), ], query_id=query_id, ) return query_struct.perform( connection, query_params={ 'hash_code': cache_id(cache), }, )
def cache_clear( connection: 'Connection', cache: Union[str, int], binary: bool = False, query_id: Optional[int] = None, ) -> 'APIResult': """ Clears the cache without notifying listeners or cache writers. :param connection: connection to GridGain server, :param cache: name or ID of the cache, :param binary: (optional) pass True to keep the value in binary form. False by default, :param query_id: (optional) a value generated by client and returned as-is in response.query_id. When the parameter is omitted, a random value is generated, :return: API result data object. Contains zero status on success, non-zero status and an error description otherwise. """ query_struct = Query( OP_CACHE_CLEAR, [ ('hash_code', Int), ('flag', Byte), ], query_id=query_id, ) return query_struct.perform( connection, query_params={ 'hash_code': cache_id(cache), 'flag': 1 if binary else 0, }, )
def cache_get_or_create( connection: 'Connection', name: str, query_id=None, ) -> 'APIResult': """ Creates a cache with a given name. Does nothing if the cache exists. :param connection: connection to GridGain server, :param name: cache name, :param query_id: (optional) a value generated by client and returned as-is in response.query_id. When the parameter is omitted, a random value is generated, :return: API result data object. Contains zero status if a cache is created successfully, non-zero status and an error description otherwise. """ query_struct = Query( OP_CACHE_GET_OR_CREATE_WITH_NAME, [ ('cache_name', String), ], query_id=query_id, ) return query_struct.perform( connection, query_params={ 'cache_name': name, }, )
def cache_get_and_replace( connection: 'Connection', cache: Union[str, int], key: Any, value: Any, key_hint: 'GridGainDataType' = None, value_hint: 'GridGainDataType' = None, binary: bool = False, query_id: Optional[int] = None, ) -> 'APIResult': """ Puts a value with a given key to cache, returning previous value for that key, if and only if there is a value currently mapped for that key. :param connection: connection to GridGain server, :param cache: name or ID of the cache, :param key: key for the cache entry. Can be of any supported type, :param value: value for the key, :param key_hint: (optional) GridGain data type, for which the given key should be converted, :param value_hint: (optional) GridGain data type, for which the given value should be converted. :param binary: pass True to keep the value in binary form. False by default, :param query_id: a value generated by client and returned as-is in response.query_id. When the parameter is omitted, a random value is generated, :return: API result data object. Contains zero status and an old value or None on success, non-zero status and an error description otherwise. """ query_struct = Query( OP_CACHE_GET_AND_REPLACE, [ ('hash_code', Int), ('flag', Byte), ('key', key_hint or AnyDataObject), ('value', value_hint or AnyDataObject), ], query_id=query_id, ) result = query_struct.perform( connection, query_params={ 'hash_code': cache_id(cache), 'flag': 1 if binary else 0, 'key': key, 'value': value, }, response_config=[ ('value', AnyDataObject), ], ) if result.status == 0: result.value = result.value['value'] return result
def cache_remove_if_equals( connection: 'Connection', cache: Union[str, int], key: Any, sample: Any, key_hint: 'GridGainDataType' = None, sample_hint: 'GridGainDataType' = None, binary: bool = False, query_id: Optional[int] = None, ) -> 'APIResult': """ Removes an entry with a given key if provided value is equal to actual value, notifying listeners and cache writers. :param connection: connection to GridGain server, :param cache: name or ID of the cache, :param key: key for the cache entry, :param sample: a sample to compare the stored value with, :param key_hint: (optional) GridGain data type, for which the given key should be converted, :param sample_hint: (optional) GridGain data type, for whic the given sample should be converted :param binary: (optional) pass True to keep the value in binary form. False by default, :param query_id: (optional) a value generated by client and returned as-is in response.query_id. When the parameter is omitted, a random value is generated, :return: API result data object. Contains zero status and a boolean success code, or non-zero status and an error description if something has gone wrong. """ query_struct = Query( OP_CACHE_REMOVE_IF_EQUALS, [ ('hash_code', Int), ('flag', Byte), ('key', key_hint or AnyDataObject), ('sample', sample_hint or AnyDataObject), ], query_id=query_id, ) result = query_struct.perform( connection, query_params={ 'hash_code': cache_id(cache), 'flag': 1 if binary else 0, 'key': key, 'sample': sample, }, response_config=[ ('success', Bool), ], ) if result.status == 0: result.value = result.value['success'] return result
def sql_fields_cursor_get_page( conn: 'Connection', cursor: int, field_count: int, query_id: int = None, ) -> APIResult: """ Retrieves the next query result page by cursor ID from `sql_fields`. :param conn: connection to GridGain server, :param cursor: cursor ID, :param field_count: a number of fields in a row, :param query_id: (optional) a value generated by client and returned as-is in response.query_id. When the parameter is omitted, a random value is generated, :return: API result data object. Contains zero status and a value of type dict with results on success, non-zero status and an error description otherwise. Value dict is of following format: * `data`: list, result values, * `more`: bool, True if more data is available for subsequent ‘sql_fields_cursor_get_page’ calls. """ query_struct = Query( OP_QUERY_SQL_FIELDS_CURSOR_GET_PAGE, [ ('cursor', Long), ], query_id=query_id, ) result = query_struct.perform( conn, query_params={ 'cursor': cursor, }, response_config=[ ('data', StructArray([ ('field_{}'.format(i), AnyDataObject) for i in range(field_count) ])), ('more', Bool), ] ) if result.status != 0: return result value = result.value result.value = { 'data': [], 'more': value['more'] } for row_dict in value['data']: row = [] for field_key in sorted(row_dict.keys()): row.append(row_dict[field_key]) result.value['data'].append(row) return result
def cache_get_size( connection: 'Connection', cache: Union[str, int], peek_modes: int = 0, binary: bool = False, query_id: Optional[int] = None, ) -> 'APIResult': """ Gets the number of entries in cache. :param connection: connection to GridGain server, :param cache: name or ID of the cache, :param peek_modes: (optional) limit count to near cache partition (PeekModes.NEAR), primary cache (PeekModes.PRIMARY), or backup cache (PeekModes.BACKUP). Defaults to all cache partitions (PeekModes.ALL), :param binary: (optional) pass True to keep the value in binary form. False by default, :param query_id: (optional) a value generated by client and returned as-is in response.query_id. When the parameter is omitted, a random value is generated, :return: API result data object. Contains zero status and a number of cache entries on success, non-zero status and an error description otherwise. """ if not isinstance(peek_modes, (list, tuple)): if peek_modes == 0: peek_modes = [] else: peek_modes = [peek_modes] query_struct = Query( OP_CACHE_GET_SIZE, [ ('hash_code', Int), ('flag', Byte), ('peek_modes', PeekModes), ], query_id=query_id, ) result = query_struct.perform( connection, query_params={ 'hash_code': cache_id(cache), 'flag': 1 if binary else 0, 'peek_modes': peek_modes, }, response_config=[ ('count', Long), ], ) if result.status == 0: result.value = result.value['count'] return result
def cache_get( connection: 'Connection', cache: Union[str, int], key: Any, key_hint: 'GridGainDataType' = None, binary: bool = False, query_id: Optional[int] = None, ) -> 'APIResult': """ Retrieves a value from cache by key. :param connection: connection to GridGain server, :param cache: name or ID of the cache, :param key: key for the cache entry. Can be of any supported type, :param key_hint: (optional) GridGain data type, for which the given key should be converted, :param binary: (optional) pass True to keep the value in binary form. False by default, :param query_id: (optional) a value generated by client and returned as-is in response.query_id. When the parameter is omitted, a random value is generated, :return: API result data object. Contains zero status and a value retrieved on success, non-zero status and an error description on failure. """ query_struct = Query( OP_CACHE_GET, [ ('hash_code', Int), ('flag', Byte), ('key', key_hint or AnyDataObject), ], query_id=query_id, ) result = query_struct.perform( connection, query_params={ 'hash_code': cache_id(cache), 'flag': 1 if binary else 0, 'key': key, }, response_config=[ ('value', AnyDataObject), ], ) if result.status != 0: return result result.value = result.value['value'] return result
def cache_put( connection: 'Connection', cache: Union[str, int], key: Any, value: Any, key_hint: 'GridGainDataType' = None, value_hint: 'GridGainDataType' = None, binary: bool = False, query_id: Optional[int] = None, ) -> 'APIResult': """ Puts a value with a given key to cache (overwriting existing value if any). :param connection: connection to GridGain server, :param cache: name or ID of the cache, :param key: key for the cache entry. Can be of any supported type, :param value: value for the key, :param key_hint: (optional) GridGain data type, for which the given key should be converted, :param value_hint: (optional) GridGain data type, for which the given value should be converted. :param binary: (optional) pass True to keep the value in binary form. False by default, :param query_id: (optional) a value generated by client and returned as-is in response.query_id. When the parameter is omitted, a random value is generated, :return: API result data object. Contains zero status if a value is written, non-zero status and an error description otherwise. """ query_struct = Query( OP_CACHE_PUT, [ ('hash_code', Int), ('flag', Byte), ('key', key_hint or AnyDataObject), ('value', value_hint or AnyDataObject), ], query_id=query_id, ) return query_struct.perform( connection, { 'hash_code': cache_id(cache), 'flag': 1 if binary else 0, 'key': key, 'value': value, })
def cache_contains_keys( connection: 'Connection', cache: Union[str, int], keys: Iterable, binary: bool = False, query_id: Optional[int] = None, ) -> 'APIResult': """ Returns a value indicating whether all given keys are present in cache. :param connection: connection to GridGain server, :param cache: name or ID of the cache, :param keys: a list of keys or (key, type hint) tuples, :param binary: pass True to keep the value in binary form. False by default, :param query_id: a value generated by client and returned as-is in response.query_id. When the parameter is omitted, a random value is generated, :return: API result data object. Contains zero status and a bool value retrieved on success: `True` when all keys are present, `False` otherwise, non-zero status and an error description on failure. """ query_struct = Query( OP_CACHE_CONTAINS_KEYS, [ ('hash_code', Int), ('flag', Byte), ('keys', AnyDataArray()), ], query_id=query_id, ) result = query_struct.perform( connection, query_params={ 'hash_code': cache_id(cache), 'flag': 1 if binary else 0, 'keys': keys, }, response_config=[ ('value', Bool), ], ) if result.status == 0: result.value = result.value['value'] return result
def cache_get_all( connection: 'Connection', cache: Union[str, int], keys: Iterable, binary: bool = False, query_id: Optional[int] = None, ) -> 'APIResult': """ Retrieves multiple key-value pairs from cache. :param connection: connection to GridGain server, :param cache: name or ID of the cache, :param keys: list of keys or tuples of (key, key_hint), :param binary: (optional) pass True to keep the value in binary form. False by default, :param query_id: (optional) a value generated by client and returned as-is in response.query_id. When the parameter is omitted, a random value is generated, :return: API result data object. Contains zero status and a dict, made of retrieved key-value pairs, non-zero status and an error description on failure. """ query_struct = Query( OP_CACHE_GET_ALL, [ ('hash_code', Int), ('flag', Byte), ('keys', AnyDataArray()), ], query_id=query_id, ) result = query_struct.perform( connection, query_params={ 'hash_code': cache_id(cache), 'flag': 1 if binary else 0, 'keys': keys, }, response_config=[ ('data', Map), ], ) if result.status == 0: result.value = dict(result.value)['data'] return result
def scan_cursor_get_page( conn: 'Connection', cursor: int, query_id: int = None, ) -> APIResult: """ Fetches the next scan query cursor page by cursor ID that is obtained from `scan` function. :param conn: connection to GridGain server, :param cursor: cursor ID, :param query_id: (optional) a value generated by client and returned as-is in response.query_id. When the parameter is omitted, a random value is generated, :return: API result data object. Contains zero status and a value of type dict with results on success, non-zero status and an error description otherwise. Value dict is of following format: * `data`: dict, result rows as key-value pairs, * `more`: bool, True if more data is available for subsequent ‘scan_cursor_get_page’ calls. """ query_struct = Query( OP_QUERY_SCAN_CURSOR_GET_PAGE, [ ('cursor', Long), ], query_id=query_id, ) result = query_struct.perform( conn, query_params={ 'cursor': cursor, }, response_config=[ ('data', Map), ('more', Bool), ], ) if result.status == 0: result.value = dict(result.value) return result
def cache_put_all( connection: 'Connection', cache: Union[str, int], pairs: dict, binary: bool = False, query_id: Optional[int] = None, ) -> 'APIResult': """ Puts multiple key-value pairs to cache (overwriting existing associations if any). :param connection: connection to GridGain server, :param cache: name or ID of the cache, :param pairs: dictionary type parameters, contains key-value pairs to save. Each key or value can be an item of representable Python type or a tuple of (item, hint), :param binary: (optional) pass True to keep the value in binary form. False by default, :param query_id: (optional) a value generated by client and returned as-is in response.query_id. When the parameter is omitted, a random value is generated, :return: API result data object. Contains zero status if key-value pairs are written, non-zero status and an error description otherwise. """ query_struct = Query( OP_CACHE_PUT_ALL, [ ('hash_code', Int), ('flag', Byte), ('data', Map), ], query_id=query_id, ) return query_struct.perform( connection, query_params={ 'hash_code': cache_id(cache), 'flag': 1 if binary else 0, 'data': pairs, }, )
def cache_get_configuration( connection: 'Connection', cache: Union[str, int], flags: int = 0, query_id=None, ) -> 'APIResult': """ Gets configuration for the given cache. :param connection: connection to GridGain server, :param cache: name or ID of the cache, :param flags: GridGain documentation is unclear on this subject, :param query_id: (optional) a value generated by client and returned as-is in response.query_id. When the parameter is omitted, a random value is generated, :return: API result data object. Result value is OrderedDict with the cache configuration parameters. """ query_struct = Query( OP_CACHE_GET_CONFIGURATION, [ ('hash_code', Int), ('flags', Byte), ], query_id=query_id, ) result = query_struct.perform( connection, query_params={ 'hash_code': cache_id(cache), 'flags': flags, }, response_config=[ ('cache_config', cache_config_struct), ], ) if result.status == 0: result.value = compact_cache_config(result.value['cache_config']) return result
def cache_remove_keys( connection: 'Connection', cache: Union[str, int], keys: Iterable, binary: bool = False, query_id: Optional[int] = None, ) -> 'APIResult': """ Removes entries with given keys, notifying listeners and cache writers. :param connection: connection to GridGain server, :param cache: name or ID of the cache, :param keys: list of keys or tuples of (key, key_hint), :param binary: (optional) pass True to keep the value in binary form. False by default, :param query_id: (optional) a value generated by client and returned as-is in response.query_id. When the parameter is omitted, a random value is generated, :return: API result data object. Contains zero status on success, non-zero status and an error description otherwise. """ query_struct = Query( OP_CACHE_REMOVE_KEYS, [ ('hash_code', Int), ('flag', Byte), ('keys', AnyDataArray()), ], query_id=query_id, ) return query_struct.perform( connection, query_params={ 'hash_code': cache_id(cache), 'flag': 1 if binary else 0, 'keys': keys, }, )
def cache_get_names(connection: 'Connection', query_id=None) -> 'APIResult': """ Gets existing cache names. :param connection: connection to GridGain server, :param query_id: (optional) a value generated by client and returned as-is in response.query_id. When the parameter is omitted, a random value is generated, :return: API result data object. Contains zero status and a list of cache names, non-zero status and an error description otherwise. """ query_struct = Query(OP_CACHE_GET_NAMES, query_id=query_id) result = query_struct.perform( connection, response_config=[ ('cache_names', StringArray), ], ) if result.status == 0: result.value = result.value['cache_names'] return result
def put_binary_type( connection: 'Connection', type_name: str, affinity_key_field: str=None, is_enum=False, schema: dict=None, query_id=None, ) -> APIResult: """ Registers binary type information in cluster. :param connection: connection to GridGain server, :param type_name: name of the data type being registered, :param affinity_key_field: (optional) name of the affinity key field, :param is_enum: (optional) register enum if True, binary object otherwise. Defaults to False, :param schema: (optional) when register enum, pass a dict of enumerated parameter names as keys and an integers as values. When register binary type, pass a dict of field names: field types. Binary type with no fields is OK, :param query_id: (optional) a value generated by client and returned as-is in response.query_id. When the parameter is omitted, a random value is generated, :return: API result data object. """ # prepare data if schema is None: schema = {} type_id = entity_id(type_name) data = { 'type_name': type_name, 'type_id': type_id, 'affinity_key_field': affinity_key_field, 'binary_fields': [], 'is_enum': is_enum, 'schema': [], } schema_id = None if is_enum: data['enums'] = [] for literal, ordinal in schema.items(): data['enums'].append({ 'literal': literal, 'type_id': ordinal, }) else: # assemble schema and calculate schema ID in one go schema_id = FNV1_OFFSET_BASIS if schema else 0 for field_name, data_type in schema.items(): # TODO: check for allowed data types field_id = entity_id(field_name) data['binary_fields'].append({ 'field_name': field_name, 'type_id': int.from_bytes( data_type.type_code, byteorder=PROTOCOL_BYTE_ORDER ), 'field_id': field_id, }) schema_id ^= (field_id & 0xff) schema_id = int_overflow(schema_id * FNV1_PRIME) schema_id ^= ((field_id >> 8) & 0xff) schema_id = int_overflow(schema_id * FNV1_PRIME) schema_id ^= ((field_id >> 16) & 0xff) schema_id = int_overflow(schema_id * FNV1_PRIME) schema_id ^= ((field_id >> 24) & 0xff) schema_id = int_overflow(schema_id * FNV1_PRIME) data['schema'].append({ 'schema_id': schema_id, 'schema_fields': [ {'schema_field_id': entity_id(x)} for x in schema ], }) # do query if is_enum: query_struct = Query( OP_PUT_BINARY_TYPE, [ ('type_id', Int), ('type_name', String), ('affinity_key_field', String), ('binary_fields', binary_fields_struct), ('is_enum', Bool), ('enums', enum_struct), ('schema', schema_struct), ], query_id=query_id, ) else: query_struct = Query( OP_PUT_BINARY_TYPE, [ ('type_id', Int), ('type_name', String), ('affinity_key_field', String), ('binary_fields', binary_fields_struct), ('is_enum', Bool), ('schema', schema_struct), ], query_id=query_id, ) result = query_struct.perform(connection, query_params=data) if result.status == 0: result.value = { 'type_id': type_id, 'schema_id': schema_id, } return result
def cache_get_node_partitions( conn: 'Connection', caches: Union[int, Iterable[int]], query_id: int = None, ) -> APIResult: """ Gets partition mapping for a GridGain cache or a number of caches. See “IEP-23: Best Effort Affinity for thin clients”. :param conn: connection to GridGain server, :param caches: cache ID(s) the mapping is provided for, :param query_id: (optional) a value generated by client and returned as-is in response.query_id. When the parameter is omitted, a random value is generated, :return: API result data object. """ query_struct = Query(OP_CACHE_PARTITIONS, [ ('cache_ids', cache_ids), ], query_id=query_id) if not is_iterable(caches): caches = [caches] result = query_struct.perform( conn, query_params={ 'cache_ids': [{ 'cache_id': cache } for cache in caches], }, response_config=[ ('version_major', Long), ('version_minor', Int), ('partition_mapping', partition_mapping), ], ) if result.status == 0: # tidying up the result value = { 'version': (result.value['version_major'], result.value['version_minor']), 'partition_mapping': [], } for i, partition_map in enumerate(result.value['partition_mapping']): cache_id = partition_map['cache_mapping'][0]['cache_id'] value['partition_mapping'].insert( i, { 'cache_id': cache_id, 'is_applicable': partition_map['is_applicable'], }) if partition_map['is_applicable']: value['partition_mapping'][i]['cache_config'] = { a['key_type_id']: a['affinity_key_field_id'] for a in partition_map['cache_mapping'][0]['cache_config'] } value['partition_mapping'][i]['node_mapping'] = { p['node_uuid']: [x['partition_id'] for x in p['node_partitions']] for p in partition_map['node_mapping'] } result.value = value return result
def cache_local_peek( conn: 'Connection', cache: Union[str, int], key: Any, key_hint: 'GridGainDataType' = None, peek_modes: int = 0, binary: bool = False, query_id: Optional[int] = None, ) -> 'APIResult': """ Peeks at in-memory cached value using default optional peek mode. This method will not load value from any persistent store or from a remote node. :param conn: connection: connection to GridGain server, :param cache: name or ID of the cache, :param key: entry key, :param key_hint: (optional) GridGain data type, for which the given key should be converted, :param peek_modes: (optional) limit count to near cache partition (PeekModes.NEAR), primary cache (PeekModes.PRIMARY), or backup cache (PeekModes.BACKUP). Defaults to all cache partitions (PeekModes.ALL), :param binary: (optional) pass True to keep the value in binary form. False by default, :param query_id: (optional) a value generated by client and returned as-is in response.query_id. When the parameter is omitted, a random value is generated, :return: API result data object. Contains zero status and a peeked value (null if not found). """ if not isinstance(peek_modes, (list, tuple)): if peek_modes == 0: peek_modes = [] else: peek_modes = [peek_modes] query_struct = Query( OP_CACHE_LOCAL_PEEK, [ ('hash_code', Int), ('flag', Byte), ('key', key_hint or AnyDataObject), ('peek_modes', PeekModes), ], query_id=query_id, ) result = query_struct.perform( conn, query_params={ 'hash_code': cache_id(cache), 'flag': 1 if binary else 0, 'key': key, 'peek_modes': peek_modes, }, response_config=[ ('value', AnyDataObject), ], ) if result.status != 0: return result result.value = result.value['value'] return result
def sql( conn: 'Connection', cache: Union[str, int], table_name: str, query_str: str, page_size: int, query_args=None, distributed_joins: bool = False, replicated_only: bool = False, local: bool = False, timeout: int = 0, binary: bool = False, query_id: int = None ) -> APIResult: """ Executes an SQL query over data stored in the cluster. The query returns the whole record (key and value). :param conn: connection to GridGain server, :param cache: name or ID of the cache, :param table_name: name of a type or SQL table, :param query_str: SQL query string, :param page_size: cursor page size, :param query_args: (optional) query arguments, :param distributed_joins: (optional) distributed joins. Defaults to False, :param replicated_only: (optional) whether query contains only replicated tables or not. Defaults to False, :param local: (optional) pass True if this query should be executed on local node only. Defaults to False, :param timeout: (optional) non-negative timeout value in ms. Zero disables timeout (default), :param binary: (optional) pass True to keep the value in binary form. False by default, :param query_id: (optional) a value generated by client and returned as-is in response.query_id. When the parameter is omitted, a random value is generated, :return: API result data object. Contains zero status and a value of type dict with results on success, non-zero status and an error description otherwise. Value dict is of following format: * `cursor`: int, cursor ID, * `data`: dict, result rows as key-value pairs, * `more`: bool, True if more data is available for subsequent ‘sql_get_page’ calls. """ if query_args is None: query_args = [] query_struct = Query( OP_QUERY_SQL, [ ('hash_code', Int), ('flag', Byte), ('table_name', String), ('query_str', String), ('query_args', AnyDataArray()), ('distributed_joins', Bool), ('local', Bool), ('replicated_only', Bool), ('page_size', Int), ('timeout', Long), ], query_id=query_id, ) result = query_struct.perform( conn, query_params={ 'hash_code': cache_id(cache), 'flag': 1 if binary else 0, 'table_name': table_name, 'query_str': query_str, 'query_args': query_args, 'distributed_joins': 1 if distributed_joins else 0, 'local': 1 if local else 0, 'replicated_only': 1 if replicated_only else 0, 'page_size': page_size, 'timeout': timeout, }, response_config=[ ('cursor', Long), ('data', Map), ('more', Bool), ], ) if result.status == 0: result.value = dict(result.value) return result
def sql_fields( conn: 'Connection', cache: Union[str, int], query_str: str, page_size: int, query_args=None, schema: str = None, statement_type: int = StatementType.ANY, distributed_joins: bool = False, local: bool = False, replicated_only: bool = False, enforce_join_order: bool = False, collocated: bool = False, lazy: bool = False, include_field_names: bool = False, max_rows: int = -1, timeout: int = 0, binary: bool = False, query_id: int = None ) -> APIResult: """ Performs SQL fields query. :param conn: connection to GridGain server, :param cache: name or ID of the cache, :param query_str: SQL query string, :param page_size: cursor page size, :param query_args: (optional) query arguments. List of values or (value, type hint) tuples, :param schema: (optional) schema for the query. Defaults to `PUBLIC`, :param statement_type: (optional) statement type. Can be: * StatementType.ALL − any type (default), * StatementType.SELECT − select, * StatementType.UPDATE − update. :param distributed_joins: (optional) distributed joins. Defaults to False, :param local: (optional) pass True if this query should be executed on local node only. Defaults to False, :param replicated_only: (optional) whether query contains only replicated tables or not. Defaults to False, :param enforce_join_order: (optional) enforce join order. Defaults to False, :param collocated: (optional) whether your data is co-located or not. Defaults to False, :param lazy: (optional) lazy query execution. Defaults to False, :param include_field_names: (optional) include field names in result. Defaults to False, :param max_rows: (optional) query-wide maximum of rows. Defaults to -1 (all rows), :param timeout: (optional) non-negative timeout value in ms. Zero disables timeout (default), :param binary: (optional) pass True to keep the value in binary form. False by default, :param query_id: (optional) a value generated by client and returned as-is in response.query_id. When the parameter is omitted, a random value is generated, :return: API result data object. Contains zero status and a value of type dict with results on success, non-zero status and an error description otherwise. Value dict is of following format: * `cursor`: int, cursor ID, * `data`: list, result values, * `more`: bool, True if more data is available for subsequent ‘sql_fields_cursor_get_page’ calls. """ if query_args is None: query_args = [] query_struct = Query( OP_QUERY_SQL_FIELDS, [ ('hash_code', Int), ('flag', Byte), ('schema', String), ('page_size', Int), ('max_rows', Int), ('query_str', String), ('query_args', AnyDataArray()), ('statement_type', StatementType), ('distributed_joins', Bool), ('local', Bool), ('replicated_only', Bool), ('enforce_join_order', Bool), ('collocated', Bool), ('lazy', Bool), ('timeout', Long), ('include_field_names', Bool), ], query_id=query_id, ) return query_struct.perform( conn, query_params={ 'hash_code': cache_id(cache), 'flag': 1 if binary else 0, 'schema': schema, 'page_size': page_size, 'max_rows': max_rows, 'query_str': query_str, 'query_args': query_args, 'statement_type': statement_type, 'distributed_joins': distributed_joins, 'local': local, 'replicated_only': replicated_only, 'enforce_join_order': enforce_join_order, 'collocated': collocated, 'lazy': lazy, 'timeout': timeout, 'include_field_names': include_field_names, }, sql=True, include_field_names=include_field_names, has_cursor=True, )
def scan( conn: 'Connection', cache: Union[str, int], page_size: int, partitions: int = -1, local: bool = False, binary: bool = False, query_id: int = None, ) -> APIResult: """ Performs scan query. :param conn: connection to GridGain server, :param cache: name or ID of the cache, :param page_size: cursor page size, :param partitions: (optional) number of partitions to query (negative to query entire cache), :param local: (optional) pass True if this query should be executed on local node only. Defaults to False, :param binary: (optional) pass True to keep the value in binary form. False by default, :param query_id: (optional) a value generated by client and returned as-is in response.query_id. When the parameter is omitted, a random value is generated, :return: API result data object. Contains zero status and a value of type dict with results on success, non-zero status and an error description otherwise. Value dict is of following format: * `cursor`: int, cursor ID, * `data`: dict, result rows as key-value pairs, * `more`: bool, True if more data is available for subsequent ‘scan_cursor_get_page’ calls. """ query_struct = Query( OP_QUERY_SCAN, [ ('hash_code', Int), ('flag', Byte), ('filter', Null), ('page_size', Int), ('partitions', Int), ('local', Bool), ], query_id=query_id, ) result = query_struct.perform( conn, query_params={ 'hash_code': cache_id(cache), 'flag': 1 if binary else 0, 'filter': None, 'page_size': page_size, 'partitions': partitions, 'local': 1 if local else 0, }, response_config=[ ('cursor', Long), ('data', Map), ('more', Bool), ], ) if result.status == 0: result.value = dict(result.value) return result