Question

通常，我正在尝试创建一个PSCmdlet，它接受一个实现IDisposeable的类型的参数，并且需要处理以避免资源泄漏。我还想为该参数接受string并创建该类型的实例，但是如果我自己创建该对象，那么我需要在从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子类化为标记它需要明确处理，但这看起来相当hacky，虽然它适用于这种情况，但如果我想处理它显然不会起作用密封的课程。

有更好的方法吗？

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;
    }
}

然后让参数取而代之。在transformation属性中，只需根据参数是NeedsDisposed还是构造一个来设置MyDisposeable。例如：

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

从ArgumentTransformationAttribute将数据传递给PSCmdlet

2 个答案: