使用std::string 换掉stringstream

Author: CppCXY

CodeService/src/FormatElement/RangeFormatContext.cpp

@@ -49,7 +49,7 @@ void RangeFormatContext::Print(TextElement& textElement)
 			}
 		}
 	}
-	_os << textElement.GetText();
+	_buffer.append(textElement.GetText());
 	_characterCount += textElement.GetText().size();
 
 	if (startLine < _formattedRange.StartLine)
@@ -74,7 +74,7 @@ void RangeFormatContext::PrintBlank(int blank)
 	{
 		for (int i = 0; i < blank; i++)
 		{
-			_os << ' ';
+			_buffer.push_back(' ');
 		}
 	}
 }
@@ -86,7 +86,7 @@ void RangeFormatContext::PrintLine(int line)
 	{
 		for (int i = 0; i < line; i++)
 		{
-			_os << _options.end_of_line;
+			_buffer.append(_options.end_of_line);
 		}
 	}
 }

CodeService/src/FormatElement/SerializeContext.cpp

@@ -32,7 +32,7 @@ void SerializeContext::Print(TextElement& textElement)
 			}
 		}
 	}
-	_os << textElement.GetText();
+	_buffer.append(textElement.GetText());
 	_characterCount += textElement.GetText().size();
 }
 
@@ -44,7 +44,7 @@ void SerializeContext::PrintLine(int line)
 	}
 	for (int i = 0; i < line; i++)
 	{
-		_os << _options.end_of_line;
+		_buffer.append(_options.end_of_line);
 		_characterCount = 0;
 	}
 }
@@ -57,14 +57,19 @@ void SerializeContext::PrintBlank(int blank)
 	}
 	for (int i = 0; i < blank; i++)
 	{
-		_os << ' ';
+		_buffer.push_back(' ');
 		_characterCount++;
 	}
 }
 
 std::string SerializeContext::GetText()
 {
-	return _os.str();
+	return std::move(_buffer);
+}
+
+void SerializeContext::SetReadySize(std::size_t size)
+{
+	_buffer.reserve(size);
 }
 
 void SerializeContext::PrintIndent(std::size_t indent, IndentStyle style)
@@ -75,7 +80,7 @@ void SerializeContext::PrintIndent(std::size_t indent, IndentStyle style)
 		{
 			for (std::size_t i = 0; i < indent; i++)
 			{
-				_os << ' ';
+				_buffer.push_back(' ');
 			}
 			_characterCount += indent;
 			break;
@@ -84,7 +89,7 @@ void SerializeContext::PrintIndent(std::size_t indent, IndentStyle style)
 		{
 			for (std::size_t i = 0; i < indent; i++)
 			{
-				_os << '\t';
+				_buffer.push_back('\t');
 			}
 			_characterCount += indent;
 			break;

CodeService/src/LuaFormatter.cpp

@@ -61,6 +61,7 @@ void LuaFormatter::BuildFormattedElement()
 std::string LuaFormatter::GetFormattedText()
 {
 	SerializeContext ctx(_parser, _options);
+	ctx.SetReadySize(_parser->GetSource().size() + _parser->GetSource().size() / 2);
 
 	_env->Format(ctx);
 

include/CodeService/FormatElement/SerializeContext.h

@@ -16,8 +16,10 @@ class SerializeContext : public FormatContext
 	virtual void PrintBlank(int blank);
 
 	virtual std::string GetText();
+
+	void SetReadySize(std::size_t size);
 protected:
 	void PrintIndent(std::size_t indent, IndentStyle style);
-	std::stringstream _os;
+	std::string _buffer;
 };