Function Java_com_cactus_CactusJNI_nativeIndexQuery

android/cactus_jni.cpp:724–767

}

JNIEXPORT jstring JNICALL
Java_com_cactus_CactusJNI_nativeIndexQuery(JNIEnv* env, jobject, jlong handle,
                                            jfloatArray embedding, jlong topK, jstring optionsJson) {
    if (handle == 0) { throwCactusException(env, "Index not initialized"); return nullptr; }

    jsize embDim = env->GetArrayLength(embedding);
    jfloat* embData = env->GetFloatArrayElements(embedding, nullptr);
    const char* options = jstring_to_cstr(env, optionsJson);

    std::vector<int> idBuffer(static_cast<size_t>(topK));
    std::vector<float> scoreBuffer(static_cast<size_t>(topK));
    size_t idBufferSize = static_cast<size_t>(topK);
    size_t scoreBufferSize = static_cast<size_t>(topK);

    const float* embPtr = embData;
    int* idPtr = idBuffer.data();
    float* scorePtr = scoreBuffer.data();

    int result = cactus_index_query(
        reinterpret_cast<cactus_index_t>(handle),
        &embPtr,
        1,
        static_cast<size_t>(embDim),
        options,
        &idPtr,
        &idBufferSize,
        &scorePtr,
        &scoreBufferSize
    );

    env->ReleaseFloatArrayElements(embedding, embData, JNI_ABORT);
    release_jstring(env, optionsJson, options);

    if (result < 0) { throwOnError(env); return nullptr; }

    std::string json = "{\"results\":[";
    for (size_t i = 0; i < idBufferSize; i++) {
        if (i > 0) json += ",";
        json += "{\"id\":" + std::to_string(idBuffer[i]) + ",\"score\":" + std::to_string(scoreBuffer[i]) + "}";
    }
    json += "]}";

    return env->NewStringUTF(json.c_str());
}

JNIEXPORT jint JNICALL
Java_com_cactus_CactusJNI_nativeIndexCompact(JNIEnv* env, jobject, jlong handle) {