mirror of
https://github.com/GreptimeTeam/greptimedb.git
synced 2025-12-25 15:40:02 +00:00
20dcaa6897
* test: for builtin functions * test: expect fail for `datetime()` * feat: add `interval()` fn(WIP) * feat: `interval()` fn in builtin(UNTEST) * refactor: move `py_vec_obj_to_array` to util.rs * style: fmt * test: simple `interval()` cases * test: `interval()` with `last()`&`first()` * doc: `ts` param of `interval()` * log: common_telemetry for logging in script crate * doc: corrsponding test fn for each .ron file * feat: change to`mpsc` for schedule_job * test: schedule_job * dep: rm rustpython dep in common-function * refactor: mv `schedule_job` into `Script` trait * test: change to use `interval` to sample datapoint * feat: add gen_none_array for generate None Array * feat: impl Missing value for `prev`&`next` * test: `sum(prev(values))` * doc: add comment for why not support Float16 in `prev()` * feat: add `interval` in py side mock module * style: cargo fmt * refactor: according to comments * refactor: extract `apply_interval_function` * style: cargo fmt * refactor: remove `schedule()` * style: cargo fmt
61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
from example.calc_rv import as_table, calc_rvs
|
|
from greptime import coprocessor, set_conn_addr, get_conn_addr, mock_tester
|
|
import sys
|
|
import json
|
|
import requests
|
|
'''
|
|
To run this script, you need to first start a http server of greptime, and
|
|
`
|
|
python3 component/script/python/test.py 地址:端口
|
|
`
|
|
|
|
'''
|
|
@coprocessor(sql='select number from numbers limit 10', args=['number'], returns=['n'])
|
|
def test(n):
|
|
return n+2
|
|
|
|
def init_table(close, open_time):
|
|
req_init = "/v1/sql?sql=create table k_line (close double, open_time bigint, TIME INDEX (open_time))"
|
|
print(get_db(req_init).text)
|
|
for c1, c2 in zip(close, open_time):
|
|
req = "/v1/sql?sql=INSERT INTO k_line(close, open_time) VALUES ({}, {})".format(c1, c2)
|
|
print(get_db(req).text)
|
|
print(get_db("/v1/sql?sql=select * from k_line").text)
|
|
|
|
def get_db(req:str):
|
|
return requests.get("http://{}{}".format(get_conn_addr(), req))
|
|
|
|
if __name__ == "__main__":
|
|
with open("component/script/python/example/kline.json", "r") as kline_file:
|
|
kline = json.load(kline_file)
|
|
table = as_table(kline["result"])
|
|
close = table["close"]
|
|
open_time = table["open_time"]
|
|
env = {"close":close, "open_time": open_time}
|
|
|
|
res = mock_tester(calc_rvs, env=env)
|
|
print("Mock result:", [i[0] for i in res])
|
|
exit()
|
|
if len(sys.argv)!=2:
|
|
raise Exception("Expect only one address as cmd's args")
|
|
set_conn_addr(sys.argv[1])
|
|
res = test()
|
|
print(res.headers)
|
|
print(res.text)
|
|
with open("component/script/python/example/kline.json", "r") as kline_file:
|
|
kline = json.load(kline_file)
|
|
# vec = vector([1,2,3], int)
|
|
# print(vec, vec.datatype())
|
|
table = as_table(kline["result"])
|
|
# print(table)
|
|
close = table["close"]
|
|
open_time = table["open_time"]
|
|
init_table(close, open_time)
|
|
|
|
real = calc_rvs()
|
|
print(real)
|
|
try:
|
|
print(real.text["error"])
|
|
except:
|
|
print(real.text)
|