Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Handle actually issuing REST requests, not just building them #1

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
anaisbetts merged 6 commits into from
Jul 23, 2013
Merged

Handle actually issuing REST requests, not just building them #1

Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
09bd47b
Write a test to verify valid return types
anaisbetts Jul 22, 2013
074d986
Add method return type validation in RestMethodInfo
anaisbetts Jul 22, 2013
7221f1d
Implement BuildRestResultFuncForMethod for Task<T>
anaisbetts Jul 22, 2013
1a66f3a
Handle Rx
anaisbetts Jul 22, 2013
6b9e551
Throw on failed HTTP requests
anaisbetts Jul 22, 2013
b913d01
Handle requests that only care about completion
anaisbetts Jul 22, 2013
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
203 changes: 202 additions & 1 deletion RequestBuilder.cs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@
using Newtonsoft.Json;
using System.IO;
using System.Web;
using System.Threading;

namespace Refit
{
Expand Down Expand Up @@ -84,6 +85,165 @@ public Func<object[], HttpRequestMessage> BuildRequestFactoryForMethod(string me
return ret;
};
}

public Func<HttpClient, object[], object> BuildRestResultFuncForMethod(string methodName)
{
if (!interfaceHttpMethods.ContainsKey(methodName)) {
throw new ArgumentException("Method must be defined and have an HTTP Method attribute");
}

var restMethod = interfaceHttpMethods[methodName];

if (restMethod.ReturnType == typeof(Task)) {
return buildVoidTaskFuncForMethod(restMethod);
} else if (restMethod.ReturnType.GetGenericTypeDefinition() == typeof(Task<>)) {
return buildTaskFuncForMethod(restMethod);
} else {
return buildRxFuncForMethod(restMethod);
}
}

Func<HttpClient, object[], Task> buildVoidTaskFuncForMethod(RestMethodInfo restMethod)
{
var factory = BuildRequestFactoryForMethod(restMethod.Name);

return async (client, paramList) => {
var rq = factory(paramList);
var resp = await client.SendAsync(rq);

resp.EnsureSuccessStatusCode();
};
}

Func<HttpClient, object[], Task<object>> buildTaskFuncForMethod(RestMethodInfo restMethod)
{
var factory = BuildRequestFactoryForMethod(restMethod.Name);

return async (client, paramList) => {
var rq = factory(paramList);
var resp = await client.SendAsync(rq);
if (restMethod.SerializedReturnType == null) {
return resp;
}

resp.EnsureSuccessStatusCode();

var content = await resp.Content.ReadAsStringAsync();
if (restMethod.SerializedReturnType == typeof(string)) {
return content;
}

return JsonConvert.DeserializeObject(content, restMethod.SerializedReturnType);
};
}

Func<HttpClient, object[], IObservable<object>> buildRxFuncForMethod(RestMethodInfo restMethod)
{
var taskFunc = buildTaskFuncForMethod(restMethod);

return (client, paramList) => {
var ret = new FakeAsyncSubject<object>();

taskFunc(client, paramList).ContinueWith(t => {
if (t.Exception != null) {
ret.OnError(t.Exception);
} else {
ret.OnNext(t.Result);
ret.OnCompleted();
}
});

return ret;
};
}

class CompletionResult
{
public bool IsCompleted { get; set; }
public Exception Error { get; set; }
}

class FakeAsyncSubject<T> : IObservable<T>, IObserver<T>
{
bool resultSet;
T result;
CompletionResult completion;
List<IObserver<T>> subscriberList = new List<IObserver<T>>();

public void OnNext(T value)
{
if (completion == null) return;

result = value;
resultSet = true;

var currentList = default(IObserver<T>[]);
lock (subscriberList) { currentList = subscriberList.ToArray(); }
foreach (var v in currentList) v.OnNext(value);
}

public void OnError(Exception error)
{
var final = Interlocked.CompareExchange(ref completion, new CompletionResult() { IsCompleted = false, Error = error }, null);
if (final.IsCompleted) return;

var currentList = default(IObserver<T>[]);
lock (subscriberList) { currentList = subscriberList.ToArray(); }
foreach (var v in currentList) v.OnError(error);

final.IsCompleted = true;
}

public void OnCompleted()
{
var final = Interlocked.CompareExchange(ref completion, new CompletionResult() { IsCompleted = false, Error = null }, null);
if (final.IsCompleted) return;

var currentList = default(IObserver<T>[]);
lock (subscriberList) { currentList = subscriberList.ToArray(); }
foreach (var v in currentList) v.OnCompleted();

final.IsCompleted = true;
}

public IDisposable Subscribe(IObserver<T> observer)
{
if (completion != null) {
if (completion.Error != null) {
observer.OnError(completion.Error);
return new AnonymousDisposable(() => {});
}

if (resultSet) observer.OnNext(result);
observer.OnCompleted();

return new AnonymousDisposable(() => {});
}

lock (subscriberList) {
subscriberList.Add(observer);
}

return new AnonymousDisposable(() => {
lock (subscriberList) { subscriberList.Remove(observer); }
});
}
}
}

sealed class AnonymousDisposable : IDisposable
{
readonly Action block;

public AnonymousDisposable(Action block)
{
this.block = block;
}

public void Dispose()
{
block();
}
}

class RestMethodInfo
Expand All @@ -96,6 +256,8 @@ class RestMethodInfo
public Dictionary<int, string> ParameterMap { get; set; }
public Tuple<BodySerializationMethod, int> BodyParameterInfo { get; set; }
public Dictionary<int, string> QueryParameterMap { get; set; }
public Type ReturnType { get; set; }
public Type SerializedReturnType { get; set; }

static readonly Regex parameterRegex = new Regex(@"^{(.*)}$");

Expand All @@ -113,6 +275,7 @@ public RestMethodInfo(Type targetInterface, MethodInfo methodInfo)
RelativePath = hma.Path;

verifyUrlPathIsSane(RelativePath);
determineReturnTypeInfo(methodInfo);

var parameterList = methodInfo.GetParameters().ToList();

Expand Down Expand Up @@ -196,6 +359,26 @@ Tuple<BodySerializationMethod, int> findBodyParameter(List<ParameterInfo> parame

return Tuple.Create(ret.BodyAttribute.SerializationMethod, parameterList.IndexOf(ret.Parameter));
}

void determineReturnTypeInfo(MethodInfo methodInfo)
{
if (methodInfo.ReturnType.IsGenericType == false && methodInfo.ReturnType != typeof(Task)) {
goto bogusMethod;
}

var genericType = methodInfo.ReturnType.GetGenericTypeDefinition();
if (genericType != typeof(Task<>) && genericType != typeof(IObservable<>)) {
goto bogusMethod;
}

ReturnType = methodInfo.ReturnType;
SerializedReturnType = methodInfo.ReturnType.GetGenericArguments()[0];
if (SerializedReturnType == typeof(HttpResponseMessage)) SerializedReturnType = null;
return;

bogusMethod:
throw new ArgumentException("All REST Methods must return either Task<T> or IObservable<T>");
}
}

/*
Expand Down Expand Up @@ -223,7 +406,10 @@ public interface IRestMethodInfoTests
Task<string> FetchSomeStuffWithAlias([AliasAs("id")] int anId);

[Get("/foo/bar/{id}")]
Task<string> FetchSomeStuffWithBody([AliasAs("id")] int anId, [Body] Dictionary<int, string> theData);
IObservable<string> FetchSomeStuffWithBody([AliasAs("id")] int anId, [Body] Dictionary<int, string> theData);

[Post("/foo/{id}")]
string AsyncOnlyBuddy(int id);
}

[TestFixture]
Expand Down Expand Up @@ -310,6 +496,21 @@ public void FindTheBodyParameter()
Assert.AreEqual(0, fixture.QueryParameterMap.Count);
Assert.AreEqual(1, fixture.BodyParameterInfo.Item2);
}

[Test]
public void SyncMethodsShouldThrow()
{
bool shouldDie = true;

try {
var input = typeof(IRestMethodInfoTests);
var fixture = new RestMethodInfo(input, input.GetMethods().First(x => x.Name == "AsyncOnlyBuddy"));
} catch (ArgumentException) {
shouldDie = false;
}

Assert.IsFalse(shouldDie);
}
}

public interface IDummyHttpApi
Expand Down