1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
|
using Google.Protobuf;
using System;
using System.IO;
using System.Net;
using System.Net.Http;
using System.Net.Http.Formatting;
using System.Net.Http.Headers;
using System.Threading.Tasks;
using System.Web.Http;
using Tango.PMR.Stubs;
using Tango.PMR.Synchronization;
namespace Tango.Web.Formatters
{
/// <summary>
/// Represents a protobuf web request/response formatter capable of formatting messages using protobuf.
/// </summary>
/// <seealso cref="System.Net.Http.Formatting.MediaTypeFormatter" />
public class ProtoBufFormatter : MediaTypeFormatter
{
private static readonly MediaTypeHeaderValue mediaType = new MediaTypeHeaderValue("application/x-protobuf");
/// <summary>
/// Initializes a new instance of the <see cref="ProtoBufFormatter"/> class.
/// </summary>
public ProtoBufFormatter()
{
SupportedMediaTypes.Add(mediaType);
}
/// <summary>
/// Gets the default type of the media.
/// </summary>
/// <value>
/// The default type of the media.
/// </value>
public static MediaTypeHeaderValue DefaultMediaType
{
get { return mediaType; }
}
/// <summary>
/// Queries whether this <see cref="T:System.Net.Http.Formatting.MediaTypeFormatter" /> can deserializean object of the specified type.
/// </summary>
/// <param name="type">The type to deserialize.</param>
/// <returns>
/// true if the <see cref="T:System.Net.Http.Formatting.MediaTypeFormatter" /> can deserialize the type; otherwise, false.
/// </returns>
public override bool CanReadType(Type type)
{
return true;
}
/// <summary>
/// Queries whether this <see cref="T:System.Net.Http.Formatting.MediaTypeFormatter" /> can serializean object of the specified type.
/// </summary>
/// <param name="type">The type to serialize.</param>
/// <returns>
/// true if the <see cref="T:System.Net.Http.Formatting.MediaTypeFormatter" /> can serialize the type; otherwise, false.
/// </returns>
public override bool CanWriteType(Type type)
{
return true;
}
/// <summary>
/// Reads from stream asynchronous.
/// </summary>
/// <param name="type">The type.</param>
/// <param name="stream">The stream.</param>
/// <param name="content">The content.</param>
/// <param name="formatterLogger">The formatter logger.</param>
/// <returns></returns>
public override Task<object> ReadFromStreamAsync(Type type, Stream stream, HttpContent content, IFormatterLogger formatterLogger)
{
var tcs = new TaskCompletionSource<object>();
try
{
MessageParser parser = type.GetProperty("Parser").GetValue(null, null) as MessageParser;
IMessage req = parser.ParseFrom(stream);
tcs.SetResult(req);
}
catch (Exception ex)
{
tcs.SetException(ex);
}
return tcs.Task;
}
/// <summary>
/// Writes to stream asynchronous.
/// </summary>
/// <param name="type">The type.</param>
/// <param name="value">The value.</param>
/// <param name="stream">The stream.</param>
/// <param name="content">The content.</param>
/// <param name="transportContext">The transport context.</param>
/// <returns></returns>
public override Task WriteToStreamAsync(Type type, object value, Stream stream, HttpContent content, TransportContext transportContext)
{
var tcs = new TaskCompletionSource<object>();
if (value is IMessage)
{
try
{
(value as IMessage).WriteTo(stream);
tcs.SetResult(null);
}
catch (Exception ex)
{
tcs.SetException(ex);
}
return tcs.Task;
}
else if (value is HttpError)
{
var httpError = value as HttpError;
try
{
HttpProtoException msg = new HttpProtoException();
msg.Message = httpError["Message"].ToString();
msg.StatusCode = (int)HttpStatusCode.InternalServerError;
msg.WriteTo(stream);
tcs.SetResult(null);
}
catch (Exception ex)
{
tcs.SetException(ex);
}
return tcs.Task;
}
return base.WriteToStreamAsync(type, value, stream, content, transportContext);
}
}
}
|
