This repository has been archived by the owner on Dec 29, 2018. It is now read-only.
/
sets.go
108 lines (84 loc) · 2.04 KB
/
sets.go
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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
package redis
import (
"fmt"
"strconv"
"github.com/go-redis/redis"
)
type StringsSet struct {
db *Database
key string
}
func (set *StringsSet) Len() (int64, error) {
return set.db.sess.SCard(set.key).Result()
}
func (set *StringsSet) Members() ([]string, error) {
result, err := set.db.sess.SMembers(set.key).Result()
if err != nil {
return nil, err
}
return result, nil
}
func (set *StringsSet) Add(values ...string) error {
members := make([]interface{}, len(values))
for i := range values {
members[i] = values[i]
}
return set.db.sess.SAdd(set.key, members...).Err()
}
func (set *StringsSet) Remove(values ...string) error {
members := make([]interface{}, len(values))
for i := range values {
members[i] = values[i]
}
return set.db.sess.SRem(set.key, members...).Err()
}
func (set *StringsSet) SortAlpha() ([]string, error) {
result, err := set.sort(&redis.Sort{Alpha: true})
if err != nil {
return nil, err
}
return result, nil
}
func (set *StringsSet) sort(sort *redis.Sort) ([]string, error) {
result, err := set.db.sess.Sort(set.key, sort).Result()
if err != nil {
return nil, err
}
return result, nil
}
type Int64Set struct {
db *Database
key string
}
func (set *Int64Set) Len() (int64, error) {
return set.db.sess.SCard(set.key).Result()
}
func (set *Int64Set) Members() ([]int64, error) {
rawResult, err := set.db.sess.SMembers(set.key).Result()
if err != nil {
return nil, err
}
result := []int64{}
for _, r := range rawResult {
n, err := strconv.ParseInt(r, 10, 64)
if err != nil {
return nil, err
}
result = append(result, n)
}
return result, nil
}
func (set *Int64Set) Add(values ...int64) error {
members := make([]interface{}, len(values))
for i := range values {
members[i] = fmt.Sprintf("%d", values[i])
}
return set.db.sess.SAdd(set.key, members...).Err()
}
func (set *Int64Set) Remove(values ...int64) error {
members := make([]interface{}, len(values))
for i := range values {
members[i] = fmt.Sprintf("%d", values[i])
}
return set.db.sess.SRem(set.key, members...).Err()
}