internal void Write(Stream stream)
{
Stream responseMessageStream = null;
Serializer responseSerializer = null;
this.actualResponseMessageWhoseHeadersMayBeOverridden.SetStream(stream);
try
{
switch (this.PayloadKind)
{
case ODataPayloadKind.BinaryValue:
Debug.Assert(
this.requestDescription.TargetKind == RequestTargetKind.OpenPropertyValue ||
this.requestDescription.TargetKind == RequestTargetKind.PrimitiveValue ||
this.requestDescription.TargetKind == RequestTargetKind.MediaResource,
this.requestDescription.TargetKind + " is PrimitiveValue or OpenPropertyValue or StreamPropertyValue");
Debug.Assert(this.messageWriter == null, "No Message writer required for writing binary payload");
responseMessageStream = this.actualResponseMessageWhoseHeadersMayBeOverridden.GetStream();
BinarySerializer binarySerializer = new BinarySerializer(responseMessageStream);
if (this.requestDescription.TargetKind == RequestTargetKind.MediaResource)
{
// If this.mediaResourceStream is null, we will set the status code to 204.
// We would not get here on a WCF host, but on other hosts we would.
Debug.Assert(this.requestDescription.IsNamedStream || this.mediaResourceStream != null, "this.mediaResourceStream cannot be null for MR request.");
if (this.mediaResourceStream != null)
{
binarySerializer.WriteRequest(this.mediaResourceStream, this.service.StreamProvider.StreamBufferSize);
}
}
else
{
binarySerializer.WriteRequest(this.queryResults.Current);
}
break;
case ODataPayloadKind.Value:
Debug.Assert(
this.requestDescription.TargetKind == RequestTargetKind.OpenPropertyValue ||
this.requestDescription.TargetKind == RequestTargetKind.PrimitiveValue,
this.requestDescription.TargetKind + " is PrimitiveValue or OpenPropertyValue");
TextSerializer textSerializer = new TextSerializer(this.messageWriter);
textSerializer.WriteRequest(this.queryResults.Current);
break;
case ODataPayloadKind.ServiceDocument:
Debug.Assert(this.requestDescription.TargetKind == RequestTargetKind.ServiceDirectory, "this.requestDescription.TargetKind == RequestTargetKind.ServiceDirectory");
ServiceDocumentSerializer serializer = new ServiceDocumentSerializer(this.messageWriter);
serializer.WriteServiceDocument(this.service.Provider);
break;
case ODataPayloadKind.EntityReferenceLink:
case ODataPayloadKind.EntityReferenceLinks:
case ODataPayloadKind.Collection:
case ODataPayloadKind.Property:
Debug.Assert(this.requestDescription.TargetKind != RequestTargetKind.Resource || this.requestDescription.LinkUri, "this.requestDescription.TargetKind != RequestTargetKind.Resource || this.requestDescription.LinkUri");
NonEntitySerializer nonEntitySerializer = new NonEntitySerializer(this.requestDescription, this.AbsoluteServiceUri, this.service, this.messageWriter);
responseSerializer = nonEntitySerializer;
nonEntitySerializer.WriteRequest(this.queryResults);
break;
case ODataPayloadKind.Entry:
case ODataPayloadKind.Feed:
Debug.Assert(this.requestDescription.TargetKind == RequestTargetKind.Resource, "TargetKind " + this.requestDescription.TargetKind + " == Resource");
EntitySerializer entitySerializer = new EntitySerializer(
this.requestDescription,
this.AbsoluteServiceUri,
this.service,
this.service.OperationContext.ResponseMessage.GetHeader(XmlConstants.HttpResponseETag),
this.messageWriter,
this.contentFormat);
responseSerializer = entitySerializer;
entitySerializer.WriteRequest(this.queryResults);
break;
default:
Debug.Assert(this.PayloadKind == ODataPayloadKind.MetadataDocument, "this.payloadKind == ODataPayloadKind.MetadataDocument");
Debug.Assert(this.requestDescription.TargetKind == RequestTargetKind.Metadata, "this.requestDescription.TargetKind == RequestTargetKind.Metadata");
this.messageWriter.WriteMetadataDocument();
break;
}
}
catch (Exception exception)
{
if (!CommonUtil.IsCatchableExceptionType(exception))
{
throw;
}
// Astoria-ODataLib-Integration: Astoria does not call flush before calling the IDataServiceHost.ProcessException method
// If the request is for an entry/feed and the data source throws an error while these results are being enumerated and written,
// the server doesn't flush the writer's stream before it calls HandleException and starts writing out the error.
// To handle this case, we'll make the EntitySerializer expose a method that calls Flush on the underlying ODL writer instance.
if (responseSerializer != null)
{
responseSerializer.Flush();
}
ErrorHandler.HandleExceptionWhileWriting(this.service, exception, this.actualResponseMessageWhoseHeadersMayBeOverridden, this.messageWriter, this.encoding, stream, this.messageWriterBuilder);
}
finally
{
WebUtil.Dispose(this.messageWriter);
WebUtil.Dispose(this.queryResults);
WebUtil.Dispose(this.mediaResourceStream);
// For batch operations, we need to dispose the stream obtained from calling
// IODataResponseMessage.GetStream to signal that we are done with the message.
if (responseMessageStream != null && this.actualResponseMessageWhoseHeadersMayBeOverridden is ODataBatchOperationResponseMessage)
{
responseMessageStream.Dispose();
}
}
}
