通常,我正在尝试创建一个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,虽然它适用于这种情况,但如果我想处理它显然不会起作用密封的课程。
有更好的方法吗?
答案 0 :(得分:0)
您可以在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();
}
答案 1 :(得分:0)
最后,我只使用接受包裹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);
}