From 50c9363876ba9e49d73429f3822530e47493c9f9 Mon Sep 17 00:00:00 2001
From: Redkale <8730487+redkale@users.noreply.github.com>
Date: Wed, 5 Dec 2018 17:59:56 +0800
Subject: [PATCH]

---
 src/org/redkale/net/http/WebSocketServlet.java | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/org/redkale/net/http/WebSocketServlet.java b/src/org/redkale/net/http/WebSocketServlet.java
index f889d2744..e264efed6 100644
--- a/src/org/redkale/net/http/WebSocketServlet.java
+++ b/src/org/redkale/net/http/WebSocketServlet.java
@@ -106,6 +106,9 @@ public abstract class WebSocketServlet extends HttpServlet implements Resourcabl
     @Resource(name = "$")
     protected WebSocketNode node;
 
+    @Resource(name = "SERVER_RESFACTORY")
+    protected ResourceFactory resourceFactory;
+
     protected WebSocketServlet() {
         Type msgtype = String.class;
         try {
@@ -145,6 +148,7 @@ public abstract class WebSocketServlet extends HttpServlet implements Resourcabl
                 if (cryptorClass != null && !cryptorClass.isEmpty()) {
                     try {
                         this.cryptor = (Cryptor) Thread.currentThread().getContextClassLoader().loadClass(cryptorClass).getDeclaredConstructor().newInstance();
+                        if (resourceFactory != null && this.cryptor != null) resourceFactory.inject(this.cryptor);
                     } catch (Exception e) {
                         throw new RuntimeException(e);
                     }