AuroraMiddleware/zstd
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Added env var function for threads

Browse Source
This commit is contained in:
senhuang42 2020-08-25 18:49:52 -04:00
parent 5d91206ebe
commit 70fa7b37ca
1 changed files with 26 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
programs/zstdcli.c
View File

@ -16,6 +16,10 @@
# define ZSTDCLI_CLEVEL_DEFAULT 3
#endif
#ifndef ZSTD_THREADS_DEFAULT
# define ZSTD_THREADS_DEFAULT 3
#endif
#ifndef ZSTDCLI_CLEVEL_MAX
# define ZSTDCLI_CLEVEL_MAX 19 /* without using --ultra */
#endif
@ -586,6 +590,7 @@ static void printVersion(void)
/* Environment variables for parameter setting */
#define ENV_CLEVEL "ZSTD_CLEVEL"
#define ENV_THREADS "ZSTD_THREADS"
/* pick up environment variable */
static int init_cLevel(void) {
@ -615,6 +620,26 @@ static int init_cLevel(void) {
return ZSTDCLI_CLEVEL_DEFAULT;
}
/* pick up environment variable */
static unsigned init_numThreads(void) {
const char* const env = getenv(ENV_THREADS);
if (env != NULL) {
const char* ptr = env;
if ((*ptr>='0') && (*ptr<='9')) {
unsigned numThreads;
if (readU32FromCharChecked(&ptr, &numThreads)) {
DISPLAYLEVEL(2, "Ignore environment variable setting %s=%s: numeric value too large \n", ENV_THREADS, env);
return ZSTD_THREADS_DEFAULT;
} else if (*ptr == 0) {
return numThreads;
}
}
DISPLAYLEVEL(2, "Ignore environment variable setting %s=%s: not a valid integer value \n", ENV_THREADS, env);
}
return ZSTD_THREADS_DEFAULT;
}
#define NEXT_FIELD(ptr) { \
if (*argument == '=') { \
ptr = ++argument; \
@ -721,7 +746,7 @@ int main(int const argCount, const char* argv[])
if ((filenames==NULL) || (file_of_names==NULL)) { DISPLAY("zstd: allocation error \n"); exit(1); }
programName = lastNameFromPath(programName);
#ifdef ZSTD_MULTITHREAD
nbWorkers = 1;
nbWorkers = init_numThreads();
#endif
/* preset behaviors */