Function main

freebsd/contrib/zstd/examples/streaming_compression_thread_pool.c:127–178

}

int main(int argc, const char** argv)
{
    const char* const exeName = argv[0];

    if (argc<=3) {
        printf("wrong arguments\n");
        printf("usage:\n");
        printf("%s POOL_SIZE LEVEL FILES\n", exeName);
        return 1;
    }

    int pool_size = atoi (argv[1]);
    CHECK(pool_size != 0, "can't parse POOL_SIZE!");

    int level = atoi (argv[2]);
    CHECK(level != 0, "can't parse LEVEL!");

    argc -= 3;
    argv += 3;

#if defined(ZSTD_STATIC_LINKING_ONLY)
    ZSTD_threadPool *pool = ZSTD_createThreadPool (pool_size);
    CHECK(pool != NULL, "ZSTD_createThreadPool() failed!");
    fprintf (stderr, "Using shared thread pool of size %d\n", pool_size);
#else
    fprintf (stderr, "All threads use its own thread pool\n");
#endif

    pthread_t *threads = malloc_orDie(argc * sizeof(pthread_t));
    compress_args_t *args = malloc_orDie(argc * sizeof(compress_args_t));

    for (unsigned i = 0; i < argc; i++)
    {
      args[i].fname = argv[i];
      args[i].outName = createOutFilename_orDie(args[i].fname);
      args[i].cLevel = level;
#if defined(ZSTD_STATIC_LINKING_ONLY)
      args[i].pool = pool;
#endif

      pthread_create (&threads[i], NULL, compressFile_orDie, &args[i]);
    }

    for (unsigned i = 0; i < argc; i++)
      pthread_join (threads[i], NULL);

#if defined(ZSTD_STATIC_LINKING_ONLY)
    ZSTD_freeThreadPool (pool);
#endif

    return 0;
}