将数据从 ArgumentTransformationAttribute 传递到 PSCmdlet

Question

一般来说，我尝试创建一个 PSCmdlet，它采用实现 IDisposeable 的类型的参数，并需要进行处置以避免泄漏资源。我还想接受该参数的字符串并创建该类型的实例，但是如果我自己创建该对象，那么我需要在从 ProcessRecord 返回之前处理它>。

我将 ArgumentTransformationAttribute 与我的参数一起使用，以便从字符串构造我的 IDisposeable 对象，但我找不到任何方法将数据从该类传递到我的 PSCmdlet 关于我是否创建了该对象。例如：

[Cmdlet("Get", "MyDisposeableName")]
public class GetMyDisposeableNameCommand : PSCmdlet
{
    [Parameter(Mandatory = true, Position = 0), MyDisposeableTransformation]
    public MyDisposeable MyDisposeable
    {
        get;
        set;
    }

    protected override void ProcessRecord()
    {
        try
        {
            WriteObject(MyDisposeable.Name);
        }
        finally
        {
            /* Should only dispose MyDisposeable if we created it... */
            MyDisposeable.Dispose();
        }
    }
}

class MyDisposeableTransformationAttribute : ArgumentTransformationAttribute
{
    public override Object Transform(EngineIntrinsics engineIntrinsics, Object input)
    {
        if (input is PSObject && ((PSObject)input).BaseObject is MyDisposeable)
        {
            /* We were passed a MyDisposeable, we should not dispose it */
            return ((PSObject)input).BaseObject;
        }

        /* We created a MyDisposeable, we *should* dispose it */
        return new MyDisposeable(input.ToString());
    }
}

我最好的猜测是对我的 MyDisposeableClass 进行子类化，只是为了标记它需要显式处置，但这看起来相当老套，虽然它在这种情况下有效，但如果我想要处理一个密封类。

有更好的方法吗？

Answer 1

您是否可以向 MyDisposable 类添加一个属性，而不是子类化？

public class MyDisposable
{
    ...   
    public bool IsAttributeCreated { get; set; }
}

然后在你的属性代码中

/* We created a MyDisposeable, we *should* dispose it */
return new MyDisposeable(input.ToString()){IsAttributeCreated=true};

最后在你的finally块中

finally
{
    /* Should only dispose MyDisposeable if we created it... */
    if (MyDisposable.IsAttributeCreated)
        MyDisposeable.Dispose();
}

Answer 2

最后，我只是使用接受包装 MyDisposeable 类型的参数。我最初担心这样做是使用内部类型作为参数会影响函数的可访问性。 （也许这会对文档产生负面影响，但在 cmdlet 中，文档完全由 XML 文件控制。）

经过一些测试，使用内部类作为参数并让转换接受公共类型似乎没有任何问题。所以我只是创建一个包装类：

public class MyDisposeableWrapper
{
    public MyDisposeable MyDisposeable
    {
        get;
        set;
    }

    public bool NeedsDisposed
    {
        get;
        set;
    }

    public MyDisposeableWrapper(MyDisposeable myDisposeable, bool needsDisposed)
    {
        MyDisposeable = myDisposeable;
        NeedsDisposed = needsDisposed;
    }
}

并让参数采用它。在转换属性中，只需根据参数是采用 MyDisposeable 还是构造参数来设置 NeedsDispose 即可。例如：

if(input is MyDisposeable)
{
    return new MyDisposeableWrapper((MyDisposeable) input, false);
}
else
{
    /* construct MyDisposeable from the input */
    return new MyDisposeableWrapper(myDisposeable, true);
}