Support some of SQL execution API

хорошая новость: получилось достать box.prepare (хоть его и нет в module.h но экспортируется без проблем). Результат:

pub fn test_sql_prepare() {
    let mut sp = Space::builder("TEST1")
        .field(Field::unsigned("id"))
        .create().unwrap();

    sp.index_builder("primary")
        .index_type(IndexType::Tree)
        .part(1)
        .create().unwrap();

    let sql = "SELECT * FROM TEST1";

    let b = Vec::<u8>::with_capacity(1000);

    let result = unsafe {
        ffi::tarantool::sql_prepare(sql.as_ptr() as *const c_char, sql.len() as u32, b.as_ptr() as *const c_char)
    };
    let e =  tarantool::error::TarantoolErrorCode::try_last();

    assert_eq!(e, None);
    assert_eq!(result, 0);
}

плохая новость: prepare работает с такой inner штукой тарантула как port, я так понял это такая мегаабстракция над чем то что можно представить как lua таблицу (или вообще над чем то что экспортит свой интерфейс куда то). Вобщем полиморфизм такой. Разбор этого port (sqlной его реализации) пока что представляется делом довольно сложным и видимо очень зависимым от внутреннего устройства тарантула (вобщем большая вероятность локнуться на конкретную версию) тк надо лезть сильно в кишки.

При условии что получится адеватно работать с port_sql, с box.execute проблем уже не будет.

Путем различных хаков получилось выполнить сценарий:

создать prepared stmt
вызвать execute
получить результат в формате msgpack

Примерный код для ознакомления:

pub fn test_sql_prepare() {
//подготавливаем тестовые данные
    let mut sp = Space::builder("TEST1")
        .field(Field::unsigned("id"))
        .field(Field::string("value"))
        .create().unwrap();

    sp.index_builder("primary")
        .index_type(IndexType::Tree)
        .part(1)
        .create().unwrap();
    sp.insert(&(1, "one")).unwrap();
    sp.insert(&(2, "two")).unwrap();
    sp.insert(&(3, "three")).unwrap();
    sp.insert(&(4, "four")).unwrap();

    let sql = "SELECT * FROM TEST1";

    let result = unsafe {
// создаем prepared statement
        let port = Port { vtab: std::mem::uninitialized(), _unused: [0; 60] };

        let r_prep = ffi::tarantool::sql_prepare(sql.as_ptr() as *const c_char, sql.len() as u32, &port as *const Port);

// получаем stmt_id
        let sql_port = &port as *const Port as *const PortSql;
        let stmt = (*sql_port).sql_stmt;
        let sql = CStr::from_ptr(ffi::tarantool::sql_stmt_query_str(stmt));
        let stmt_id = ffi::tarantool::sql_stmt_calculate_id(sql.as_ptr(), sql.to_str().unwrap().len());

// получаем текущий файбер, тк для execute нам нужен region (fiber->gc)
        let fiber_ptr = ffi::tarantool::fiber_self();
// грязный хак
        let region = fiber_ptr.offset(48);

// вызываем execute
        let port2 = Port { vtab: std::mem::uninitialized(), _unused: [0; 60] };
        let r_ex = ffi::tarantool::sql_execute_prepared(stmt_id, std::ptr::null::<c_void>(), 0, &port2 as *const Port, region);

// создаем буфер 
        let slab_c = ffi::tarantool::cord_slab_cache();
        let mut obuf = Obuf::new();

        ffi::tarantool::obuf_create(&mut obuf as *mut Obuf, slab_c, 1024);

// дергаем метод dump_msgpack для того чтобы буфер заполнился результатами запроса упакованными в msgpack
        ((*port2.vtab).dump_msgpack)(&port2 as *const Port, &obuf as *const Obuf);

        let iov_ptr = obuf.iov[0].iov_base as *mut u8;
        let v = std::slice::from_raw_parts(iov_ptr, obuf.iov[0].iov_len);
       
// в v у нас msgpack encoded результаты
        println!("result: {:?}", v);


        ffi::tarantool::port_destroy(&port2 as *const Port);
        ffi::tarantool::port_destroy(&port as *const Port);

        (r_prep, r_ex)
    };
    let e = tarantool::error::TarantoolErrorCode::try_last();

    assert_eq!(e, None);
    assert_eq!(result.0, 0);
    assert_eq!(result.1, 0);
}

У данного кода 2 проблемы:

после консультации с @gmoshkin выяснилось что почти все используемое CAPI не относится к экспорируемому, и его уже не возможно использовать в версиях 2.10+
ручной разбор структуры fiber - максимально не безопасен ввиду наличия в оригинальной структуре полей спрятанных за ifdef назначение которых понять сложновато

Предлагаемый пример работы с rust api (в случае если получится утрясти имеющиеся проблемы):

pub fn prepare_and_execute_some_sql_example() {
    let query = "SELECT * FROM t1 WHERE a = ? and b = ?";

    {
        let statement = tarantool::sql::prepare(query);

        let result1 = statement.execute::<Vec<T1Row>>(&(1, 2));
        let result2 = statement.execute::<Vec<T1Row>>(&(2, 3));

        // box.unprepare(statement) unexplicitly called when drop statement
    }
}

mentioned in issue picodata/tarantool#2

mentioned in commit fdebeccd

mentioned in commit d0ffb81f

mentioned in commit bb2b831b

mentioned in merge request !144 (closed)

mentioned in commit 4e1af37c

mentioned in commit 6ece570a

mentioned in commit 4c626f4d

mentioned in commit 1ff787ce

mentioned in commit 00f66ec0

mentioned in issue #130 (moved)

mentioned in issue picodata/tarantool#30

Поддержка добавляется по необходимости

closed

Support some of SQL execution API

Designs

Child items ...

Activity