cKey
6 years ago
committed by
GitHub
GitHub
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 Ocelot.Multiplexer; |
|||
using System.Collections.Generic; |
|||
using System.Linq; |
|||
using System.Net; |
|||
using System.Net.Http; |
|||
using System.Net.Http.Headers; |
|||
using System.Text; |
|||
using System.Threading.Tasks; |
|||
|
|||
namespace Ocelot.Middleware.Multiplexer |
|||
{ |
|||
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); |
|||
} |
|||
|
|||
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; |
|||
JsonMergeSettings mergeSetting = new JsonMergeSettings(); |
|||
mergeSetting.MergeArrayHandling = MergeArrayHandling.Union; |
|||
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(); |
|||
if (contexts.Count == 1) |
|||
var content = await httpResponse.Items.DownstreamResponse().Content.ReadAsStringAsync(); |
|||
var contentObject = JsonConvert.DeserializeObject(content); |
|||
if (responseObject == null) |
|||
{ |
|||
if (contexts[0].IsError) |
|||
{ |
|||
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); |
|||
} |
|||
responseObject = JObject.FromObject(contentObject); |
|||
} |
|||
else |
|||
{ |
|||
for (var i = 0; i < contexts.Count; i++) |
|||
{ |
|||
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); |
|||
} |
|||
} |
|||
responseObject.Merge(contentObject, mergeSetting); |
|||
} |
|||
} |
|||
|
|||
var stringContent = new StringContent(responseObject.ToString()) |
|||
{ |
|||
Headers = { ContentType = new MediaTypeHeaderValue("application/json") } |
|||
}; |
|||
stringContent.Headers.Add("_abperrorformat", "true"); |
|||
return new DownstreamResponse(stringContent, HttpStatusCode.OK, |
|||
return new DownstreamResponse(stringContent, HttpStatusCode.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