1058

Committed 17 Sep 2020 08:04PM UTC coverage: 67.44%. First build

Build # 1058

Build Type

Pull #134

travis-ci

Committed by

web-flow

Commit Message

<a href="https://github.com/topfreegames/pitaya/commit/<a class=hub.com/topfreegames/pitaya/commit/<a class="double-link" href="https://git"><a class=hub.com/topfreegames/pitaya/commit/<a class="double-link" href="https://git"><a class=hub.com/topfreegames/pitaya/commit/e248e63d2474f24e92e882a6f798a14c0ccffd37">e248e63d2">&lt;a href=&quot;https://github.com/topfreegames/pitaya/commit/</a><a class="double-link" href="https://github.com/topfreegames/pitaya/commit/&lt;a class=&quot;double-link&quot; href=&quot;https://git">&lt;a class=</a>hub.com/topfreegames/pitaya/commit/&lt;a class=&quot;double-link&quot; href=&quot;https://git">&lt;a class=</a>hub.com/topfreegames/pitaya/commit/&lt;a class=&quot;double-link&quot; href=&quot;https://git">&lt;a class=</a>hub.com/topfreegames/pitaya/commit/&lt;a class=&quot;double-link&quot; href=&quot;https://git">&lt;a class=</a>hub.com/topfreegames/pitaya/commit/&lt;a class=&quot;double-link&quot; href=&quot;https://git">&lt;a class=</a>hub.com/topfreegames/pitaya/commit/e248e63d2474f24e92e882a6f798a14c0ccffd37">e248e63d2</a><a href="https://github.com/topfreegames/pitaya/commit/e248e63d2474f24e92e882a6f798a14c0ccffd37">&quot;&gt;&amp;lt;a href=&amp;quot;https://github.com/topfreegames/pitaya/commit/&lt;/a&gt;&lt;a class=&quot;double-link&quot; href=&quot;https://github.com/topfreegames/pitaya/commit/&amp;lt;a class=&amp;quot;double-link&amp;quot; href=&amp;quot;https://git&quot;&gt;&amp;lt;a class=&lt;/a&gt;hub.com/topfreegames/pitaya/commit/&amp;lt;a class=&amp;quot;double-link&amp;quot; href=&amp;quot;https://git&quot;&gt;&amp;lt;a class=&lt;/a&gt;hub.com/topfreegames/pitaya/commit/&amp;lt;a class=&amp;quot;double-link&amp;quot; href=&amp;quot;https://git&quot;&gt;&amp;lt;a class=&lt;/a&gt;hub.com/topfreegames/pitaya/commit/&amp;lt;a class=&amp;quot;double-link&amp;quot; href=&amp;quot;https://git&quot;&gt;&amp;lt;a class=&lt;/a&gt;hub.com/topfreegames/pitaya/commit/&amp;lt;a class=&amp;quot;double-link&amp;quot; href=&amp;quot;https://git&quot;&gt;&amp;lt;a class=&lt;/a&gt;hub.com/topfreegames/pitaya/commit/e248e63d2474f24e92e882a6f798a14c0ccffd37&quot;&gt;e248e63d2&lt;/a&gt;&lt;a href=&quot;https://github.com/topfreegames/pitaya/commi... (continued)

Pull Request Pull Request #134: Refactor/hardcoded configs

Run Details

624 of 624 new or added lines in 18 files covered. (100.0%)

4797 of 7113 relevant lines covered (67.44%)

0.75 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

73.58

/groups/etcd_group_service.go

package groups

import (
        "context"
        "fmt"
        "sync"
        "time"

        "github.com/coreos/etcd/clientv3"
        "github.com/coreos/etcd/clientv3/namespace"
        "github.com/coreos/etcd/mvcc/mvccpb"
        "github.com/topfreegames/pitaya/v2/config"
        "github.com/topfreegames/pitaya/v2/constants"
        "github.com/topfreegames/pitaya/v2/logger"
)

var (
        clientInstance     *clientv3.Client
        transactionTimeout time.Duration
        etcdOnce           sync.Once
)

// EtcdGroupService base ETCD struct solution
type EtcdGroupService struct {
}

// NewEtcdGroupService returns a new group instance
func NewEtcdGroupService(conf config.EtcdGroupServiceConfig, clientOrNil *clientv3.Client) (*EtcdGroupService, error) {
        err := initClientInstance(conf, clientOrNil)
        if err != nil {
                return nil, err
        }
        return &EtcdGroupService{}, err
}

func initClientInstance(config config.EtcdGroupServiceConfig, clientOrNil *clientv3.Client) error {
        var err error
        etcdOnce.Do(func() {
                if clientOrNil != nil {
                        clientInstance = clientOrNil
                } else {
                        clientInstance, err = createBaseClient(config)
                }
                if err != nil {
                        logger.Log.Fatalf("error initializing singleton etcd client in groups: %s", err.Error())
                        return
                }
                transactionTimeout = config.TransactionTimeout
        })
        return err
}

func createBaseClient(config config.EtcdGroupServiceConfig) (*clientv3.Client, error) {
        cli, err := clientv3.New(clientv3.Config{
                Endpoints:   config.Endpoints,
                DialTimeout: config.DialTimeout,
        })
        if err != nil {
                return nil, err
        }
        cli.KV = namespace.NewKV(cli.KV, config.Prefix)
        return cli, nil
}

func groupKey(groupName string) string {
        return fmt.Sprintf("groups/%s", groupName)
}

func memberKey(groupName, uid string) string {
        return fmt.Sprintf("%s/uids/%s", groupKey(groupName), uid)
}

func getGroupKV(ctx context.Context, groupName string) (*mvccpb.KeyValue, error) {
        ctxT, cancel := context.WithTimeout(ctx, transactionTimeout)
        defer cancel()
        etcdRes, err := clientInstance.Get(ctxT, groupKey(groupName))
        if err != nil {
                return nil, err
        }
        if etcdRes.Count == 0 {
                return nil, constants.ErrGroupNotFound
        }
        return etcdRes.Kvs[0], nil
}

func (c *EtcdGroupService) createGroup(ctx context.Context, groupName string, leaseID clientv3.LeaseID) error {
        var etcdRes *clientv3.TxnResponse
        var err error

        ctxT, cancel := context.WithTimeout(ctx, transactionTimeout)
        defer cancel()
        if leaseID != 0 {
                etcdRes, err = clientInstance.Txn(ctxT).
                        If(clientv3.Compare(clientv3.CreateRevision(groupKey(groupName)), "=", 0)).
                        Then(clientv3.OpPut(groupKey(groupName), "", clientv3.WithLease(leaseID))).
                        Commit()
        } else {
                etcdRes, err = clientInstance.Txn(ctxT).
                        If(clientv3.Compare(clientv3.CreateRevision(groupKey(groupName)), "=", 0)).
                        Then(clientv3.OpPut(groupKey(groupName), "")).
                        Commit()
        }

        if err != nil {
                return err
        }
        if !etcdRes.Succeeded {
                return constants.ErrGroupAlreadyExists
        }
        return nil
}

// GroupCreate creates a group struct inside ETCD, without TTL
func (c *EtcdGroupService) GroupCreate(ctx context.Context, groupName string) error {
        return c.createGroup(ctx, groupName, 0)
}

// GroupCreateWithTTL creates a group struct inside ETCD, with TTL, using leaseID
func (c *EtcdGroupService) GroupCreateWithTTL(ctx context.Context, groupName string, ttlTime time.Duration) error {
        ctxT, cancel := context.WithTimeout(ctx, transactionTimeout)
        defer cancel()
        lease, err := clientInstance.Grant(ctxT, int64(ttlTime.Seconds()))
        if err != nil {
                return err
        }
        return c.createGroup(ctx, groupName, lease.ID)
}

// GroupMembers returns all member's UIDs
func (c *EtcdGroupService) GroupMembers(ctx context.Context, groupName string) ([]string, error) {
        prefix := memberKey(groupName, "")
        ctxT, cancel := context.WithTimeout(ctx, transactionTimeout)
        defer cancel()
        etcdRes, err := clientInstance.Txn(ctxT).
                If(clientv3.Compare(clientv3.CreateRevision(groupKey(groupName)), ">", 0)).
                Then(clientv3.OpGet(prefix, clientv3.WithPrefix(), clientv3.WithKeysOnly())).
                Commit()

        if err != nil {
                return nil, err
        }
        if !etcdRes.Succeeded {
                return nil, constants.ErrGroupNotFound
        }

        getRes := etcdRes.Responses[0].GetResponseRange()
        members := make([]string, getRes.GetCount())
        for i, kv := range getRes.GetKvs() {
                members[i] = string(kv.Key)[len(prefix):]
        }
        return members, nil
}

// GroupContainsMember checks whether a UID is contained in current group or not
func (c *EtcdGroupService) GroupContainsMember(ctx context.Context, groupName, uid string) (bool, error) {
        ctxT, cancel := context.WithTimeout(ctx, transactionTimeout)
        defer cancel()
        etcdRes, err := clientInstance.Txn(ctxT).
                If(clientv3.Compare(clientv3.CreateRevision(groupKey(groupName)), ">", 0)).
                Then(clientv3.OpGet(memberKey(groupName, uid), clientv3.WithCountOnly())).
                Commit()

        if err != nil {
                return false, err
        }
        if !etcdRes.Succeeded {
                return false, constants.ErrGroupNotFound
        }
        return etcdRes.Responses[0].GetResponseRange().GetCount() > 0, nil
}

// GroupAddMember adds UID to group
func (c *EtcdGroupService) GroupAddMember(ctx context.Context, groupName, uid string) error {
        var etcdRes *clientv3.TxnResponse
        kv, err := getGroupKV(ctx, groupName)
        if err != nil {
                return err
        }

        ctxT, cancel := context.WithTimeout(ctx, transactionTimeout)
        defer cancel()
        if kv.Lease != 0 {
                etcdRes, err = clientInstance.Txn(ctxT).
                        If(clientv3.Compare(clientv3.CreateRevision(groupKey(groupName)), ">", 0),
                                clientv3.Compare(clientv3.CreateRevision(memberKey(groupName, uid)), "=", 0)).
                        Then(clientv3.OpPut(memberKey(groupName, uid), "", clientv3.WithLease(clientv3.LeaseID(kv.Lease)))).
                        Commit()
        } else {
                etcdRes, err = clientInstance.Txn(ctxT).
                        If(clientv3.Compare(clientv3.CreateRevision(groupKey(groupName)), ">", 0),
                                clientv3.Compare(clientv3.CreateRevision(memberKey(groupName, uid)), "=", 0)).
                        Then(clientv3.OpPut(memberKey(groupName, uid), "")).
                        Commit()
        }

        if err != nil {
                return err
        }
        if !etcdRes.Succeeded {
                return constants.ErrMemberAlreadyExists
        }
        return nil
}

// GroupRemoveMember removes specified UID from group
func (c *EtcdGroupService) GroupRemoveMember(ctx context.Context, groupName, uid string) error {
        ctxT, cancel := context.WithTimeout(ctx, transactionTimeout)
        defer cancel()
        etcdRes, err := clientInstance.Txn(ctxT).
                If(clientv3.Compare(clientv3.CreateRevision(memberKey(groupName, uid)), ">", 0)).
                Then(clientv3.OpDelete(memberKey(groupName, uid))).
                Commit()

        if err != nil {
                return err
        }
        if !etcdRes.Succeeded {
                return constants.ErrMemberNotFound
        }
        return nil
}

// GroupRemoveAll clears all UIDs in the group
func (c *EtcdGroupService) GroupRemoveAll(ctx context.Context, groupName string) error {
        ctxT, cancel := context.WithTimeout(ctx, transactionTimeout)
        defer cancel()
        etcdRes, err := clientInstance.Txn(ctxT).
                If(clientv3.Compare(clientv3.CreateRevision(groupKey(groupName)), ">", 0)).
                Then(clientv3.OpDelete(memberKey(groupName, ""), clientv3.WithPrefix())).
                Commit()

        if err != nil {
                return err
        }
        if !etcdRes.Succeeded {
                return constants.ErrGroupNotFound
        }
        return nil
}

// GroupDelete deletes the whole group, including members and base group
func (c *EtcdGroupService) GroupDelete(ctx context.Context, groupName string) error {
        ctxT, cancel := context.WithTimeout(ctx, transactionTimeout)
        defer cancel()
        etcdRes, err := clientInstance.Txn(ctxT).
                If(clientv3.Compare(clientv3.CreateRevision(groupKey(groupName)), ">", 0)).
                Then(clientv3.OpDelete(memberKey(groupName, ""), clientv3.WithPrefix()),
                        clientv3.OpDelete(groupKey(groupName))).
                Commit()

        if err != nil {
                return err
        }
        if !etcdRes.Succeeded {
                return constants.ErrGroupNotFound
        }
        return nil
}

// GroupCountMembers get current member amount in group
func (c *EtcdGroupService) GroupCountMembers(ctx context.Context, groupName string) (int, error) {
        ctxT, cancel := context.WithTimeout(ctx, transactionTimeout)
        defer cancel()
        etcdRes, err := clientInstance.Get(ctxT, memberKey(groupName, ""), clientv3.WithPrefix(), clientv3.WithCountOnly())
        if err != nil {
                return 0, err
        }
        return int(etcdRes.Count), nil
}

// GroupRenewTTL will renew ETCD lease TTL
func (c *EtcdGroupService) GroupRenewTTL(ctx context.Context, groupName string) error {
        kv, err := getGroupKV(ctx, groupName)
        if err != nil {
                return err
        }
        if kv.Lease != 0 {
                ctxT, cancel := context.WithTimeout(ctx, transactionTimeout)
                defer cancel()
                _, err = clientInstance.KeepAliveOnce(ctxT, clientv3.LeaseID(kv.Lease))
                return err
        }
        return constants.ErrEtcdLeaseNotFound
}

1	package groups
2
3	import (
4	"context"
5	"fmt"
6	"sync"
7	"time"
8
9	"github.com/coreos/etcd/clientv3"
10	"github.com/coreos/etcd/clientv3/namespace"
11	"github.com/coreos/etcd/mvcc/mvccpb"
12	"github.com/topfreegames/pitaya/v2/config"
13	"github.com/topfreegames/pitaya/v2/constants"
14	"github.com/topfreegames/pitaya/v2/logger"
15	)
16
17	var (
18	clientInstance *clientv3.Client
19	transactionTimeout time.Duration
20	etcdOnce sync.Once
21	)
22
23	// EtcdGroupService base ETCD struct solution
24	type EtcdGroupService struct {
25	}
26
27	// NewEtcdGroupService returns a new group instance
28	func NewEtcdGroupService(conf config.EtcdGroupServiceConfig, clientOrNil clientv3.Client) (EtcdGroupService, error) {	1✔
29	err := initClientInstance(conf, clientOrNil)	1✔
30	if err != nil {	1✔
31	return nil, err	×
32	}	×
33	return &EtcdGroupService{}, err	1✔
34	}
35
36	func initClientInstance(config config.EtcdGroupServiceConfig, clientOrNil *clientv3.Client) error {	1✔
37	var err error	1✔
38	etcdOnce.Do(func() {	2✔
39	if clientOrNil != nil {	2✔
40	clientInstance = clientOrNil	1✔
41	} else {	1✔
42	clientInstance, err = createBaseClient(config)	×
43	}	×
44	if err != nil {	1✔
45	logger.Log.Fatalf("error initializing singleton etcd client in groups: %s", err.Error())	×
46	return	×
47	}	×
48	transactionTimeout = config.TransactionTimeout	1✔
49	})
50	return err	1✔
51	}
52
53	func createBaseClient(config config.EtcdGroupServiceConfig) (*clientv3.Client, error) {	×
54	cli, err := clientv3.New(clientv3.Config{	×
55	Endpoints: config.Endpoints,	×
56	DialTimeout: config.DialTimeout,	×
57	})	×
58	if err != nil {	×
59	return nil, err	×
60	}	×
61	cli.KV = namespace.NewKV(cli.KV, config.Prefix)	×
62	return cli, nil	×
63	}
64
65	func groupKey(groupName string) string {	1✔
66	return fmt.Sprintf("groups/%s", groupName)	1✔
67	}	1✔
68
69	func memberKey(groupName, uid string) string {	1✔
70	return fmt.Sprintf("%s/uids/%s", groupKey(groupName), uid)	1✔
71	}	1✔
72
73	func getGroupKV(ctx context.Context, groupName string) (*mvccpb.KeyValue, error) {	1✔
74	ctxT, cancel := context.WithTimeout(ctx, transactionTimeout)	1✔
75	defer cancel()	1✔
76	etcdRes, err := clientInstance.Get(ctxT, groupKey(groupName))	1✔
77	if err != nil {	1✔
78	return nil, err	×
79	}	×
80	if etcdRes.Count == 0 {	1✔
81	return nil, constants.ErrGroupNotFound	×
82	}	×
83	return etcdRes.Kvs[0], nil	1✔
84	}
85
86	func (c *EtcdGroupService) createGroup(ctx context.Context, groupName string, leaseID clientv3.LeaseID) error {	1✔
87	var etcdRes *clientv3.TxnResponse	1✔
88	var err error	1✔
89		1✔
90	ctxT, cancel := context.WithTimeout(ctx, transactionTimeout)	1✔
91	defer cancel()	1✔
92	if leaseID != 0 {	2✔
93	etcdRes, err = clientInstance.Txn(ctxT).	1✔
94	If(clientv3.Compare(clientv3.CreateRevision(groupKey(groupName)), "=", 0)).	1✔
95	Then(clientv3.OpPut(groupKey(groupName), "", clientv3.WithLease(leaseID))).	1✔
96	Commit()	1✔
97	} else {	2✔
98	etcdRes, err = clientInstance.Txn(ctxT).	1✔
99	If(clientv3.Compare(clientv3.CreateRevision(groupKey(groupName)), "=", 0)).	1✔
100	Then(clientv3.OpPut(groupKey(groupName), "")).	1✔
101	Commit()	1✔
102	}	1✔
103
104	if err != nil {	1✔
105	return err	×
106	}	×
107	if !etcdRes.Succeeded {	2✔
108	return constants.ErrGroupAlreadyExists	1✔
109	}	1✔
110	return nil	1✔
111	}
112
113	// GroupCreate creates a group struct inside ETCD, without TTL
114	func (c *EtcdGroupService) GroupCreate(ctx context.Context, groupName string) error {	1✔
115	return c.createGroup(ctx, groupName, 0)	1✔
116	}	1✔
117
118	// GroupCreateWithTTL creates a group struct inside ETCD, with TTL, using leaseID
119	func (c *EtcdGroupService) GroupCreateWithTTL(ctx context.Context, groupName string, ttlTime time.Duration) error {	1✔
120	ctxT, cancel := context.WithTimeout(ctx, transactionTimeout)	1✔
121	defer cancel()	1✔
122	lease, err := clientInstance.Grant(ctxT, int64(ttlTime.Seconds()))	1✔
123	if err != nil {	1✔
124	return err	×
125	}	×
126	return c.createGroup(ctx, groupName, lease.ID)	1✔
127	}
128
129	// GroupMembers returns all member's UIDs
130	func (c *EtcdGroupService) GroupMembers(ctx context.Context, groupName string) ([]string, error) {	1✔
131	prefix := memberKey(groupName, "")	1✔
132	ctxT, cancel := context.WithTimeout(ctx, transactionTimeout)	1✔
133	defer cancel()	1✔
134	etcdRes, err := clientInstance.Txn(ctxT).	1✔
135	If(clientv3.Compare(clientv3.CreateRevision(groupKey(groupName)), ">", 0)).	1✔
136	Then(clientv3.OpGet(prefix, clientv3.WithPrefix(), clientv3.WithKeysOnly())).	1✔
137	Commit()	1✔
138		1✔
139	if err != nil {	1✔
140	return nil, err	×
141	}	×
142	if !etcdRes.Succeeded {	1✔
143	return nil, constants.ErrGroupNotFound	×
144	}	×
145
146	getRes := etcdRes.Responses[0].GetResponseRange()	1✔
147	members := make([]string, getRes.GetCount())	1✔
148	for i, kv := range getRes.GetKvs() {	2✔
149	members[i] = string(kv.Key)[len(prefix):]	1✔
150	}	1✔
151	return members, nil	1✔
152	}
153
154	// GroupContainsMember checks whether a UID is contained in current group or not
155	func (c *EtcdGroupService) GroupContainsMember(ctx context.Context, groupName, uid string) (bool, error) {	1✔
156	ctxT, cancel := context.WithTimeout(ctx, transactionTimeout)	1✔
157	defer cancel()	1✔
158	etcdRes, err := clientInstance.Txn(ctxT).	1✔
159	If(clientv3.Compare(clientv3.CreateRevision(groupKey(groupName)), ">", 0)).	1✔
160	Then(clientv3.OpGet(memberKey(groupName, uid), clientv3.WithCountOnly())).	1✔
161	Commit()	1✔
162		1✔
163	if err != nil {	1✔
164	return false, err	×
165	}	×
166	if !etcdRes.Succeeded {	2✔
167	return false, constants.ErrGroupNotFound	1✔
168	}	1✔
169	return etcdRes.Responses[0].GetResponseRange().GetCount() > 0, nil	1✔
170	}
171
172	// GroupAddMember adds UID to group
173	func (c *EtcdGroupService) GroupAddMember(ctx context.Context, groupName, uid string) error {	1✔
174	var etcdRes *clientv3.TxnResponse	1✔
175	kv, err := getGroupKV(ctx, groupName)	1✔
176	if err != nil {	1✔
177	return err	×
178	}	×
179
180	ctxT, cancel := context.WithTimeout(ctx, transactionTimeout)	1✔
181	defer cancel()	1✔
182	if kv.Lease != 0 {	1✔
183	etcdRes, err = clientInstance.Txn(ctxT).	×
184	If(clientv3.Compare(clientv3.CreateRevision(groupKey(groupName)), ">", 0),	×
185	clientv3.Compare(clientv3.CreateRevision(memberKey(groupName, uid)), "=", 0)).	×
186	Then(clientv3.OpPut(memberKey(groupName, uid), "", clientv3.WithLease(clientv3.LeaseID(kv.Lease)))).	×
187	Commit()	×
188	} else {	1✔
189	etcdRes, err = clientInstance.Txn(ctxT).	1✔
190	If(clientv3.Compare(clientv3.CreateRevision(groupKey(groupName)), ">", 0),	1✔
191	clientv3.Compare(clientv3.CreateRevision(memberKey(groupName, uid)), "=", 0)).	1✔
192	Then(clientv3.OpPut(memberKey(groupName, uid), "")).	1✔
193	Commit()	1✔
194	}	1✔
195
196	if err != nil {	1✔
197	return err	×
198	}	×
199	if !etcdRes.Succeeded {	2✔
200	return constants.ErrMemberAlreadyExists	1✔
201	}	1✔
202	return nil	1✔
203	}
204
205	// GroupRemoveMember removes specified UID from group
206	func (c *EtcdGroupService) GroupRemoveMember(ctx context.Context, groupName, uid string) error {	1✔
207	ctxT, cancel := context.WithTimeout(ctx, transactionTimeout)	1✔
208	defer cancel()	1✔
209	etcdRes, err := clientInstance.Txn(ctxT).	1✔
210	If(clientv3.Compare(clientv3.CreateRevision(memberKey(groupName, uid)), ">", 0)).	1✔
211	Then(clientv3.OpDelete(memberKey(groupName, uid))).	1✔
212	Commit()	1✔
213		1✔
214	if err != nil {	1✔
215	return err	×
216	}	×
217	if !etcdRes.Succeeded {	1✔
218	return constants.ErrMemberNotFound	×
219	}	×
220	return nil	1✔
221	}
222
223	// GroupRemoveAll clears all UIDs in the group
224	func (c *EtcdGroupService) GroupRemoveAll(ctx context.Context, groupName string) error {	1✔
225	ctxT, cancel := context.WithTimeout(ctx, transactionTimeout)	1✔
226	defer cancel()	1✔
227	etcdRes, err := clientInstance.Txn(ctxT).	1✔
228	If(clientv3.Compare(clientv3.CreateRevision(groupKey(groupName)), ">", 0)).	1✔
229	Then(clientv3.OpDelete(memberKey(groupName, ""), clientv3.WithPrefix())).	1✔
230	Commit()	1✔
231		1✔
232	if err != nil {	1✔
233	return err	×
234	}	×
235	if !etcdRes.Succeeded {	1✔
236	return constants.ErrGroupNotFound	×
237	}	×
238	return nil	1✔
239	}
240
241	// GroupDelete deletes the whole group, including members and base group
242	func (c *EtcdGroupService) GroupDelete(ctx context.Context, groupName string) error {	1✔
243	ctxT, cancel := context.WithTimeout(ctx, transactionTimeout)	1✔
244	defer cancel()	1✔
245	etcdRes, err := clientInstance.Txn(ctxT).	1✔
246	If(clientv3.Compare(clientv3.CreateRevision(groupKey(groupName)), ">", 0)).	1✔
247	Then(clientv3.OpDelete(memberKey(groupName, ""), clientv3.WithPrefix()),	1✔
248	clientv3.OpDelete(groupKey(groupName))).	1✔
249	Commit()	1✔
250		1✔
251	if err != nil {	1✔
252	return err	×
253	}	×
254	if !etcdRes.Succeeded {	1✔
255	return constants.ErrGroupNotFound	×
256	}	×
257	return nil	1✔
258	}
259
260	// GroupCountMembers get current member amount in group
261	func (c *EtcdGroupService) GroupCountMembers(ctx context.Context, groupName string) (int, error) {	1✔
262	ctxT, cancel := context.WithTimeout(ctx, transactionTimeout)	1✔
263	defer cancel()	1✔
264	etcdRes, err := clientInstance.Get(ctxT, memberKey(groupName, ""), clientv3.WithPrefix(), clientv3.WithCountOnly())	1✔
265	if err != nil {	1✔
266	return 0, err	×
267	}	×
268	return int(etcdRes.Count), nil	1✔
269	}
270
271	// GroupRenewTTL will renew ETCD lease TTL
272	func (c *EtcdGroupService) GroupRenewTTL(ctx context.Context, groupName string) error {	1✔
273	kv, err := getGroupKV(ctx, groupName)	1✔
274	if err != nil {	1✔
275	return err	×
276	}	×
277	if kv.Lease != 0 {	2✔
278	ctxT, cancel := context.WithTimeout(ctx, transactionTimeout)	1✔
279	defer cancel()	1✔
280	_, err = clientInstance.KeepAliveOnce(ctxT, clientv3.LeaseID(kv.Lease))	1✔
281	return err	1✔
282	}	1✔
283	return constants.ErrEtcdLeaseNotFound	1✔
284	}

topfreegames / pitaya / 1058

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous