require("pgsql")

conn = pgsql.connect("hostaddr='10.6.6.1' port='5432' dbname='gmtest' user='gmtest' password='z0rteh' sslmode='disable'")

function conn:onInitialConnect()
    print("Initial connection made")
end

function conn:onConnect()
    print("Connection made")
end

function conn:onError(query, err)
    ErrorNoHalt("pqtest: error in query '" .. query .. "';\n   " .. err:gsub("\n", "\n   "):sub(1, -4))
end

concommand.Add("pgsql_reset", function()
    if IsValid(ply) then return end
    conn:reset()
end )

concommand.Add("pgsql_query", function(ply, cmd, arg)
    if IsValid(ply) then return end

    conn:query(arg[1], function(...)
        for k, result in ipairs({...}) do
            if result:success() then
                print(unpack(result.columnNames))

                for row in result do
                    print(unpack(row))
                end
            end
        end
    end)
end )

concommand.Add("pgsql_test", function()
    conn:query("SELECT generate_series(1, 2000) as \"Test a!\"", function(res)
        if not res:success() then return print("UH OH") end

        local sum = 0

        for row in res do
            sum = sum + row["Test a!"]
        end

        print(sum)
    end)
end )

concommand.Add("pgsql_multitest", function()
    for i = 1, 100 do
        conn:query(Format("SELECT %i as A", i), function(res)
            if not res:success() then
                print(res.error)
                return
            end

            print(i, tonumber(res:row(1)[1]) == i )
        end)
    end
end )