TDengine/tests/pytest/util/sub.py
Linhe Huo 5823bcfc3b
[TD-3048]<feature>: support lines/stream/query_a/stop_query/ and so on. (#7079)
* [TD-3048]<feature>: support lines/stream/query_a/stop_query/ and so on.

commit 0edc106a76a95b28e65019c2ee4e4ed84530ad35
Author: Huo Linhe <linhehuo@gmail.com>
Date:   Thu Jul 29 21:13:13 2021 +0800

    doc: improve document for python connector

commit 84915de0831b49c8378a16242dd0dbba8aaf386f
Author: Huo Linhe <linhehuo@gmail.com>
Date:   Thu Jul 29 20:35:45 2021 +0800

    chore: add time elapsed

commit 1e8822e01bda96388a359363776792e261260b88
Author: Huo Linhe <linhehuo@gmail.com>
Date:   Thu Jul 29 20:26:01 2021 +0800

    feat: support multi bind

commit 82d823f6194471fd3324b50f7dfba0ee9f10a7dd
Author: Huo Linhe <linhehuo@gmail.com>
Date:   Thu Jul 29 16:42:05 2021 +0800

    feat: python support for async query and subscribe with callback

commit 8b9d36de2945906821225189cb47958e153d81e2
Author: Huo Linhe <linhehuo@gmail.com>
Date:   Wed Jul 28 18:09:09 2021 +0800

    feat: finish stream and stmt interface

commit bc871aa43e9fc28dd1b3f8784a8ac6ee988564b5
Author: Huo Linhe <linhehuo@gmail.com>
Date:   Mon Jul 26 20:21:27 2021 +0800

    feat: basic full-fetured implementations

commit e5f7a5213e9016c377a57485e8d5c12139ce0957
Author: Huo Linhe <linhehuo@gmail.com>
Date:   Fri Jul 23 10:33:14 2021 +0800

    tmp: refactor

* chore: fix insert/line_insert.py error

* [TD-3048]<fix>: fix tests error

* [TD-3049]<feature>: support stop_query in python connector cursor

* [TD-3048]<doc>: improve python connector document

* [TD-3048]<doc>: improve python connection README

* [TD-3048]<hotfix>: fix python2 install and runtime error

* chore: replace insertLines with insert_lines

* chore: fix misspellings

* chore: fix crash gen error in threading

* feat: support __del__ method for simplify

* test: fix subscription test result check

* chore: compatible with 2.0.20

* chore: fix python connector subscription test case

* [TD-3048]<fix>: fix schemaless insert test

* [TD-3048]<fix>: fix memory leak in crash_gen test case

* [TD-3048]<fix>: minor fix for crash gen memory leak

* [TD-3048]<fix>: set minimal required python3 as 3.4

* chore: update version in setup.py

* [TD-3048]<fix>: fix runtime errors in python3.4

* [TD-3048]<fix>: add typing as a dependency
2021-08-07 08:00:37 +08:00

44 lines
1.2 KiB
Python

###################################################################
# Copyright (c) 2020 by TAOS Technologies, Inc.
# All rights reserved.
#
# This file is proprietary and confidential to TAOS Technologies.
# No part of this file may be reproduced, stored, transmitted,
# disclosed or used in any form or by any means other than as
# expressly provided by the written permission from Jianhui Tao
#
###################################################################
# -*- coding: utf-8 -*-
import sys
import os
import time
import datetime
from util.log import *
class TDSub:
def __init__(self):
self.consumedRows = 0
self.consumedCols = 0
def init(self, sub):
self.sub = sub
def close(self, keepProgress):
self.sub.close(keepProgress)
def consume(self):
self.result = self.sub.consume()
self.result.fetch_all()
self.consumedRows = self.result.row_count
self.consumedCols = self.result.field_count
return self.consumedRows
def checkRows(self, expectRows):
if self.consumedRows != expectRows:
tdLog.exit("consumed rows:%d != expect:%d" % (self.consumedRows, expectRows))
tdLog.info("consumed rows:%d == expect:%d" % (self.consumedRows, expectRows))
tdSub = TDSub()