Function Java_com_cactus_CactusJNI_nativeIndexAdd

android/cactus_jni.cpp:651–704

}

JNIEXPORT jint JNICALL
Java_com_cactus_CactusJNI_nativeIndexAdd(JNIEnv* env, jobject, jlong handle,
                                          jintArray ids, jobjectArray documents,
                                          jobjectArray metadatas, jobjectArray embeddings,
                                          jlong embeddingDim) {
    if (handle == 0) return -1;

    jsize count = env->GetArrayLength(ids);
    jint* idData = env->GetIntArrayElements(ids, nullptr);

    std::vector<const char*> docPtrs(count);
    std::vector<const char*> metaPtrs(count);
    std::vector<const float*> embPtrs(count);
    std::vector<jstring> docStrings(count);
    std::vector<jstring> metaStrings(count);
    std::vector<jfloatArray> embArrays(count);

    for (jsize i = 0; i < count; i++) {
        docStrings[i] = static_cast<jstring>(env->GetObjectArrayElement(documents, i));
        docPtrs[i] = jstring_to_cstr(env, docStrings[i]);

        if (metadatas != nullptr) {
            metaStrings[i] = static_cast<jstring>(env->GetObjectArrayElement(metadatas, i));
            metaPtrs[i] = jstring_to_cstr(env, metaStrings[i]);
        } else {
            metaPtrs[i] = nullptr;
        }

        embArrays[i] = static_cast<jfloatArray>(env->GetObjectArrayElement(embeddings, i));
        embPtrs[i] = env->GetFloatArrayElements(embArrays[i], nullptr);
    }

    int result = cactus_index_add(
        reinterpret_cast<cactus_index_t>(handle),
        reinterpret_cast<const int*>(idData),
        docPtrs.data(),
        metadatas != nullptr ? metaPtrs.data() : nullptr,
        embPtrs.data(),
        static_cast<size_t>(count),
        static_cast<size_t>(embeddingDim)
    );

    for (jsize i = 0; i < count; i++) {
        release_jstring(env, docStrings[i], docPtrs[i]);
        if (metadatas != nullptr) {
            release_jstring(env, metaStrings[i], metaPtrs[i]);
        }
        env->ReleaseFloatArrayElements(embArrays[i], const_cast<jfloat*>(embPtrs[i]), JNI_ABORT);
    }

    env->ReleaseIntArrayElements(ids, idData, JNI_ABORT);
    if (result < 0) { throwOnError(env); return -1; }
    return result;
}

JNIEXPORT jint JNICALL
Java_com_cactus_CactusJNI_nativeIndexDelete(JNIEnv* env, jobject, jlong handle, jintArray ids) {