local driver = require "luaconnector"

local config = {
 host = "127.0.0.1",
 port = 6030,
 database = "",
 user = "root",
 password = "taosdata",
 max_packet_size = 1024 * 1024 
}

local conn
local res = driver.connect(config)
if res.code ~=0 then
 print("connect--- failed: "..res.error)
 return
else
 conn = res.conn
 print("connect--- pass.")
end

local res = driver.query(conn,"drop database if exists demo")

res = driver.query(conn,"create database demo")
if res.code ~=0 then
 print("create db--- failed: "..res.error)
 return
else
 print("create db--- pass.")
end

res = driver.query(conn,"use demo")
if res.code ~=0 then
 print("select db--- failed: "..res.error)
 return
else
 print("select db--- pass.")
end

res = driver.query(conn,"create table m1 (ts timestamp, speed int, owner binary(20), mark nchar(30))")
if res.code ~=0 then
 print("create table---failed: "..res.error)
 return
else
 print("create table--- pass.")
end

res = driver.query(conn,"insert into m1 values ('2019-09-01 00:00:00.001', 0, 'robotspace', '世界人民大团结万岁'), ('2019-09-01 00:00:00.002', 1, 'Hilink', '鬥鬯鬲鬼魚鳥鹵鹿麥麻黃黍黑黹黽鼎鼓鼠鼻齊齒龍龜龠'),('2019-09-01 00:00:00.003', 2, 'Harmony', '₠₡₢₣₤₥₦₧Rs₩₪₫€₭₮₯₰₱₲₳₴₵')")
if res.code ~=0 then
 print("insert records failed: "..res.error)
 return
else
 if(res.affected == 3) then
 print("insert records--- pass")
 else
 print("insert records---failed: expect 3 affected records, actually affected "..res.affected)
 end 
end

res = driver.query(conn,"select * from m1")

if res.code ~=0 then
 print("select failed: "..res.error)
 return
else
 if (#(res.item) == 3) then
 print("select--- pass")
	print(res.item[1].mark)
	print(res.item[2].mark)
	print(res.item[3].mark)

 else
	print("select--- failed: expect 3 affected records, actually received "..#(res.item))
 end

end

res = driver.query(conn,"CREATE TABLE thermometer (ts timestamp, degree double) TAGS(location binary(20), type int)")
if res.code ~=0 then
 print(res.error)
 return
else
 print("create super table--- pass")
end
res = driver.query(conn,"CREATE TABLE therm1 USING thermometer TAGS ('beijing', 1)")
if res.code ~=0 then
 print(res.error)
 return
else
 print("create table--- pass")
end

res = driver.query(conn,"INSERT INTO therm1 VALUES ('2019-09-01 00:00:00.001', 20),('2019-09-01 00:00:00.002', 21)")

if res.code ~=0 then
 print(res.error)
 return
else
 if(res.affected == 2) then
 print("insert records--- pass")
 else
 print("insert records---failed: expect 2 affected records, actually affected "..res.affected)
 end 
end

res = driver.query(conn,"SELECT COUNT(*) count, AVG(degree) AS av, MAX(degree), MIN(degree) FROM thermometer WHERE location='beijing' or location='tianjin' GROUP BY location, type")
if res.code ~=0 then
 print("select from super table--- failed:"..res.error)
 return
else
 print("select from super table--- pass")
 for i = 1, #(res.item) do
 print("res:"..res.item[i].count)
 end
end

function async_query_callback(res)
 if res.code ~=0 then
 print("async_query_callback--- failed:"..res.error)
 return
 else

 if(res.affected == 3) then
 print("async_query_callback, insert records--- pass")
 else
 print("async_query_callback, insert records---failed: expect 3 affected records, actually affected "..res.affected)
 end 

 end
end

driver.query_a(conn,"INSERT INTO therm1 VALUES ('2019-09-01 00:00:00.005', 100),('2019-09-01 00:00:00.006', 101),('2019-09-01 00:00:00.007', 102)", async_query_callback)


function stream_callback(t)
 print("------------------------")
 print("continuous query result:")
 for key, value in pairs(t) do
 print("key:"..key..", value:"..value)
 end
end

local stream
res = driver.open_stream(conn,"SELECT COUNT(*) as count, AVG(degree) as avg, MAX(degree) as max, MIN(degree) as min FROM thermometer interval(2s) sliding(2s);)",0, stream_callback)
if res.code ~=0 then
 print("open stream--- failed:"..res.error)
 return
else
 print("open stream--- pass")
 stream = res.stream
end

print("From now on we start continous insert in an definite (infinite if you want) loop.")
local loop_index = 0
while loop_index < 30 do
 local t = os.time()*1000
 local v = loop_index
 res = driver.query(conn,string.format("INSERT INTO therm1 VALUES (%d, %d)",t,v))

 if res.code ~=0 then
 print("continous insertion--- failed:" .. res.error)
 return
 else
 --print("insert successfully, affected:"..res.affected)
 end
 os.execute("sleep " .. 1)
 loop_index = loop_index + 1
end

driver.close_stream(stream)

driver.close(conn)