Add config option for the read buffer size of the NNTP connection

nzbget-ng · Jan 5, 2023 · bfc68a4 · bfc68a4
1 parent 942f8a7
commit bfc68a4
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 1 deletion.
diff --git a/daemon/main/Options.cpp b/daemon/main/Options.cpp
@@ -140,6 +140,7 @@ static const char* OPTION_QUOTASTARTDAY			= "QuotaStartDay";
 static const char* OPTION_DAILYQUOTA			= "DailyQuota";
 static const char* OPTION_REORDERFILES			= "ReorderFiles";
 static const char* OPTION_UPDATECHECK			= "UpdateCheck";
+static const char* OPTION_CONNECTIONREADBUFFER		= "ConnectionReadBuffer";
 
 // obsolete options
 static const char* OPTION_POSTLOGKIND			= "PostLogKind";
@@ -525,6 +526,7 @@ void Options::InitDefaults()
 	SetOption(OPTION_DAILYQUOTA, "0");
 	SetOption(OPTION_REORDERFILES, "no");
 	SetOption(OPTION_UPDATECHECK, "none");
+	SetOption(OPTION_CONNECTIONREADBUFFER, "4");
 }
 
 void Options::InitOptFile()
@@ -728,6 +730,7 @@ void Options::InitOptions()
 	m_monthlyQuota			= ParseIntValue(OPTION_MONTHLYQUOTA, 10);
 	m_quotaStartDay			= ParseIntValue(OPTION_QUOTASTARTDAY, 10);
 	m_dailyQuota			= ParseIntValue(OPTION_DAILYQUOTA, 10);
+	m_connectionReadBuffer		= ParseIntValue(OPTION_CONNECTIONREADBUFFER, 10) * 1024;
 
 	m_nzbLog				= (bool)ParseEnumValue(OPTION_NZBLOG, BoolCount, BoolNames, BoolValues);
 	m_appendCategoryDir		= (bool)ParseEnumValue(OPTION_APPENDCATEGORYDIR, BoolCount, BoolNames, BoolValues);

diff --git a/daemon/main/Options.h b/daemon/main/Options.h
@@ -305,6 +305,8 @@ class Options
 	Categories* GetCategories() { return &m_categories; }
 	Category* FindCategory(const char* name, bool searchAliases) { return m_categories.FindCategory(name, searchAliases); }
 
+	int GetConnectionReadBuffer() { return m_connectionReadBuffer; }
+
 	// Current state
 	void SetServerMode(bool serverMode) { m_serverMode = serverMode; }
 	bool GetServerMode() { return m_serverMode; }
@@ -434,6 +436,7 @@ class Options
 	bool m_reorderFiles = false;
 	EFileNaming m_fileNaming = nfArticle;
 	int m_downloadRate = 0;
+	int m_connectionReadBuffer = 0;
 
 	// Application mode
 	bool m_serverMode = false;

diff --git a/daemon/nntp/ArticleDownloader.cpp b/daemon/nntp/ArticleDownloader.cpp
@@ -335,7 +335,7 @@ ArticleDownloader::EStatus ArticleDownloader::Download()
 	m_decoder.SetRawMode(g_Options->GetRawArticle());
 
 	status = adRunning;
-	CharBuffer lineBuf(1024*4);
+	CharBuffer lineBuf(g_Options->GetConnectionReadBuffer());
 
 	while (!IsStopped() && !m_decoder.GetEof())
 	{

diff --git a/nzbget.conf b/nzbget.conf
@@ -1071,6 +1071,10 @@ QuotaStartDay=1
 DailyQuota=0
 
 
+# Readbuffer size for the connections to the news server (kilobytes).
+ConnectionReadBuffer=4
+
+
 ##############################################################################
 ### LOGGING                                                                ###