summaryrefslogtreecommitdiffstats
path: root/src/dma/pkg/common/redispool.go
blob: 3c19a195ed51c13e580d324786afbd9854d8d8a5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
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
}