Merge pull request #64 from carlbenson/master

Allow headers to come from other row types than strings.
thorbenprimke · Jul 8, 2016 · dd2e594 · dd2e594
2 parents 9687e3b + e811c40
commit dd2e594
Showing 1 changed file with 38 additions and 4 deletions.
diff --git a/library/src/main/java/io/realm/RealmBasedRecyclerViewAdapter.java b/library/src/main/java/io/realm/RealmBasedRecyclerViewAdapter.java
@@ -82,6 +82,7 @@ public RowWrapper(boolean isRealm, int realmIndex, int sectionHeaderIndex, Strin
     private final int LOAD_MORE_VIEW_TYPE = 101;
     private final int FOOTER_VIEW_TYPE = 102;
 
+    private Context context;
     protected LayoutInflater inflater;
     protected RealmResults<T> realmResults;
     protected List ids;
@@ -151,6 +152,7 @@ public RealmBasedRecyclerViewAdapter(
             throw new IllegalArgumentException("Context cannot be null");
         }
 
+        this.context = context;
         this.animateResults = animateResults;
         this.addSectionHeaders = addSectionHeaders;
         this.headerColumnName = headerColumnName;
@@ -231,6 +233,10 @@ public void onBindHeaderViewHolder(RealmViewHolder holder, int position) {
         }
     }
 
+    public Context getContext() {
+        return context;
+    }
+
     /**
      * DON'T OVERRIDE THIS METHOD. Implement onCreateRealmViewHolder instead.
      */
@@ -354,8 +360,20 @@ public void updateRealmResults(RealmResults<T> queryResults) {
      * Method that creates the header string that should be used. Override this method to have
      * a custom header.
      */
-    public String createHeaderFromColumnValue(String columnValue) {
-        return columnValue.substring(0, 1);
+    public String createHeaderFromColumnValue(Object columnValue) {
+        String result = null;
+        if (columnValue instanceof Boolean) {
+            result = columnValue.toString();
+        } else if (columnValue instanceof String) {
+            result = ((String) columnValue).substring(0, 1);
+        } else if (columnValue instanceof Long) {
+            result = columnValue.toString();
+        } else {
+            throw new IllegalStateException("columnType not supported");
+        }
+
+        return result;
+
     }
 
     private List getIdsOfRealmResults() {
@@ -426,8 +444,24 @@ private void updateRowWrappers() {
             final long headerIndex = realmResults.getTable().getColumnIndex(headerColumnName);
             int i = 0;
             for (RealmModel result : realmResults) {
-                String rawHeader = ((RealmObjectProxy) result)
-                        .realmGet$proxyState().getRow$realm().getString(headerIndex);
+                Object rawHeader;
+                RealmFieldType fieldType = ((RealmObjectProxy) result)
+                        .realmGet$proxyState().getRow$realm().getColumnType(headerIndex);
+
+                if (fieldType == RealmFieldType.STRING) {
+                    rawHeader = ((RealmObjectProxy) result)
+                            .realmGet$proxyState().getRow$realm().getString(headerIndex);
+
+                } else if (fieldType == RealmFieldType.BOOLEAN) {
+                    rawHeader = ((RealmObjectProxy) result)
+                            .realmGet$proxyState().getRow$realm().getBoolean(headerIndex);
+                } else if (fieldType == RealmFieldType.INTEGER) {
+                    rawHeader = ((RealmObjectProxy) result)
+                            .realmGet$proxyState().getRow$realm().getLong(headerIndex);
+                } else {
+                    throw new IllegalStateException("columnValue type not supported");
+                }
+
                 String header = createHeaderFromColumnValue(rawHeader);
                 if (!TextUtils.equals(lastHeader, header)) {
                     // Insert new header view and update section data.