feature: support hlen protocol (#28)

funky-eyes · Apr 18, 2024 · 192c9cf · 192c9cf
1 parent c66b872
commit 192c9cf
Show file tree

Hide file tree

Showing 37 changed files with 219 additions and 111 deletions.
diff --git a/src/main/java/icu/funkye/redispike/handler/RedisCommandHandler.java b/src/main/java/icu/funkye/redispike/handler/RedisCommandHandler.java
@@ -16,38 +16,36 @@
  */
 package icu.funkye.redispike.handler;
 
-import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
-import java.util.concurrent.Executor;
 import java.util.concurrent.ExecutorService;
-import java.util.concurrent.Executors;
 import java.util.concurrent.ForkJoinPool;
 import com.alipay.remoting.CommandCode;
 import com.alipay.remoting.CommandHandler;
 import com.alipay.remoting.RemotingContext;
 import com.alipay.remoting.RemotingProcessor;
 import icu.funkye.redispike.handler.process.impl.GetRequestProcessor;
-import icu.funkye.redispike.handler.process.impl.HDelRequestProcessor;
-import icu.funkye.redispike.handler.process.impl.HExistsRequestProcessor;
-import icu.funkye.redispike.handler.process.impl.HGetAllRequestProcessor;
-import icu.funkye.redispike.handler.process.impl.HGetRequestProcessor;
-import icu.funkye.redispike.handler.process.impl.HIncrbyRequestProcessor;
-import icu.funkye.redispike.handler.process.impl.HIncrbyfloatRequestProcessor;
-import icu.funkye.redispike.handler.process.impl.HMgetRequestProcessor;
-import icu.funkye.redispike.handler.process.impl.HSetRequestProcessor;
-import icu.funkye.redispike.handler.process.impl.HValsRequestProcessor;
+import icu.funkye.redispike.handler.process.impl.hash.HDelRequestProcessor;
+import icu.funkye.redispike.handler.process.impl.hash.HExistsRequestProcessor;
+import icu.funkye.redispike.handler.process.impl.hash.HGetAllRequestProcessor;
+import icu.funkye.redispike.handler.process.impl.hash.HGetRequestProcessor;
+import icu.funkye.redispike.handler.process.impl.hash.HIncrbyRequestProcessor;
+import icu.funkye.redispike.handler.process.impl.hash.HIncrbyfloatRequestProcessor;
+import icu.funkye.redispike.handler.process.impl.hash.HLenRequestProcessor;
+import icu.funkye.redispike.handler.process.impl.hash.HMgetRequestProcessor;
+import icu.funkye.redispike.handler.process.impl.hash.HSetRequestProcessor;
+import icu.funkye.redispike.handler.process.impl.hash.HValsRequestProcessor;
 import icu.funkye.redispike.handler.process.impl.KeysRequestProcessor;
-import icu.funkye.redispike.handler.process.impl.SCardRequestProcessor;
-import icu.funkye.redispike.handler.process.impl.SPopRequestProcessor;
-import icu.funkye.redispike.handler.process.impl.SRandmemberRequestProcessor;
-import icu.funkye.redispike.handler.process.impl.SRemRequestProcessor;
+import icu.funkye.redispike.handler.process.impl.set.SCardRequestProcessor;
+import icu.funkye.redispike.handler.process.impl.set.SPopRequestProcessor;
+import icu.funkye.redispike.handler.process.impl.set.SRandmemberRequestProcessor;
+import icu.funkye.redispike.handler.process.impl.set.SRemRequestProcessor;
 import icu.funkye.redispike.handler.process.impl.SetRequestProcessor;
 import icu.funkye.redispike.handler.process.impl.CommandRequestProcessor;
 import icu.funkye.redispike.handler.process.impl.DelRequestProcessor;
-import icu.funkye.redispike.handler.process.impl.SAddRequestProcessor;
-import icu.funkye.redispike.handler.process.impl.SMembersRequestProcessor;
+import icu.funkye.redispike.handler.process.impl.set.SAddRequestProcessor;
+import icu.funkye.redispike.handler.process.impl.set.SMembersRequestProcessor;
 import icu.funkye.redispike.protocol.RedisRequest;
 import icu.funkye.redispike.protocol.response.BulkResponse;
 import org.slf4j.Logger;
@@ -100,6 +98,8 @@ public RedisCommandHandler() {
         processorMap.put(hIncrbyRequestProcessor.getCmdCode().value(), hIncrbyRequestProcessor);
         HIncrbyfloatRequestProcessor hIncrbyfloatRequestProcessor = new HIncrbyfloatRequestProcessor();
         processorMap.put(hIncrbyfloatRequestProcessor.getCmdCode().value(), hIncrbyfloatRequestProcessor);
+        HLenRequestProcessor hLenRequestProcessor = new HLenRequestProcessor();
+        processorMap.put(hLenRequestProcessor.getCmdCode().value(), hLenRequestProcessor);
     }
 
     @Override

diff --git a/...er/process/impl/HDelRequestProcessor.java → ...ocess/impl/hash/HDelRequestProcessor.java b/...er/process/impl/HDelRequestProcessor.java → ...ocess/impl/hash/HDelRequestProcessor.java
@@ -14,7 +14,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-package icu.funkye.redispike.handler.process.impl;
+package icu.funkye.redispike.handler.process.impl.hash;
 
 import java.util.ArrayList;
 import java.util.List;
@@ -30,7 +30,7 @@
 import icu.funkye.redispike.factory.AeroSpikeClientFactory;
 import icu.funkye.redispike.handler.process.AbstractRedisRequestProcessor;
 import icu.funkye.redispike.protocol.RedisRequestCommandCode;
-import icu.funkye.redispike.protocol.request.HDelRequest;
+import icu.funkye.redispike.protocol.request.hash.HDelRequest;
 import icu.funkye.redispike.util.IntegerUtils;
 
 public class HDelRequestProcessor extends AbstractRedisRequestProcessor<HDelRequest> {

diff --git a/...process/impl/HExistsRequestProcessor.java → ...ss/impl/hash/HExistsRequestProcessor.java b/...process/impl/HExistsRequestProcessor.java → ...ss/impl/hash/HExistsRequestProcessor.java
@@ -14,9 +14,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-package icu.funkye.redispike.handler.process.impl;
-
-import java.util.Optional;
+package icu.funkye.redispike.handler.process.impl.hash;
 
 import com.aerospike.client.AerospikeException;
 import com.aerospike.client.Key;
@@ -27,7 +25,7 @@
 import icu.funkye.redispike.factory.AeroSpikeClientFactory;
 import icu.funkye.redispike.handler.process.AbstractRedisRequestProcessor;
 import icu.funkye.redispike.protocol.RedisRequestCommandCode;
-import icu.funkye.redispike.protocol.request.HExistsRequest;
+import icu.funkye.redispike.protocol.request.hash.HExistsRequest;
 import icu.funkye.redispike.util.IntegerUtils;
 
 public class HExistsRequestProcessor extends AbstractRedisRequestProcessor<HExistsRequest> {

diff --git a/...process/impl/HGetAllRequestProcessor.java → ...ss/impl/hash/HGetAllRequestProcessor.java b/...process/impl/HGetAllRequestProcessor.java → ...ss/impl/hash/HGetAllRequestProcessor.java
@@ -14,20 +14,18 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-package icu.funkye.redispike.handler.process.impl;
+package icu.funkye.redispike.handler.process.impl.hash;
 
 import com.aerospike.client.AerospikeException;
 import com.aerospike.client.Key;
 import com.aerospike.client.Record;
 import com.aerospike.client.listener.RecordListener;
 import com.alipay.remoting.RemotingContext;
-import com.alipay.sofa.common.profile.StringUtil;
 
 import icu.funkye.redispike.factory.AeroSpikeClientFactory;
 import icu.funkye.redispike.handler.process.AbstractRedisRequestProcessor;
 import icu.funkye.redispike.protocol.RedisRequestCommandCode;
-import icu.funkye.redispike.protocol.request.HGetAllRequest;
-import icu.funkye.redispike.protocol.request.HGetRequest;
+import icu.funkye.redispike.protocol.request.hash.HGetAllRequest;
 import icu.funkye.redispike.util.IntegerUtils;
 
 public class HGetAllRequestProcessor extends AbstractRedisRequestProcessor<HGetAllRequest> {

diff --git a/...er/process/impl/HGetRequestProcessor.java → ...ocess/impl/hash/HGetRequestProcessor.java b/...er/process/impl/HGetRequestProcessor.java → ...ocess/impl/hash/HGetRequestProcessor.java
@@ -14,19 +14,18 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-package icu.funkye.redispike.handler.process.impl;
+package icu.funkye.redispike.handler.process.impl.hash;
 
 import com.aerospike.client.AerospikeException;
 import com.aerospike.client.Key;
 import com.aerospike.client.Record;
 import com.aerospike.client.listener.RecordListener;
 import com.alipay.remoting.RemotingContext;
-import com.alipay.sofa.common.profile.StringUtil;
 
 import icu.funkye.redispike.factory.AeroSpikeClientFactory;
 import icu.funkye.redispike.handler.process.AbstractRedisRequestProcessor;
 import icu.funkye.redispike.protocol.RedisRequestCommandCode;
-import icu.funkye.redispike.protocol.request.HGetRequest;
+import icu.funkye.redispike.protocol.request.hash.HGetRequest;
 import icu.funkye.redispike.util.IntegerUtils;
 
 public class HGetRequestProcessor extends AbstractRedisRequestProcessor<HGetRequest> {

diff --git a/...process/impl/HIncrbyRequestProcessor.java → ...ss/impl/hash/HIncrbyRequestProcessor.java b/...process/impl/HIncrbyRequestProcessor.java → ...ss/impl/hash/HIncrbyRequestProcessor.java
@@ -14,28 +14,21 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-package icu.funkye.redispike.handler.process.impl;
-
-import java.util.ArrayList;
-import java.util.List;
+package icu.funkye.redispike.handler.process.impl.hash;
 
 import com.aerospike.client.AerospikeException;
 import com.aerospike.client.Bin;
 import com.aerospike.client.Key;
 import com.aerospike.client.Operation;
 import com.aerospike.client.Record;
 import com.aerospike.client.listener.RecordListener;
-import com.aerospike.client.listener.WriteListener;
-import com.aerospike.client.policy.RecordExistsAction;
 import com.aerospike.client.policy.WritePolicy;
 import com.alipay.remoting.RemotingContext;
 
 import icu.funkye.redispike.factory.AeroSpikeClientFactory;
 import icu.funkye.redispike.handler.process.AbstractRedisRequestProcessor;
 import icu.funkye.redispike.protocol.RedisRequestCommandCode;
-import icu.funkye.redispike.protocol.request.HIncrbyRequest;
-import icu.funkye.redispike.protocol.request.HSetRequest;
-import icu.funkye.redispike.protocol.request.conts.Operate;
+import icu.funkye.redispike.protocol.request.hash.HIncrbyRequest;
 import icu.funkye.redispike.util.IntegerUtils;
 
 public class HIncrbyRequestProcessor extends AbstractRedisRequestProcessor<HIncrbyRequest> {
@@ -44,7 +37,6 @@ public class HIncrbyRequestProcessor extends AbstractRedisRequestProcessor<HIncr
     public HIncrbyRequestProcessor() {
         this.cmdCode = new RedisRequestCommandCode(IntegerUtils.hashCodeToShort(HIncrbyRequest.class.hashCode()));
         this.defaultWritePolicy = client.getWritePolicyDefault();
-        this.defaultWritePolicy.sendKey = true;
     }
 
     @Override

diff --git a/...ss/impl/HIncrbyfloatRequestProcessor.java → ...pl/hash/HIncrbyfloatRequestProcessor.java b/...ss/impl/HIncrbyfloatRequestProcessor.java → ...pl/hash/HIncrbyfloatRequestProcessor.java
@@ -14,7 +14,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-package icu.funkye.redispike.handler.process.impl;
+package icu.funkye.redispike.handler.process.impl.hash;
 
 import com.aerospike.client.AerospikeException;
 import com.aerospike.client.Bin;
@@ -28,8 +28,7 @@
 import icu.funkye.redispike.factory.AeroSpikeClientFactory;
 import icu.funkye.redispike.handler.process.AbstractRedisRequestProcessor;
 import icu.funkye.redispike.protocol.RedisRequestCommandCode;
-import icu.funkye.redispike.protocol.request.HIncrbyRequest;
-import icu.funkye.redispike.protocol.request.HIncrbyfloatRequest;
+import icu.funkye.redispike.protocol.request.hash.HIncrbyfloatRequest;
 import icu.funkye.redispike.util.IntegerUtils;
 
 public class HIncrbyfloatRequestProcessor extends AbstractRedisRequestProcessor<HIncrbyfloatRequest> {

diff --git a/src/main/java/icu/funkye/redispike/handler/process/impl/hash/HLenRequestProcessor.java b/src/main/java/icu/funkye/redispike/handler/process/impl/hash/HLenRequestProcessor.java
@@ -0,0 +1,68 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package icu.funkye.redispike.handler.process.impl.hash;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+
+import com.aerospike.client.AerospikeException;
+import com.aerospike.client.Bin;
+import com.aerospike.client.Key;
+import com.aerospike.client.Language;
+import com.aerospike.client.Record;
+import com.aerospike.client.listener.DeleteListener;
+import com.aerospike.client.listener.ExecuteListener;
+import com.aerospike.client.listener.RecordListener;
+import com.aerospike.client.listener.WriteListener;
+import com.aerospike.client.task.RegisterTask;
+import com.alipay.remoting.RemotingContext;
+
+import icu.funkye.redispike.factory.AeroSpikeClientFactory;
+import icu.funkye.redispike.handler.process.AbstractRedisRequestProcessor;
+import icu.funkye.redispike.protocol.RedisRequestCommandCode;
+import icu.funkye.redispike.protocol.request.hash.HDelRequest;
+import icu.funkye.redispike.protocol.request.hash.HLenRequest;
+import icu.funkye.redispike.util.IntegerUtils;
+
+public class HLenRequestProcessor extends AbstractRedisRequestProcessor<HLenRequest> {
+
+    public HLenRequestProcessor() {
+        this.cmdCode = new RedisRequestCommandCode(IntegerUtils.hashCodeToShort(HLenRequest.class.hashCode()));
+        RegisterTask task = client.register(null, this.getClass().getClassLoader(), "lua/hlen.lua", "hlen.lua",
+            Language.LUA);
+        task.waitTillComplete();
+    }
+
+    @Override
+    public void handle(RemotingContext ctx, HLenRequest request) {
+        Key key = new Key(AeroSpikeClientFactory.namespace, AeroSpikeClientFactory.set, request.getKey());
+        client.execute(AeroSpikeClientFactory.eventLoops.next(), new ExecuteListener() {
+            @Override
+            public void onSuccess(Key key, Object obj) {
+                request.setResponse(obj.toString());
+                write(ctx, request);
+            }
+
+            @Override
+            public void onFailure(AerospikeException exception) {
+                logger.error(exception.getMessage(), exception);
+                write(ctx, request);
+            }
+        }, client.getWritePolicyDefault(), key, "hlen", "hash_count_bins");
+    }
+}
diff --git a/...r/process/impl/HMgetRequestProcessor.java → ...cess/impl/hash/HMgetRequestProcessor.java b/...r/process/impl/HMgetRequestProcessor.java → ...cess/impl/hash/HMgetRequestProcessor.java
@@ -14,7 +14,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-package icu.funkye.redispike.handler.process.impl;
+package icu.funkye.redispike.handler.process.impl.hash;
 
 import java.util.Optional;
 import com.aerospike.client.AerospikeException;
@@ -26,7 +26,7 @@
 import icu.funkye.redispike.factory.AeroSpikeClientFactory;
 import icu.funkye.redispike.handler.process.AbstractRedisRequestProcessor;
 import icu.funkye.redispike.protocol.RedisRequestCommandCode;
-import icu.funkye.redispike.protocol.request.HMgetRequest;
+import icu.funkye.redispike.protocol.request.hash.HMgetRequest;
 import icu.funkye.redispike.util.IntegerUtils;
 
 public class HMgetRequestProcessor extends AbstractRedisRequestProcessor<HMgetRequest> {

diff --git a/...er/process/impl/HSetRequestProcessor.java → ...ocess/impl/hash/HSetRequestProcessor.java b/...er/process/impl/HSetRequestProcessor.java → ...ocess/impl/hash/HSetRequestProcessor.java
@@ -14,7 +14,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-package icu.funkye.redispike.handler.process.impl;
+package icu.funkye.redispike.handler.process.impl.hash;
 
 import java.util.ArrayList;
 import java.util.List;
@@ -29,7 +29,7 @@
 import icu.funkye.redispike.factory.AeroSpikeClientFactory;
 import icu.funkye.redispike.handler.process.AbstractRedisRequestProcessor;
 import icu.funkye.redispike.protocol.RedisRequestCommandCode;
-import icu.funkye.redispike.protocol.request.HSetRequest;
+import icu.funkye.redispike.protocol.request.hash.HSetRequest;
 import icu.funkye.redispike.protocol.request.conts.Operate;
 import icu.funkye.redispike.util.IntegerUtils;
 

diff --git a/...r/process/impl/HValsRequestProcessor.java → ...cess/impl/hash/HValsRequestProcessor.java b/...r/process/impl/HValsRequestProcessor.java → ...cess/impl/hash/HValsRequestProcessor.java
@@ -14,7 +14,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-package icu.funkye.redispike.handler.process.impl;
+package icu.funkye.redispike.handler.process.impl.hash;
 
 import java.util.Optional;
 import com.aerospike.client.AerospikeException;
@@ -26,7 +26,7 @@
 import icu.funkye.redispike.factory.AeroSpikeClientFactory;
 import icu.funkye.redispike.handler.process.AbstractRedisRequestProcessor;
 import icu.funkye.redispike.protocol.RedisRequestCommandCode;
-import icu.funkye.redispike.protocol.request.HValsRequest;
+import icu.funkye.redispike.protocol.request.hash.HValsRequest;
 import icu.funkye.redispike.util.IntegerUtils;
 
 public class HValsRequestProcessor extends AbstractRedisRequestProcessor<HValsRequest> {

diff --git a/...er/process/impl/SAddRequestProcessor.java → ...rocess/impl/set/SAddRequestProcessor.java b/...er/process/impl/SAddRequestProcessor.java → ...rocess/impl/set/SAddRequestProcessor.java
@@ -14,7 +14,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-package icu.funkye.redispike.handler.process.impl;
+package icu.funkye.redispike.handler.process.impl.set;
 
 import java.util.ArrayList;
 import java.util.List;
@@ -29,7 +29,7 @@
 import icu.funkye.redispike.factory.AeroSpikeClientFactory;
 import icu.funkye.redispike.handler.process.AbstractRedisRequestProcessor;
 import icu.funkye.redispike.protocol.RedisRequestCommandCode;
-import icu.funkye.redispike.protocol.request.SAddRequest;
+import icu.funkye.redispike.protocol.request.set.SAddRequest;
 import icu.funkye.redispike.util.IntegerUtils;
 
 public class SAddRequestProcessor extends AbstractRedisRequestProcessor<SAddRequest> {

diff --git a/...r/process/impl/SCardRequestProcessor.java → ...ocess/impl/set/SCardRequestProcessor.java b/...r/process/impl/SCardRequestProcessor.java → ...ocess/impl/set/SCardRequestProcessor.java
@@ -14,21 +14,19 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-package icu.funkye.redispike.handler.process.impl;
+package icu.funkye.redispike.handler.process.impl.set;
 
 import com.aerospike.client.AerospikeException;
 import com.aerospike.client.Key;
 import com.aerospike.client.Language;
-import com.aerospike.client.Value;
 import com.aerospike.client.listener.ExecuteListener;
 import com.aerospike.client.task.RegisterTask;
 import com.alipay.remoting.RemotingContext;
 
 import icu.funkye.redispike.factory.AeroSpikeClientFactory;
 import icu.funkye.redispike.handler.process.AbstractRedisRequestProcessor;
 import icu.funkye.redispike.protocol.RedisRequestCommandCode;
-import icu.funkye.redispike.protocol.request.SCardRequest;
-import icu.funkye.redispike.protocol.request.SRemRequest;
+import icu.funkye.redispike.protocol.request.set.SCardRequest;
 import icu.funkye.redispike.util.IntegerUtils;
 
 public class SCardRequestProcessor extends AbstractRedisRequestProcessor<SCardRequest> {

diff --git a/...rocess/impl/SMembersRequestProcessor.java → ...ss/impl/set/SMembersRequestProcessor.java b/...rocess/impl/SMembersRequestProcessor.java → ...ss/impl/set/SMembersRequestProcessor.java
@@ -14,7 +14,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-package icu.funkye.redispike.handler.process.impl;
+package icu.funkye.redispike.handler.process.impl.set;
 
 import com.aerospike.client.AerospikeException;
 import com.aerospike.client.Key;
@@ -25,7 +25,7 @@
 import icu.funkye.redispike.factory.AeroSpikeClientFactory;
 import icu.funkye.redispike.handler.process.AbstractRedisRequestProcessor;
 import icu.funkye.redispike.protocol.RedisRequestCommandCode;
-import icu.funkye.redispike.protocol.request.SMembersRequest;
+import icu.funkye.redispike.protocol.request.set.SMembersRequest;
 import icu.funkye.redispike.util.IntegerUtils;
 
 public class SMembersRequestProcessor extends AbstractRedisRequestProcessor<SMembersRequest> {