faiss/tests/test_util.h

/**
 * Copyright (c) Facebook, Inc. and its affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 */

#ifndef FAISS_TEST_UTIL_H
#define FAISS_TEST_UTIL_H

#include <faiss/IndexIVFPQ.h>
#include <unistd.h>
#include <cstdlib>

struct Tempfilename {
    pthread_mutex_t* mutex;
    std::string filename;

    Tempfilename(pthread_mutex_t* mutex, std::string filename) {
        this->mutex = mutex;
        this->filename = filename;
        pthread_mutex_lock(mutex);
        int fd = mkstemp(&filename[0]);
        close(fd);
        pthread_mutex_unlock(mutex);
    }

    ~Tempfilename() {
        if (access(filename.c_str(), F_OK)) {
            unlink(filename.c_str());
        }
    }

    const char* c_str() {
        return filename.c_str();
    }
};

#endif // FAISS_TEST_UTIL_H