Adding post- hooks for :find and :search in the indirection class.

1 files changed, 8 insertions, 1 deletions

diff --git a/lib/puppet/indirector/indirection.rb b/lib/puppet/indirector/indirection.rb
index 50a27f771..60278be98 100644
--- a/lib/puppet/indirector/indirection.rb
+++ b/lib/puppet/indirector/indirection.rb
@@ -124,6 +124,9 @@ class Puppet::Indirector::Indirection
                 Puppet.info "Caching %s %s" % [self.name, key]
                 cache.save(result, *args)
             end
+
+            terminus(terminus_name).post_find(result) if terminus(terminus_name).respond_to?(:post_find)
+
             return result
         end
     end
@@ -137,7 +140,11 @@ class Puppet::Indirector::Indirection
     def search(*args)
         check_authorization(:search, terminus_class, args)
 
-        terminus.search(*args)
+        result = terminus.search(*args)
+
+        terminus().post_search(result) if terminus().respond_to?(:post_search)
+
+        result
     end
 
     # these become instance methods