diff options
Diffstat (limited to 'src/backend')
| -rw-r--r-- | src/backend/replication/basebackup_zstd.c | 45 |
1 files changed, 28 insertions, 17 deletions
diff --git a/src/backend/replication/basebackup_zstd.c b/src/backend/replication/basebackup_zstd.c index 5496eaa72b7..f6876f48118 100644 --- a/src/backend/replication/basebackup_zstd.c +++ b/src/backend/replication/basebackup_zstd.c @@ -25,8 +25,8 @@ typedef struct bbsink_zstd /* Common information for all types of sink. */ bbsink base; - /* Compression level */ - int compresslevel; + /* Compression options */ + bc_specification *compress; ZSTD_CCtx *cctx; ZSTD_outBuffer zstd_outBuf; @@ -67,22 +67,13 @@ bbsink_zstd_new(bbsink *next, bc_specification *compress) return NULL; /* keep compiler quiet */ #else bbsink_zstd *sink; - int compresslevel; Assert(next != NULL); - if ((compress->options & BACKUP_COMPRESSION_OPTION_LEVEL) == 0) - compresslevel = 0; - else - { - compresslevel = compress->level; - Assert(compresslevel >= 1 && compresslevel <= 22); - } - sink = palloc0(sizeof(bbsink_zstd)); *((const bbsink_ops **) &sink->base.bbs_ops) = &bbsink_zstd_ops; sink->base.bbs_next = next; - sink->compresslevel = compresslevel; + sink->compress = compress; return &sink->base; #endif @@ -99,16 +90,36 @@ bbsink_zstd_begin_backup(bbsink *sink) bbsink_zstd *mysink = (bbsink_zstd *) sink; size_t output_buffer_bound; size_t ret; + bc_specification *compress = mysink->compress; mysink->cctx = ZSTD_createCCtx(); if (!mysink->cctx) elog(ERROR, "could not create zstd compression context"); - ret = ZSTD_CCtx_setParameter(mysink->cctx, ZSTD_c_compressionLevel, - mysink->compresslevel); - if (ZSTD_isError(ret)) - elog(ERROR, "could not set zstd compression level to %d: %s", - mysink->compresslevel, ZSTD_getErrorName(ret)); + if ((compress->options & BACKUP_COMPRESSION_OPTION_LEVEL) != 0) + { + ret = ZSTD_CCtx_setParameter(mysink->cctx, ZSTD_c_compressionLevel, + compress->level); + if (ZSTD_isError(ret)) + elog(ERROR, "could not set zstd compression level to %d: %s", + compress->level, ZSTD_getErrorName(ret)); + } + + if ((compress->options & BACKUP_COMPRESSION_OPTION_WORKERS) != 0) + { + /* + * On older versions of libzstd, this option does not exist, and trying + * to set it will fail. Similarly for newer versions if they are + * compiled without threading support. + */ + ret = ZSTD_CCtx_setParameter(mysink->cctx, ZSTD_c_nbWorkers, + compress->workers); + if (ZSTD_isError(ret)) + ereport(ERROR, + errcode(ERRCODE_INVALID_PARAMETER_VALUE), + errmsg("could not set compression worker count to %d: %s", + compress->workers, ZSTD_getErrorName(ret))); + } /* * We need our own buffer, because we're going to pass different data to |