resolve some bugs caused by mismatched operation types (#258)

Author: WeiXinChan

src/main/java/com/alipay/oceanbase/rpc/mutation/BatchOperation.java

@@ -268,6 +268,9 @@ private BatchOperationResult executeWithNormalBatchOp() throws Exception {
                     query.getSelectColumns().toArray((new String[0])));
             } else if (operation instanceof Get) {
                 Get get = (Get) operation;
+                if (get.getRowKey() == null) {
+                    throw new IllegalArgumentException("RowKey is null in Get operation");
+                }
                 batchOps.get(get.getRowKey().getValues(), get.getSelectColumns());
             } else {
                 throw new ObTableException("unknown operation " + operation);
@@ -314,6 +317,9 @@ private BatchOperationResult executeWithLSBatchOp() throws Exception {
                     }
                 } else if (operation instanceof Get) {
                     Get get = (Get) operation;
+                    if (get.getRowKey() == null) {
+                        throw new IllegalArgumentException("RowKey is null in Get operation");
+                    }
                     batchOps.addOperation(get);
                 } else if (operation instanceof TableQuery) {
                     TableQuery query = (TableQuery) operation;

src/main/java/com/alipay/oceanbase/rpc/table/ObTableClientLSBatchOpsImpl.java

@@ -315,18 +315,22 @@ public List<Object> execute() throws Exception {
      */
     public List<Object> executeWithResult() throws Exception {
         List<Object> results = new ArrayList<Object>(batchOperation.size());
-        for (ObTableSingleOpResult result : executeInternal()) {
+        ObTableSingleOpResult[] singleResults = executeInternal();
+        for (int i = 0; i < singleResults.length; i++) {
+            ObTableSingleOpResult result = singleResults[i];
+            // Sometimes the server does not set the operation type，so we use request operation type
+            ObTableOperationType opType = batchOperation.get(i).getSingleOpType();
             int errCode = result.getHeader().getErrno();
             if (errCode == ResultCodes.OB_SUCCESS.errorCode) {
-                if (result.getOperationType() == ObTableOperationType.GET) {
+                if (opType == ObTableOperationType.GET) {
                     results.add(new GetResult(result));
                 } else {
                     results.add(new MutationResult(result));
                 }
             } else {
                 results.add(ExceptionUtil.convertToObTableException(result.getExecuteHost(),
-                    result.getExecutePort(), result.getSequence(), result.getUniqueId(), errCode,
-                    result.getHeader().getErrMsg()));
+                        result.getExecutePort(), result.getSequence(), result.getUniqueId(), errCode,
+                        result.getHeader().getErrMsg()));
             }
         }
         return results;