src/QueryCat.IntegrationTests/Plugins/SamplePluginRowsInput.cs - develop - krasninja/querycat

src/QueryCat.IntegrationTests/Plugins/SamplePluginRowsInput.cs

99 строк2 KB

Ivan Kozhin

Move ErrorCode to Core namespace

28 дек 2023, 07:18

28 дек 2023, 07:18035fe6b

using System.ComponentModel;
using System.Diagnostics;
using QueryCat.Backend.Core;
using QueryCat.Backend.Core.Data;
using QueryCat.Backend.Core.Functions;
using QueryCat.Backend.Core.Types;

namespace QueryCat.IntegrationTests.Plugins;

/// <summary>
/// Example simple rows input plugin.
/// </summary>
public class SamplePluginRowsInput : IRowsInput
{
    private const long MaxValue = 9;

    [Description("Sample input.")]
    [FunctionSignature("plugin(start: integer = 0): object<IRowsInput>")]
    public static VariantValue SamplePlugin(FunctionCallInfo args)
    {
        var startValue = args.GetAt(0).AsInteger;
        var rowsSource = new SamplePluginRowsInput(startValue);
        return VariantValue.CreateFromObject(rowsSource);
    }

    private long _currentState;

    /// <inheritdoc />
    public Column[] Columns { get; } =
    {
        new("id", DataType.Integer, "Key.")
    };

    /// <inheritdoc />
    public string[] UniqueKey { get; } = Array.Empty<string>();

    /// <inheritdoc />
    public QueryContext QueryContext
    {
        get => NullQueryContext.Instance;
        set => Trace.WriteLine($"Set {nameof(QueryContext)}");
    }

    public SamplePluginRowsInput(long initialValue)
    {
        _currentState = initialValue;
    }

    /// <inheritdoc />
    public void Open()
    {
        Trace.WriteLine(nameof(Open));
    }

    /// <inheritdoc />
    public void Close()
    {
        Trace.WriteLine(nameof(Close));
    }

    /// <inheritdoc />
    public ErrorCode ReadValue(int columnIndex, out VariantValue value)
    {
        Trace.WriteLine(nameof(ReadValue));
        if (columnIndex == 0)
        {
            value = new VariantValue(_currentState);
            return ErrorCode.OK;
        }

        value = default;
        return ErrorCode.InvalidColumnIndex;
    }

    /// <inheritdoc />
    public bool ReadNext()
    {
        Trace.WriteLine(nameof(ReadNext));
        if (_currentState >= MaxValue)
        {
            return false;
        }
        _currentState++;
        return true;
    }

    /// <inheritdoc />
    public void Reset()
    {
        Trace.WriteLine(nameof(Reset));
        _currentState = 0;
    }

    /// <inheritdoc />
    public void Explain(IndentedStringBuilder stringBuilder)
    {
        stringBuilder.AppendLine("Sample");
    }
}