typo fix

[pykit.git] / Publisher.cc

diff --git a/Publisher.cc b/Publisher.cc
index aa96b67..30a6789 100644 (file)
--- a/Publisher.cc
+++ b/Publisher.cc
@@ -50,6 +50,8 @@ namespace detail {
         virtual void abort();
 
         using QNetworkReply::setHeader;
+        using QNetworkReply::setRawHeader;
+        using QNetworkReply::setAttribute;
 
         void clearResponse();
         QByteArray postData() const;
@@ -155,7 +157,6 @@ prefix_ void pykit::detail::InternalServerReply::processRequest()
 prefix_ void pykit::detail::InternalServerReply::complete()
 {
     emit readyRead();
-    emit readChannelFinished();
     emit finished();
 }
 
@@ -200,7 +201,12 @@ prefix_ bool pykit::detail::InternalServerReply::waitForReadyRead(int msecs)
 { return buffer_.waitForReadyRead(msecs); }
 
 prefix_ qint64 pykit::detail::InternalServerReply::readData(char * data, qint64 maxSize)
-{ return buffer_.read(data,maxSize); }
+{
+    qint64 n (buffer_.read(data,maxSize));
+    if (buffer_.atEnd())
+        emit readChannelFinished();
+    return n;
+}
 
 prefix_ qint64 pykit::detail::InternalServerReply::writeData(char const * data, qint64 maxSize)
 { return buffer_.write(data,maxSize); }
@@ -248,6 +254,16 @@ prefix_ void pykit::Request::setLocation(std::string const & location)
     reply_.setHeader(QNetworkRequest::LocationHeader, QVariant(location.c_str()));
 }
 
+prefix_ void pykit::Request::setHeader(std::string const & header, std::string const & value)
+{
+    reply_.setRawHeader(header.c_str(), value.c_str());
+}
+
+prefix_ void pykit::Request::setStatusCode(int code)
+{
+    reply_.setAttribute(QNetworkRequest::HttpStatusCodeAttribute, code);
+}
+
 prefix_ QUrl pykit::Request::url()
     const
 {