You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
86 lines
2.9 KiB
86 lines
2.9 KiB
// Copyright 2012 Gary Burd |
|
// |
|
// 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 redis |
|
|
|
import ( |
|
"crypto/sha1" |
|
"encoding/hex" |
|
"io" |
|
"strings" |
|
) |
|
|
|
// Script encapsulates the source, hash and key count for a Lua script. See |
|
// http://redis.io/commands/eval for information on scripts in Redis. |
|
type Script struct { |
|
keyCount int |
|
src string |
|
hash string |
|
} |
|
|
|
// NewScript returns a new script object. If keyCount is greater than or equal |
|
// to zero, then the count is automatically inserted in the EVAL command |
|
// argument list. If keyCount is less than zero, then the application supplies |
|
// the count as the first value in the keysAndArgs argument to the Do, Send and |
|
// SendHash methods. |
|
func NewScript(keyCount int, src string) *Script { |
|
h := sha1.New() |
|
io.WriteString(h, src) |
|
return &Script{keyCount, src, hex.EncodeToString(h.Sum(nil))} |
|
} |
|
|
|
func (s *Script) args(spec string, keysAndArgs []interface{}) []interface{} { |
|
var args []interface{} |
|
if s.keyCount < 0 { |
|
args = make([]interface{}, 1+len(keysAndArgs)) |
|
args[0] = spec |
|
copy(args[1:], keysAndArgs) |
|
} else { |
|
args = make([]interface{}, 2+len(keysAndArgs)) |
|
args[0] = spec |
|
args[1] = s.keyCount |
|
copy(args[2:], keysAndArgs) |
|
} |
|
return args |
|
} |
|
|
|
// Do evaluates the script. Under the covers, Do optimistically evaluates the |
|
// script using the EVALSHA command. If the command fails because the script is |
|
// not loaded, then Do evaluates the script using the EVAL command (thus |
|
// causing the script to load). |
|
func (s *Script) Do(c Conn, keysAndArgs ...interface{}) (interface{}, error) { |
|
v, err := c.Do("EVALSHA", s.args(s.hash, keysAndArgs)...) |
|
if e, ok := err.(Error); ok && strings.HasPrefix(string(e), "NOSCRIPT ") { |
|
v, err = c.Do("EVAL", s.args(s.src, keysAndArgs)...) |
|
} |
|
return v, err |
|
} |
|
|
|
// SendHash evaluates the script without waiting for the reply. The script is |
|
// evaluated with the EVALSHA command. The application must ensure that the |
|
// script is loaded by a previous call to Send, Do or Load methods. |
|
func (s *Script) SendHash(c Conn, keysAndArgs ...interface{}) error { |
|
return c.Send("EVALSHA", s.args(s.hash, keysAndArgs)...) |
|
} |
|
|
|
// Send evaluates the script without waiting for the reply. |
|
func (s *Script) Send(c Conn, keysAndArgs ...interface{}) error { |
|
return c.Send("EVAL", s.args(s.src, keysAndArgs)...) |
|
} |
|
|
|
// Load loads the script without evaluating it. |
|
func (s *Script) Load(c Conn) error { |
|
_, err := c.Do("SCRIPT", "LOAD", s.src) |
|
return err |
|
}
|
|
|