re
/
redis
forked from mirror/redis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

multi: Restrict Discard usage.

This commit is contained in:
Vladimir Mihailenco 2012-08-25 22:51:42 +03:00
parent dbcfb0984e
commit 037888ee0f
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
multi.go
View File

@ -42,13 +42,20 @@ func (c *MultiClient) Unwatch(keys ...string) *StatusReq {
func (c *MultiClient) Discard() { func (c *MultiClient) Discard() {
c.reqsMtx.Lock() c.reqsMtx.Lock()
c.reqs = []Req{NewStatusReq("MULTI")} if c.reqs == nil {
panic("Discard can be used only inside Exec")
}
c.reqs = c.reqs[:1]
c.reqsMtx.Unlock() c.reqsMtx.Unlock()
} }
func (c *MultiClient) Exec(do func()) ([]Req, error) { func (c *MultiClient) Exec(do func()) ([]Req, error) {
c.Discard() c.reqsMtx.Lock()
c.reqs = []Req{NewStatusReq("MULTI")}
c.reqsMtx.Unlock()
do() do()
c.Queue(NewIfaceSliceReq("EXEC")) c.Queue(NewIfaceSliceReq("EXEC"))
c.reqsMtx.Lock() c.reqsMtx.Lock()