032fbaefb4
Signed-off-by: Vincent Biret <vibiret@microsoft.com>
92 lines
3.1 KiB
C#
92 lines
3.1 KiB
C#
// ------------------------------------------------------------
|
|
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License (MIT). See LICENSE in the repo root for license information.
|
|
// ------------------------------------------------------------
|
|
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using Microsoft.OpenApi.Models;
|
|
using Microsoft.OpenApi.OData.Common;
|
|
using Microsoft.OpenApi.OData.Edm;
|
|
using Microsoft.OpenApi.OData.Generator;
|
|
|
|
namespace Microsoft.OpenApi.OData.Operation
|
|
{
|
|
/// <summary>
|
|
/// Retrieve a $count get
|
|
/// </summary>
|
|
internal class DollarCountGetOperationHandler : OperationHandler
|
|
{
|
|
/// <inheritdoc/>
|
|
public override OperationType OperationType => OperationType.Get;
|
|
|
|
/// <summary>
|
|
/// Gets/sets the segment before $count.
|
|
/// this segment could be "entity set", "Collection property", "Composable function whose return is collection",etc.
|
|
/// </summary>
|
|
internal ODataSegment LastSecondSegment { get; set; }
|
|
private const int SecondLastSegmentIndex = 2;
|
|
/// <inheritdoc/>
|
|
protected override void Initialize(ODataContext context, ODataPath path)
|
|
{
|
|
base.Initialize(context, path);
|
|
|
|
// get the last second segment
|
|
int count = path.Segments.Count;
|
|
if(count >= SecondLastSegmentIndex)
|
|
LastSecondSegment = path.Segments.ElementAt(count - SecondLastSegmentIndex);
|
|
}
|
|
|
|
/// <inheritdoc/>
|
|
protected override void SetBasicInfo(OpenApiOperation operation)
|
|
{
|
|
// Summary
|
|
operation.Summary = $"Get the number of the resource";
|
|
|
|
// OperationId
|
|
if (Context.Settings.EnableOperationId)
|
|
{
|
|
operation.OperationId = $"Get.Count.{LastSecondSegment.Identifier}";
|
|
}
|
|
|
|
base.SetBasicInfo(operation);
|
|
}
|
|
|
|
/// <inheritdoc/>
|
|
protected override void SetResponses(OpenApiOperation operation)
|
|
{
|
|
OpenApiSchema schema = new()
|
|
{
|
|
Reference = new() {
|
|
Type = ReferenceType.Schema,
|
|
Id = Constants.DollarCountSchemaName
|
|
}
|
|
};
|
|
|
|
operation.Responses = new OpenApiResponses
|
|
{
|
|
{
|
|
Constants.StatusCode200,
|
|
new OpenApiResponse
|
|
{
|
|
Description = "The count of the resource",
|
|
Content = new Dictionary<string, OpenApiMediaType>
|
|
{
|
|
{
|
|
"text/plain",
|
|
new OpenApiMediaType
|
|
{
|
|
Schema = schema
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
};
|
|
operation.Responses.Add(Constants.StatusCodeDefault, Constants.StatusCodeDefault.GetResponse());
|
|
|
|
base.SetResponses(operation);
|
|
}
|
|
}
|
|
}
|