cKey
6 years ago
5 changed files with 75 additions and 169 deletions
@ -1,149 +1,49 @@ |
|||||
using Newtonsoft.Json; |
using Microsoft.AspNetCore.Http; |
||||
|
using Newtonsoft.Json; |
||||
using Newtonsoft.Json.Linq; |
using Newtonsoft.Json.Linq; |
||||
|
using Ocelot.Multiplexer; |
||||
using System.Collections.Generic; |
using System.Collections.Generic; |
||||
using System.Linq; |
|
||||
using System.Net; |
using System.Net; |
||||
using System.Net.Http; |
using System.Net.Http; |
||||
using System.Net.Http.Headers; |
using System.Net.Http.Headers; |
||||
using System.Text; |
|
||||
using System.Threading.Tasks; |
using System.Threading.Tasks; |
||||
|
|
||||
namespace Ocelot.Middleware.Multiplexer |
namespace Ocelot.Middleware.Multiplexer |
||||
{ |
{ |
||||
public class AbpApiDefinitionAggregator : IDefinedAggregator |
public class AbpApiDefinitionAggregator : IDefinedAggregator |
||||
{ |
{ |
||||
public async Task<DownstreamResponse> Aggregate(List<DownstreamContext> responses) |
public async Task<DownstreamResponse> Aggregate(List<HttpContext> responses) |
||||
{ |
{ |
||||
// Ocelot抛弃了下游主机返回的HttpHeaders，所以没法判断是否为abp返回格式
|
|
||||
// var isAbpResponse = responses.Any(response => response.DownstreamResponse.Headers.Any(h => h.Key.Equals("_abperrorformat")));
|
|
||||
return await MapAbpApiDefinitionAggregateContentAsync(responses); |
return await MapAbpApiDefinitionAggregateContentAsync(responses); |
||||
} |
} |
||||
|
|
||||
protected virtual async Task<DownstreamResponse> MapAbpApiDefinitionAggregateContentAsync(List<DownstreamContext> downstreamContexts) |
protected virtual async Task<DownstreamResponse> MapAbpApiDefinitionAggregateContentAsync(List<HttpContext> responses) |
||||
{ |
{ |
||||
var responseKeys = downstreamContexts.Select(s => s.DownstreamReRoute.Key).Distinct().ToList(); |
|
||||
JObject responseObject = null; |
JObject responseObject = null; |
||||
JsonMergeSettings mergeSetting = new JsonMergeSettings(); |
JsonMergeSettings mergeSetting = new JsonMergeSettings(); |
||||
mergeSetting.MergeArrayHandling = MergeArrayHandling.Union; |
mergeSetting.MergeArrayHandling = MergeArrayHandling.Union; |
||||
mergeSetting.PropertyNameComparison = System.StringComparison.CurrentCultureIgnoreCase; |
mergeSetting.PropertyNameComparison = System.StringComparison.CurrentCultureIgnoreCase; |
||||
for (var k = 0; k < responseKeys.Count; k++) |
foreach (var httpResponse in responses) |
||||
{ |
{ |
||||
var contexts = downstreamContexts.Where(w => w.DownstreamReRoute.Key == responseKeys[k]).ToList(); |
var content = await httpResponse.Items.DownstreamResponse().Content.ReadAsStringAsync(); |
||||
if (contexts.Count == 1) |
var contentObject = JsonConvert.DeserializeObject(content); |
||||
|
if (responseObject == null) |
||||
{ |
{ |
||||
if (contexts[0].IsError) |
responseObject = JObject.FromObject(contentObject); |
||||
{ |
|
||||
return contexts[0].DownstreamResponse; |
|
||||
} |
|
||||
|
|
||||
var content = await contexts[0].DownstreamResponse.Content.ReadAsStringAsync(); |
|
||||
var contentObject = JsonConvert.DeserializeObject(content); |
|
||||
if (responseObject == null) |
|
||||
{ |
|
||||
responseObject = JObject.FromObject(contentObject); |
|
||||
} |
|
||||
else |
|
||||
{ |
|
||||
responseObject.Merge(contentObject, mergeSetting); |
|
||||
} |
|
||||
} |
} |
||||
else |
else |
||||
{ |
{ |
||||
for (var i = 0; i < contexts.Count; i++) |
responseObject.Merge(contentObject, mergeSetting); |
||||
{ |
|
||||
if (contexts[i].IsError) |
|
||||
{ |
|
||||
return contexts[i].DownstreamResponse; |
|
||||
} |
|
||||
|
|
||||
var content = await contexts[i].DownstreamResponse.Content.ReadAsStringAsync(); |
|
||||
var contentObject = JsonConvert.DeserializeObject(content); |
|
||||
if (responseObject == null) |
|
||||
{ |
|
||||
responseObject = JObject.FromObject(contentObject); |
|
||||
} |
|
||||
else |
|
||||
{ |
|
||||
responseObject.Merge(contentObject, mergeSetting); |
|
||||
} |
|
||||
} |
|
||||
} |
} |
||||
} |
} |
||||
|
|
||||
var stringContent = new StringContent(responseObject.ToString()) |
var stringContent = new StringContent(responseObject.ToString()) |
||||
{ |
{ |
||||
Headers = { ContentType = new MediaTypeHeaderValue("application/json") } |
Headers = { ContentType = new MediaTypeHeaderValue("application/json") } |
||||
}; |
}; |
||||
stringContent.Headers.Add("_abperrorformat", "true"); |
stringContent.Headers.Add("_abperrorformat", "true"); |
||||
return new DownstreamResponse(stringContent, HttpStatusCode.OK, |
return new DownstreamResponse(stringContent, HttpStatusCode.OK, |
||||
new List<KeyValuePair<string, IEnumerable<string>>>(), "OK"); |
new List<KeyValuePair<string, IEnumerable<string>>>(), "OK"); |
||||
} |
} |
||||
|
|
||||
protected virtual async Task<DownstreamResponse> MapSimpleJsonAggregateContentAsync(List<DownstreamContext> downstreamContexts) |
|
||||
{ |
|
||||
var contentBuilder = new StringBuilder(); |
|
||||
|
|
||||
contentBuilder.Append("{"); |
|
||||
|
|
||||
var responseKeys = downstreamContexts.Select(s => s.DownstreamReRoute.Key).Distinct().ToList(); |
|
||||
|
|
||||
for (var k = 0; k < responseKeys.Count; k++) |
|
||||
{ |
|
||||
var contexts = downstreamContexts.Where(w => w.DownstreamReRoute.Key == responseKeys[k]).ToList(); |
|
||||
if (contexts.Count == 1) |
|
||||
{ |
|
||||
if (contexts[0].IsError) |
|
||||
{ |
|
||||
return contexts[0].DownstreamResponse; |
|
||||
} |
|
||||
|
|
||||
var content = await contexts[0].DownstreamResponse.Content.ReadAsStringAsync(); |
|
||||
contentBuilder.Append($"\"{responseKeys[k]}\":{content}"); |
|
||||
} |
|
||||
else |
|
||||
{ |
|
||||
contentBuilder.Append($"\"{responseKeys[k]}\":"); |
|
||||
contentBuilder.Append("["); |
|
||||
|
|
||||
for (var i = 0; i < contexts.Count; i++) |
|
||||
{ |
|
||||
if (contexts[i].IsError) |
|
||||
{ |
|
||||
return contexts[i].DownstreamResponse; |
|
||||
} |
|
||||
|
|
||||
var content = await contexts[i].DownstreamResponse.Content.ReadAsStringAsync(); |
|
||||
if (string.IsNullOrWhiteSpace(content)) |
|
||||
{ |
|
||||
continue; |
|
||||
} |
|
||||
|
|
||||
contentBuilder.Append($"{content}"); |
|
||||
|
|
||||
if (i + 1 < contexts.Count) |
|
||||
{ |
|
||||
contentBuilder.Append(","); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
contentBuilder.Append("]"); |
|
||||
} |
|
||||
|
|
||||
if (k + 1 < responseKeys.Count) |
|
||||
{ |
|
||||
contentBuilder.Append(","); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
contentBuilder.Append("}"); |
|
||||
|
|
||||
var stringContent = new StringContent(contentBuilder.ToString()) |
|
||||
{ |
|
||||
Headers = { ContentType = new MediaTypeHeaderValue("application/json") } |
|
||||
}; |
|
||||
|
|
||||
return new DownstreamResponse(stringContent, HttpStatusCode.OK, new List<KeyValuePair<string, IEnumerable<string>>>(), "OK"); |
|
||||
} |
|
||||
|
|
||||
} |
} |
||||
} |
} |
||||
|
|||||
Loading…
Reference in new issue