summaryrefslogtreecommitdiffstats
path: root/src/dma/pkg/common/redispool.go
diff options
context:
space:
mode:
Diffstat (limited to 'src/dma/pkg/common/redispool.go')
-rw-r--r--src/dma/pkg/common/redispool.go72
1 files changed, 72 insertions, 0 deletions
diff --git a/src/dma/pkg/common/redispool.go b/src/dma/pkg/common/redispool.go
new file mode 100644
index 00000000..3c19a195
--- /dev/null
+++ b/src/dma/pkg/common/redispool.go
@@ -0,0 +1,72 @@
+/*
+ * Copyright 2018 NEC Corporation
+ *
+ * Licensed 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 common
+
+import (
+ "github.com/go-redis/redis"
+ "log"
+)
+
+// RedisPool is an implementation of Pool by redis.
+type RedisPool struct {
+ Client *redis.Client
+}
+
+// Set is to set data in redis.
+func (thisPool RedisPool) Set(key string, data string) error {
+ key = redisLabel + "/" + key
+ err := thisPool.Client.Set(key, data, 0).Err()
+ if err != nil {
+ log.Printf("redis Set error: %s", err)
+ }
+ return err
+}
+
+// Get is to get data in redis.
+func (thisPool RedisPool) Get(key string) (string, error) {
+ key = redisLabel + "/" + key
+ value, err := thisPool.Client.Get(key).Result()
+ if err != nil {
+ log.Printf("redis Get error: %s", err)
+ }
+ return value, err
+}
+
+// Del is to delete data in redis.
+func (thisPool RedisPool) Del(key string) error {
+ key = redisLabel + "/" + key
+ err := thisPool.Client.Del(key).Err()
+ if err != nil {
+ log.Printf("redis Del error: %s", err)
+ }
+ return err
+}
+
+// DelAll is to delete all data, begins with <redisLabel>, in redis.
+func (thisPool RedisPool) DelAll() error {
+ pattern := redisLabel + "/*"
+
+ keys, err := thisPool.Client.Keys(pattern).Result()
+ if err != nil {
+ log.Printf("redis Keys error :%s", err)
+ }
+
+ for _, v := range keys {
+ thisPool.Client.Del(v)
+ }
+ return err
+}