fix key and range subpartition not support

oceanbase · Jan 14, 2025 · 36966f9 · 36966f9
1 parent 0870e42
commit 36966f9
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/main/java/com/alipay/oceanbase/rpc/location/LocationUtil.java b/src/main/java/com/alipay/oceanbase/rpc/location/LocationUtil.java
@@ -1771,8 +1771,14 @@ private static Map<Long, Long> parseKeyHashPart(ResultSet rs, TableEntry tableEn
                         long subPartNum = rs.getLong("sub_part_num");
                         subHashPartDesc.setPartNum((int) subPartNum);
                     }
+                } else if (subPartDesc instanceof ObRangePartDesc) {
+                    ObRangePartDesc subRangePartDesc = (ObRangePartDesc) subPartDesc;
+                    if (!isSubPart && subRangePartDesc.getPartNum() == 0) {
+                        long subPartNum = rs.getLong("sub_part_num");
+                        subRangePartDesc.setPartNum((int) subPartNum);
+                    }
                 } else {
-                    throw new IllegalArgumentException("sub part desc is not key or hash part desc");
+                    throw new IllegalArgumentException("sub part desc is not key,hash and range part desc");
                 }
             }
             Long tabletId = rs.getLong("tablet_id");