From 4b722c2d9a85244abb3c30e646c5bb3b4ba1c00e Mon Sep 17 00:00:00 2001
From: Erik Brakkee <erik@brakkee.org>
Date: Fri, 30 Jul 2010 19:08:09 +0000
Subject: [PATCH]

---
 .../ComponentInstantiationInjector.java       |  1 +
 .../wicket/inject/InjectionBehavior.java      | 47 +++++++++++++++++++
 2 files changed, 48 insertions(+)
 create mode 100644 wicket/joe/src/main/java/org/wamblee/wicket/inject/InjectionBehavior.java

diff --git a/wicket/joe/src/main/java/org/wamblee/wicket/inject/ComponentInstantiationInjector.java b/wicket/joe/src/main/java/org/wamblee/wicket/inject/ComponentInstantiationInjector.java
index 9c604ebe..a4a0dcf9 100644
--- a/wicket/joe/src/main/java/org/wamblee/wicket/inject/ComponentInstantiationInjector.java
+++ b/wicket/joe/src/main/java/org/wamblee/wicket/inject/ComponentInstantiationInjector.java
@@ -59,6 +59,7 @@ public class ComponentInstantiationInjector implements
     @Override
     public void onInstantiation(Component aComponent) {
         injector.inject(aComponent);
+        aComponent.add(new InjectionBehavior());
     }
 
 }
diff --git a/wicket/joe/src/main/java/org/wamblee/wicket/inject/InjectionBehavior.java b/wicket/joe/src/main/java/org/wamblee/wicket/inject/InjectionBehavior.java
new file mode 100644
index 00000000..3d648c4e
--- /dev/null
+++ b/wicket/joe/src/main/java/org/wamblee/wicket/inject/InjectionBehavior.java
@@ -0,0 +1,47 @@
+/*
+ * Copyright 2005-2010 the original author or authors.
+ * 
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ * 
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.wamblee.wicket.inject;
+
+import org.apache.wicket.Component;
+import org.apache.wicket.behavior.AbstractBehavior;
+import org.wamblee.inject.InjectorBuilder;
+
+/**
+ * Injection behavior that performs dependency injection after
+ * serialization/deserialisation of the object. 
+ * 
+ * @author Erik Brakkee
+ *
+ */
+public class InjectionBehavior extends AbstractBehavior {
+
+    private transient boolean injectionUptodate;
+    
+    /**
+     * Constructs the behavior.
+     */
+    public InjectionBehavior() { 
+        injectionUptodate = true; 
+    }
+    
+    @Override
+    public void beforeRender(Component aComponent) {
+        if (!injectionUptodate) { 
+            InjectorBuilder.getInjector().inject(aComponent);
+            injectionUptodate = true; 
+        }
+    }
+}
-- 
2.31.1