Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

BLPOP and RPUSH command #12

Open
wants to merge 8 commits into
base: master
Choose a base branch
from
Open

BLPOP and RPUSH command #12

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
0ff4e72
BLPOP and RPUSH command
blindlf Jan 4, 2017
99b4cec
Change argument name
blindlf Jan 4, 2017
f99f4ab
Output raw value
blindlf Jan 5, 2017
e884128
Echo newline after BLPOP key
blindlf Jan 5, 2017
656d2c5
Raw input send to redis
blindlf Jan 7, 2017
dfe06dd
Command hget, hset, del, hincrby
blindlf Mar 6, 2017
1588a3b
Bugfix hset character in UTF8
blindlf Mar 6, 2017
84267ae
default host
blindlf Apr 9, 2018
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
27 changes: 24 additions & 3 deletions README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,15 +7,18 @@ Redi.sh is a primitive Redis client, written entirely in Bash. It allows you to
> By default redi.sh reads input from stdin and interprets it as a variable or array (if -a is used). To avoid setting redis hostname and port number with each command, you can export REDIS_HOST and REDIS_PORT variables.

```
./redi.sh [-a] [-g <variable|array>] [-p <password>] [-H <hostname>] [-P <port>]
./redi.sh [-a] [-w] [-g <variable|array>] [-p <password>] [-H <hostname>] [-P <port>]

-a : Tells the script that we are working with arrays, instead of regular variables.
-w : Input as raw instead of read line by line, don't replace "\n" as 0x0a.
-r <min,max> : When used with -a, defines the range of elements to get from the array. Default is all (0,-1).
-g <name> : Get the variable/array specified by <name> and output it to stdout.
-s <name> : Set the variable/array specified by <name> with the input from stdin.
-s <name> : Set the variable/array specified by <name> with the input from stdin.
-S <name> : RPUSH command. This also set -a -s.
-G <name> : BLPOP command, the timeout is 0. This also set -g, overrides -a, they are mutually exclusive.
-p <password> : Use "AUTH <password>" before running the SET/GET command to authenticate to redis.
-H <hostname> : Specify a custom hostname to connect to. Default is localhost.
-d <number> : Specify a custom database number from range 0-15\. Default is 0
-d <number> : Specify a custom database number from range 0-15\. Default is 0
-P <port> : Specify a custom port to connect to. Default is 6379.
```

Expand All @@ -35,6 +38,24 @@ green
blue
```

```shell
$ echo yellow | ./redi.sh -as queue
$ ./redi.sh -ag queue
yellow
$ echo red green blue | ./redi.sh -S queue
$ ./redi.sh -ag queue
yellow
red
green
blue
```

```shell
$ ./redi.sh -G queue
queue
yellow
```

## License

MIT
147 changes: 125 additions & 22 deletions redi.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ function redis_read_bulk() {
exit 1
fi

echo $(dd bs=1 count=$BYTE_COUNT status=noxfer <&$FILE_DESC 2>/dev/null)
dd bs=1 count=$BYTE_COUNT status=noxfer of=/dev/stdout <&$FILE_DESC 2>/dev/null
dd bs=1 count=2 status=noxfer <&$FILE_DESC 1>/dev/null 2>&1 # we are removing the extra character \r
}

Expand Down Expand Up @@ -73,6 +73,7 @@ do
if [[ ! -z $PARAM_COUNT ]]; then
if [[ $PARAM_CUR -lt $PARAM_COUNT ]]; then
((PARAM_CUR+=1))
echo
continue
else
break
Expand All @@ -97,15 +98,57 @@ function redis_select_db() {


function redis_get_var() {
typeset REDIS_VAR="$@"
printf %b "*2\r\n\$3\r\nGET\r\n\$${#REDIS_VAR}\r\n$REDIS_VAR\r\n"
if [ -z $REDIS_HASH ]; then
typeset REDIS_VAR="$@"
printf %b "*2\r\n\$3\r\nGET\r\n\$${#REDIS_VAR}\r\n$REDIS_VAR\r\n"
else
typeset REDIS_VAR="$1"
typeset REDIS_FIELD=$REDIS_HASH
printf %b "*3\r\n\$4\r\nHGET\r\n\$${#REDIS_VAR}\r\n$REDIS_VAR\r\n\$${#REDIS_FIELD}\r\n${REDIS_FIELD}\r\n"
fi
}

function redis_blpop_var() {
((number=$#+1))
protocol="*$number\r\n\$5\r\nBLPOP\r\n"
for i in "$@"; do
protocol="$protocol\$${#i}\r\n$i\r\n"
done
printf %b $protocol
}

function redis_set_var() {
typeset REDIS_VAR="$1"
shift
typeset REDIS_VAR_VAL="$@"
printf %b "*3\r\n\$3\r\nSET\r\n\$${#REDIS_VAR}\r\n$REDIS_VAR\r\n\$${#REDIS_VAR_VAL}\r\n$REDIS_VAR_VAL\r\n"
if [ -z $INPUT_RAW ]; then
printf %b "*3\r\n\$3\r\nSET\r\n\$${#REDIS_VAR}\r\n$REDIS_VAR\r\n\$${#REDIS_VAR_VAL}\r\n$REDIS_VAR_VAL\r\n"
else
printf %b "*3\r\n\$3\r\nSET\r\n\$${#REDIS_VAR}\r\n$REDIS_VAR\r\n\$${#REDIS_VAR_VAL}\r\n"
echo -n "$REDIS_VAR_VAL"
printf %b "\r\n"
fi
}

function redis_hset_var() {
typeset REDIS_VAR="$1"
typeset REDIS_FIELD="$2"
typeset REDIS_VALUE="$3"
typeset BYTES=`echo -n ${REDIS_VALUE} | wc -c`
printf %b "*4\r\n\$4\r\nHSET\r\n\$${#REDIS_VAR}\r\n$REDIS_VAR\r\n\$${#REDIS_FIELD}\r\n${REDIS_FIELD}\r\n\$${BYTES}\r\n${REDIS_VALUE}\r\n"
}

function redis_del_var() {
typeset REDIS_VAR="$1"
printf %b "*2\r\n\$3\r\nDEL\r\n\$${#REDIS_VAR}\r\n$REDIS_VAR\r\n"
}

function redis_hincrby() {
typeset REDIS_VAR="$1"
typeset REDIS_FIELD="$2"
typeset REDIS_VALUE="$3"
typeset BYTES=`echo -n ${REDIS_VALUE} | wc -c`
printf %b "*4\r\n\$7\r\nHINCRBY\r\n\$${#REDIS_VAR}\r\n$REDIS_VAR\r\n\$${#REDIS_FIELD}\r\n${REDIS_FIELD}\r\n\$${BYTES}\r\n${REDIS_VALUE}\r\n"
}

function redis_get_array() {
Expand All @@ -118,15 +161,24 @@ function redis_get_array() {
function redis_set_array() {
typeset REDIS_ARRAY="$1"
typeset -a REDIS_ARRAY_VAL=("${!2}")
typeset REDIS_RAW_VAL="$2"

printf %b "*2\r\n\$3\r\nDEL\r\n\$${#REDIS_ARRAY}\r\n$REDIS_ARRAY\r\n"
for i in "${REDIS_ARRAY_VAL[@]}"
do
printf %b "*3\r\n\$5\r\nRPUSH\r\n\$${#REDIS_ARRAY}\r\n$REDIS_ARRAY\r\n\$${#i}\r\n$i\r\n"
done
if [ -z $REDIS_PUSH ]; then
printf %b "*2\r\n\$3\r\nDEL\r\n\$${#REDIS_ARRAY}\r\n$REDIS_ARRAY\r\n"
fi
if [ -z $INPUT_RAW ]; then
for i in "${REDIS_ARRAY_VAL[@]}"
do
printf %b "*3\r\n\$5\r\nRPUSH\r\n\$${#REDIS_ARRAY}\r\n$REDIS_ARRAY\r\n\$${#i}\r\n$i\r\n"
done
else
printf %b "*3\r\n\$5\r\nRPUSH\r\n\$${#REDIS_ARRAY}\r\n$REDIS_ARRAY\r\n\$${#REDIS_RAW_VAL}\r\n"
echo -n "$REDIS_RAW_VAL"
printf %b "\r\n"
fi
}

while getopts g:s:r:P:H:p:d:ha opt; do
while getopts g:s:r:P:H:p:d:G:S:D:I:f:haw opt; do
case $opt in
p)
REDIS_PW=${OPTARG}
Expand All @@ -143,27 +195,51 @@ while getopts g:s:r:P:H:p:d:ha opt; do
a)
REDIS_ARRAY=1
;;
w)
INPUT_RAW=1
;;
r)
REDIS_ARRAY_RANGE=${OPTARG}
;;
s)
REDIS_SET=${OPTARG}
;;
G)
REDIS_ARRAY=0
REDIS_POP=1
REDIS_GET=${OPTARG}
;;
S)
REDIS_ARRAY=1
REDIS_PUSH=1
REDIS_SET=${OPTARG}
;;
D)
REDIS_DEL=${OPTARG}
;;
I)
REDIS_HINCRBY=$2
REDIS_HINCRBY_FIELD=$3
REDIS_HINCRBY_VALUE=$4
;;
f)
REDIS_HASH=${OPTARG}
;;
d)
REDIS_DB=${OPTARG}
;;
h)
echo
echo USAGE:
echo " $0 [-a] [-r <range>] [-s <var>] [-g <var>] [-p <password>] [-d <database_number>] [-H <hostname>] [-P <port>]"
echo " $0 [-a] [-w] [-r <range>] [-s <var>] [-g <var>] [-S <var>] [-G <var>] [-p <password>] [-d <database_number>] [-H <hostname>] [-P <port>] [-D <key>] [-I <key> <field> <value>]"
echo
exit 1
;;
esac
done

if [[ -z $REDIS_GET ]] && [[ -z $REDIS_SET ]]; then
echo "You must either GET(-g) or SET(-s)" >&2
if [[ -z $REDIS_HINCRBY ]] && [[ -z $REDIS_DEL ]] && [[ -z $REDIS_GET ]] && [[ -z $REDIS_SET ]]; then
echo "You must either DEL(-D) or HINCRBY(-I) or GET(-g) or SET(-s) or BLPOP(-G) or RPUSH(-S)" >&2
exit 1
fi

Expand All @@ -187,6 +263,10 @@ if [[ ! -z $REDIS_GET ]]; then
echo $i
done

elif [ ! -z $REDIS_POP ]; then
redis_blpop_var "$REDIS_GET" 0 >&$FD
redis_read $FD

else
redis_get_var "$REDIS_GET" >&$FD
redis_read $FD
Expand All @@ -196,21 +276,44 @@ if [[ ! -z $REDIS_GET ]]; then
exit 0
fi

while read -r line
do
REDIS_TODO=$line
done </dev/stdin
if [[ ! -z $REDIS_DEL ]]; then
redis_del_var "$REDIS_DEL" >&$FD
redis_read $FD
exec {FD}>&-
exit 0
fi

if [[ ! -z $REDIS_HINCRBY ]]; then
redis_hincrby "$REDIS_HINCRBY" "$REDIS_HINCRBY_FIELD" "$REDIS_HINCRBY_VALUE" >&$FD
redis_read $FD
exec {FD}>&-
exit 0
fi

if [[ -z $INPUT_RAW ]]; then
while read -r line
do
REDIS_TODO=$line
done </dev/stdin
else
REDIS_TODO=`cat`
fi

if [[ ! -z $REDIS_SET ]]; then
if [[ $REDIS_ARRAY -eq 1 ]]; then
set -- $REDIS_TODO
typeset -a temparray=( $@ )
redis_set_array "$REDIS_SET" temparray[@] >&$FD
redis_read $FD 1>/dev/null 2>&1
if [[ -z $INPUT_RAW ]]; then
set -- $REDIS_TODO
typeset -a temparray=( $@ )
redis_set_array "$REDIS_SET" temparray[@] >&$FD
else
redis_set_array "$REDIS_SET" "$REDIS_TODO" >&$FD
fi
elif [[ ! -z $REDIS_HASH ]]; then
redis_hset_var "$REDIS_SET" "$REDIS_HASH" "$REDIS_TODO" >&$FD
else
redis_set_var "$REDIS_SET" "$REDIS_TODO" >&$FD
redis_read $FD 1>/dev/null 2>&1
fi
redis_read $FD 1>/dev/null 2>&1
exec {FD}>&-
exit 0
fi