From 2cbc51cfdffa24326fb49180098cd1124b009e8b Mon Sep 17 00:00:00 2001
From: Redkale <22250530@qq.com>
Date: Wed, 14 Mar 2018 21:21:23 +0800
Subject: [PATCH]

---
 src/org/redkale/source/EntityCache.java | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/org/redkale/source/EntityCache.java b/src/org/redkale/source/EntityCache.java
index b55f96d27..7e5d354cd 100644
--- a/src/org/redkale/source/EntityCache.java
+++ b/src/org/redkale/source/EntityCache.java
@@ -638,6 +638,12 @@ public final class EntityCache<T> {
             } else if (ft == AtomicLong.class) {
                 newval = new AtomicLong(numb.longValue());
             }
+        } else {
+            if (ft == AtomicInteger.class && newval != null && newval.getClass() != AtomicInteger.class) {
+                newval = new AtomicInteger(((Number) newval).intValue());
+            } else if (ft == AtomicLong.class && newval != null && newval.getClass() != AtomicLong.class) {
+                newval = new AtomicLong(((Number) newval).longValue());
+            }
         }
         attr.set(rs, (V) newval);
         return rs;