TransactionExtend.cs 2.9 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192
  1. using System.Collections.Generic;
  2. using Autodesk.Revit.DB;
  3. namespace SAGA.RevitUtils.Extends
  4. {
  5. public static class TransactionExtend
  6. {
  7. /// <summary>
  8. /// 提交事务时删除所有警告信息
  9. /// </summary>
  10. /// <param name="trans"></param>
  11. /// <param name="deleteWarning"></param>
  12. public static void CommitTransaction(this Transaction trans, bool deleteWarning)
  13. {
  14. if (deleteWarning)
  15. {
  16. trans.CommitTransaction(null);
  17. }
  18. else trans.Commit();
  19. }
  20. /// <summary>
  21. /// 提交事务并删除所有警告
  22. /// </summary>
  23. /// <param name="trans"></param>
  24. /// <param name="failureIdList">所要删除警告集合</param>
  25. public static void CommitTransaction(this Transaction trans, List<FailureDefinitionId> failureIdList = null)
  26. {
  27. FailuresPreprocessor failuresPreprocessor = null;
  28. if (failureIdList == null)
  29. {
  30. failuresPreprocessor = new FailuresPreprocessor();
  31. }
  32. else
  33. {
  34. failuresPreprocessor = new FailuresPreprocessor(failureIdList);
  35. }
  36. FailureHandlingOptions failureOptions = trans.GetFailureHandlingOptions();
  37. failureOptions.SetFailuresPreprocessor(failuresPreprocessor);
  38. trans.SetFailureHandlingOptions(failureOptions);
  39. trans.Commit(failureOptions);
  40. }
  41. }
  42. public class FailuresPreprocessor : IFailuresPreprocessor
  43. {
  44. private List<FailureDefinitionId> _failureIdList = new List<FailureDefinitionId>();
  45. public FailuresPreprocessor()
  46. {
  47. }
  48. public FailuresPreprocessor(FailureDefinitionId id)
  49. {
  50. _failureIdList.Add(id);
  51. }
  52. public FailuresPreprocessor(List<FailureDefinitionId> idList)
  53. {
  54. this._failureIdList = idList;
  55. }
  56. FailureProcessingResult IFailuresPreprocessor.PreprocessFailures(FailuresAccessor failuresAccessor)
  57. {
  58. IList<FailureMessageAccessor> failList = new List<FailureMessageAccessor>();
  59. failList = failuresAccessor.GetFailureMessages();
  60. if (_failureIdList.Count == 0)
  61. {
  62. failuresAccessor.DeleteAllWarnings();//删除所有警告信息
  63. }
  64. else
  65. {
  66. //删除部分警告信息
  67. foreach (FailureMessageAccessor failure in failList)
  68. {
  69. FailureDefinitionId failId = failure.GetFailureDefinitionId();
  70. if (_failureIdList.Exists(p => p == failId))
  71. {
  72. failuresAccessor.DeleteWarning(failure);
  73. }
  74. }
  75. }
  76. return FailureProcessingResult.Continue;
  77. }
  78. }
  79. }