From bcd073a4025add29c68b7e6a48834f5df75b7e75 Mon Sep 17 00:00:00 2001
From: Lukas Eder <lukas.eder@gmail.com>
Date: Fri, 6 Dec 2013 11:29:23 +0100
Subject: [PATCH] [#2869] DefaultRecordMapper should attach resulting records
 according to Settings.attachRecords, if target type implements Attachable

---
 .../org/jooq/test/_/testcases/FetchTests.java | 22 ++++++++++++++++++-
 .../src/org/jooq/test/jOOQAbstractTest.java   |  5 +++++
 .../org/jooq/impl/DefaultRecordMapper.java    | 19 +++++++++++++++-
 jOOQ/src/main/java/org/jooq/impl/Utils.java   |  4 ++--
 4 files changed, 46 insertions(+), 4 deletions(-)

diff --git a/jOOQ-test/src/org/jooq/test/_/testcases/FetchTests.java b/jOOQ-test/src/org/jooq/test/_/testcases/FetchTests.java
index 254169da46..33e7be40f3 100644
--- a/jOOQ-test/src/org/jooq/test/_/testcases/FetchTests.java
+++ b/jOOQ-test/src/org/jooq/test/_/testcases/FetchTests.java
@@ -77,6 +77,7 @@ import java.util.Queue;
 
 import junit.framework.Assert;
 
+import org.jooq.AttachableInternal;
 import org.jooq.Cursor;
 import org.jooq.DSLContext;
 import org.jooq.Field;
@@ -95,6 +96,7 @@ import org.jooq.Select;
 import org.jooq.SelectQuery;
 import org.jooq.TableRecord;
 import org.jooq.UpdatableRecord;
+import org.jooq.conf.Settings;
 import org.jooq.exception.DataAccessException;
 import org.jooq.exception.InvalidResultException;
 import org.jooq.exception.MappingException;
@@ -1193,7 +1195,6 @@ extends BaseTest<A, AP, B, S, B2S, BS, L, X, DATE, BOOL, D, T, U, UU, I, IPK, T7
 
     @Test
     public void testFetchIntoTableRecords() throws Exception {
-        jOOQAbstractTest.reset = false;
 
         // [#1819] Check if only applicable setters are used
         // JOIN two tables into a generated UpdatableRecord
@@ -1219,6 +1220,25 @@ extends BaseTest<A, AP, B, S, B2S, BS, L, X, DATE, BOOL, D, T, U, UU, I, IPK, T7
         }
     }
 
+    @Test
+    public void testFetchAttachables() throws Exception {
+        // [#2869] DefaultRecordMapper should recognise Attachable types and attach them
+        // according to the Settings.
+
+        B b1 = create().selectFrom(TBook()).where(TBook_ID().eq(1)).fetchOne();
+        B b2 = b1.into(TBook().getRecordType());
+        
+        assertNotNull(((AttachableInternal) b1).configuration());
+        assertNotNull(((AttachableInternal) b2).configuration());
+        
+        B b3 = create(new Settings().withAttachRecords(false))
+                    .selectFrom(TBook()).where(TBook_ID().eq(1)).fetchOne();
+        B b4 = b3.into(TBook().getRecordType());
+        
+        assertNull(((AttachableInternal) b3).configuration());
+        assertNull(((AttachableInternal) b4).configuration());
+    }
+
     @Test
     public void testFetchIntoTableRecordsWithUDTs() throws Exception {
         if (cUAddressType() == null) {
diff --git a/jOOQ-test/src/org/jooq/test/jOOQAbstractTest.java b/jOOQ-test/src/org/jooq/test/jOOQAbstractTest.java
index 50084b4a01..37ef1b8a7f 100644
--- a/jOOQ-test/src/org/jooq/test/jOOQAbstractTest.java
+++ b/jOOQ-test/src/org/jooq/test/jOOQAbstractTest.java
@@ -1495,6 +1495,11 @@ public abstract class jOOQAbstractTest<
         new FetchTests(this).testFetchIntoTableRecords();
     }
 
+    @Test
+    public void testFetchAttachables() throws Exception {
+        new FetchTests(this).testFetchAttachables();
+    }
+
     @Test
     public void testFetchIntoTableRecordsWithUDTs() throws Exception {
         new FetchTests(this).testFetchIntoTableRecordsWithUDTs();
diff --git a/jOOQ/src/main/java/org/jooq/impl/DefaultRecordMapper.java b/jOOQ/src/main/java/org/jooq/impl/DefaultRecordMapper.java
index 4b28e4bdc6..3c41c471ad 100644
--- a/jOOQ/src/main/java/org/jooq/impl/DefaultRecordMapper.java
+++ b/jOOQ/src/main/java/org/jooq/impl/DefaultRecordMapper.java
@@ -62,6 +62,8 @@ import java.util.List;
 
 import javax.persistence.Column;
 
+import org.jooq.Attachable;
+import org.jooq.AttachableInternal;
 import org.jooq.Configuration;
 import org.jooq.Field;
 import org.jooq.Record;
@@ -278,7 +280,7 @@ public class DefaultRecordMapper<R extends Record, E> implements RecordMapper<R,
         }
 
         try {
-            return delegate.map(record);
+            return attach(delegate.map(record), record);
         }
 
         // Pass MappingExceptions on to client code
@@ -543,4 +545,19 @@ public class DefaultRecordMapper<R extends Record, E> implements RecordMapper<R,
             }
         }
     }
+
+    private static <E> E attach(E attachable, Record record) {
+        // [#2869] Attach the mapped outcome if it is Attachable and if the context's
+        // Settings.attachRecords flag is set
+        if (attachable instanceof Attachable && record instanceof AttachableInternal) {
+            Attachable a = (Attachable) attachable;
+            AttachableInternal r = (AttachableInternal) record;
+
+            if (Utils.attachRecords(r.configuration())) {
+                a.attach(r.configuration());
+            }
+        }
+
+        return attachable;
+    }
 }
diff --git a/jOOQ/src/main/java/org/jooq/impl/Utils.java b/jOOQ/src/main/java/org/jooq/impl/Utils.java
index 77c2b11947..3aabeec262 100644
--- a/jOOQ/src/main/java/org/jooq/impl/Utils.java
+++ b/jOOQ/src/main/java/org/jooq/impl/Utils.java
@@ -405,7 +405,7 @@ final class Utils {
         return configuration(configuration).settings();
     }
 
-    private static final boolean attachRecords(Configuration configuration) {
+    static final boolean attachRecords(Configuration configuration) {
         if (configuration != null) {
             Settings settings = configuration.settings();
 
@@ -1293,7 +1293,7 @@ final class Utils {
      */
     static final Field<String> escapeForLike(Object value, Configuration configuration) {
         if (value != null && value.getClass() == String.class) {
-            
+
             /* [pro] xx
             xx xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx xx xxxxxxx x
                 xxxxxx xxxxxxx x xxxxx x xxxxx