Fixing issue chennaione#243

Conflicts: library/build.gradle sugartestapp/src/androidTest/java/me/jivimberg/android/sugartestapp/robolectric/SugarRecordTest.java sugartestapp/src/main/java/me/jivimberg/android/sugartestapp/model/annotated/Car.java
jivimberg · Feb 1, 2015 · 7e7df2f · 7e7df2f
1 parent eadd351
commit 7e7df2f
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 0 deletions.
diff --git a/library/build.gradle b/library/build.gradle
@@ -12,6 +12,7 @@ buildscript {
 
     dependencies {
         compile 'com.android.support:support-v4:21.0.3'
+        compile group: 'com.google.guava', name: 'guava', version: '12.0'
     }
 
     android {

diff --git a/library/src/com/orm/SugarContext.java b/library/src/com/orm/SugarContext.java
@@ -2,15 +2,21 @@
 
 import android.content.Context;
 
+import com.google.common.collect.MapMaker;
+
+import java.util.concurrent.ConcurrentMap;
+
 public class SugarContext {
 
     private static SugarContext instance = null;
     private SugarDb sugarDb;
     private Context context;
+    private ConcurrentMap<Object, Long> entitiesMap;
 
     private SugarContext(Context context) {
         this.context = context;
         this.sugarDb = new SugarDb(context);
+        this.entitiesMap = new MapMaker().weakKeys().makeMap();
     }
 
     public static SugarContext getSugarContext() {
@@ -47,4 +53,7 @@ protected SugarDb getSugarDb() {
         return sugarDb;
     }
 
+    ConcurrentMap<Object, Long> getEntitiesMap() {
+        return entitiesMap;
+    }
 }
diff --git a/library/src/com/orm/SugarRecord.java b/library/src/com/orm/SugarRecord.java
@@ -180,23 +180,39 @@ public static long save(Object object) {
     }
 
     static long save(SQLiteDatabase db, Object object) {
+        Map<Object, Long> entitiesMap = getSugarContext().getEntitiesMap();
         List<Field> columns = ReflectionUtil.getTableFields(object.getClass());
         ContentValues values = new ContentValues(columns.size());
         for (Field column : columns) {
             ReflectionUtil.addFieldValueToColumn(values, column, object);
         }
 
+        boolean isSugarEntity = isSugarEntity(object);
+        if (isSugarEntity && entitiesMap.containsKey(object)) {
+                values.put("id", entitiesMap.get(object));
+        }
+
         long id = db.insertWithOnConflict(NamingHelper.toSQLName(object.getClass()), null, values,
                 SQLiteDatabase.CONFLICT_REPLACE);
 
+        //TODO should we remove the id field from SugarRecord?
         if (SugarRecord.class.isAssignableFrom(object.getClass())) {
             ReflectionUtil.setFieldValueForId(object, id);
         }
+
+        if (isSugarEntity) {
+            entitiesMap.put(object, id);
+        }
+
         Log.i("Sugar", object.getClass().getSimpleName() + " saved : " + id);
 
         return id;
     }
 
+    private static boolean isSugarEntity(Object object) {
+        return object.getClass().isAnnotationPresent(Table.class) || SugarRecord.class.isAssignableFrom(object.getClass());
+    }
+
     private static void inflate(Cursor cursor, Object object) {
         List<Field> columns = ReflectionUtil.getTableFields(object.getClass());