Table of Contents
In this tutorial, you’ll learn how to use Python with Redis (pronouncedRED-iss, or maybeREE-diss orRed-DEES, depending on who you ask), which is a lightning fast in-memory key-value store that can be used for anything from A to Z. Here’s whatSeven Databases in Seven Weeks, a popular book on databases, has to say about Redis:
It’s not simply easy to use; it’s a joy. If an API is UX for programmers, then Redis should be in the Museum of Modern Art alongside the Mac Cube.
…
And when it comes to speed, Redis is hard to beat. Reads are fast, and writes are even faster, handling upwards of 100,000
SET
operations per second by some benchmarks. (Source)
Intrigued? This tutorial is built for the Python programmer who may have zero to little Redis experience. We’ll tackle two tools at once and introduce both Redis itself as well as one of its Python client libraries,redis-py
.
redis-py
(which youimport as justredis
) is one of many Python clients for Redis, but it has the distinction of being billed as“currently the way to go for Python” by the Redis developers themselves. It lets you call Redis commands from Python, and get back familiar Python objects in return.
In this tutorial, you’ll cover:
redis-py
while also seeing glimpses of how it implements Redis’ protocolFree Download:Get a sample chapter from Python Tricks: The Book that shows you Python’s best practices with simple examples you can apply instantly to write more beautiful + Pythonic code.
As my great-great-grandfather said, nothing builds grit like installing from source. This section will walk you through downloading, making, and installing Redis. I promise that this won’t hurt one bit!
Note: This section is oriented towards installation on Mac OS X or Linux. If you’re using Windows, there is a Microsoftfork of Redis that can be installed as a Windows Service. Suffice it to say that Redis as a program lives most comfortably on a Linux box and that setup and use on Windows may be finicky.
First, download the Redis source code as a tarball:
$redisurl="http://download.redis.io/redis-stable.tar.gz"$curl-s-oredis-stable.tar.gz$redisurl
Next, switch over toroot
and extract the archive’s source code to/usr/local/lib/
:
$sudosuroot$mkdir-p/usr/local/lib/$chmoda+w/usr/local/lib/$tar-C/usr/local/lib/-xzfredis-stable.tar.gz
Optionally, you can now remove the archive itself:
$rmredis-stable.tar.gz
This will leave you with a source code repository at/usr/local/lib/redis-stable/
. Redis is written in C, so you’ll need to compile, link, and install with themake
utility:
$cd/usr/local/lib/redis-stable/$make&&makeinstall
Usingmake install
does two actions:
The firstmake
command compiles and links the source code.
Themake install
part takes the binaries and copies them to/usr/local/bin/
so that you can run them from anywhere (assuming that/usr/local/bin/
is inPATH
).
Here are all the steps so far:
$redisurl="http://download.redis.io/redis-stable.tar.gz"$curl-s-oredis-stable.tar.gz$redisurl$sudosuroot$mkdir-p/usr/local/lib/$chmoda+w/usr/local/lib/$tar-C/usr/local/lib/-xzfredis-stable.tar.gz$rmredis-stable.tar.gz$cd/usr/local/lib/redis-stable/$make&&makeinstall
At this point, take a moment to confirm that Redis is in yourPATH
and check its version:
$redis-cli--versionredis-cli 5.0.3
If your shell can’t findredis-cli
, check to make sure that/usr/local/bin/
is on yourPATH
environment variable, and add it if not.
In addition toredis-cli
,make install
actually leads to a handful of different executable files (and one symlink) being placed at/usr/local/bin/
:
$# A snapshot of executables that come bundled with Redis$ls-hFG/usr/local/bin/redis-*|sort/usr/local/bin/redis-benchmark*/usr/local/bin/redis-check-aof*/usr/local/bin/redis-check-rdb*/usr/local/bin/redis-cli*/usr/local/bin/redis-sentinel@/usr/local/bin/redis-server*
While all of these have some intended use, the two you’ll probably care about most areredis-cli
andredis-server
, which we’ll outline shortly. But before we get to that, setting up some baseline configuration is in order.
Redis is highly configurable. While it runs fine out of the box, let’s take a minute to set some bare-bones configuration options that relate to database persistence and basic security:
$sudosuroot$mkdir-p/etc/redis/$touch/etc/redis/6379.conf
Now, write the following to/etc/redis/6379.conf
. We’ll cover what most of these mean gradually throughout the tutorial:
# /etc/redis/6379.confport 6379daemonize yessave 60 1bind 127.0.0.1tcp-keepalive 300dbfilename dump.rdbdir ./rdbcompression yes
Redis configuration is self-documenting, with thesampleredis.conf
file located in the Redis source for your reading pleasure. If you’re using Redis in a production system, it pays to block out all distractions and take the time to read this sample file in full to familiarize yourself with the ins and outs of Redis and fine-tune your setup.
Some tutorials, including parts of Redis’ documentation, may also suggest running the Shell scriptinstall_server.sh
located inredis/utils/install_server.sh
. You’re by all means welcome to run this as a more comprehensive alternative to the above, but take note of a few finer points aboutinstall_server.sh
:
/etc/redis/6379.conf
.init
script to/etc/init.d/redis_6379
that will let you dosudo service redis_6379 start
.The Redis quickstart guide also contains a section on amore proper Redis setup, but the configuration options above should be totally sufficient for this tutorial and getting started.
Security Note: A few years back, the author of Redis pointed out security vulnerabilities in earlier versions of Redis if no configuration was set. Redis 3.2 (the current version 5.0.3 as of March 2019) made steps to prevent this intrusion, setting theprotected-mode
option toyes
by default.
We explicitly setbind 127.0.0.1
to let Redis listen for connections only from the localhost interface, although you would need to expand this whitelist in a real production server. The point ofprotected-mode
is as a safeguard that will mimic this bind-to-localhost behavior if you don’t otherwise specify anything under thebind
option.
With that squared away, we can now dig into using Redis itself.
This section will provide you with just enough knowledge of Redis to be dangerous, outlining its design and basic usage.
Redis has aclient-server architecture and uses arequest-response model. This means that you (the client) connect to a Redis server through TCP connection, on port 6379 by default. You request some action (like some form of reading, writing, getting, setting, or updating), and the serverserves you back a response.
There can be many clients talking to the same server, which is really what Redis or any client-server application is all about. Each client does a (typically blocking) read on a socket waiting for the server response.
Thecli
inredis-cli
stands forcommand line interface, and theserver
inredis-server
is for, well, running a server. In the same way that you would runpython
at the command line, you can runredis-cli
to jump into an interactive REPL (Read Eval Print Loop) where you can run client commands directly from the shell.
First, however, you’ll need to launchredis-server
so that you have a running Redis server to talk to. A common way to do this in development is to start a server atlocalhost (IPv4 address127.0.0.1
), which is the default unless you tell Redis otherwise. You can also passredis-server
the name of your configuration file, which is akin to specifying all of its key-value pairs ascommand-line arguments:
$redis-server/etc/redis/6379.conf31829:C 07 Mar 2019 08:45:04.030 # oO0OoO0OoO0Oo Redis is starting oO0OoO0OoO0Oo31829:C 07 Mar 2019 08:45:04.030 # Redis version=5.0.3, bits=64, commit=00000000, modified=0, pid=31829, just started31829:C 07 Mar 2019 08:45:04.030 # Configuration loaded
We set thedaemonize
configuration option toyes
, so the server runs in the background. (Otherwise, use--daemonize yes
as an option toredis-server
.)
Now you’re ready to launch the Redis REPL. Enterredis-cli
on your command line. You’ll see the server’shost:port pair followed by a>
prompt:
127.0.0.1:6379>
Here’s one of the simplest Redis commands,PING
, which just tests connectivity to the server and returns"PONG"
if things are okay:
127.0.0.1:6379>PINGPONG
Redis commands are case-insensitive, although their Python counterparts are most definitely not.
Note: As another sanity check, you can search for the process ID of the Redis server withpgrep
:
$pgrepredis-server26983
To kill the server, usepkill redis-server
from the command line. On Mac OS X, you can also useredis-cli shutdown
.
Next, we’ll use some of the common Redis commands and compare them to what they would look like in pure Python.
Redis stands forRemote Dictionary Service.
“You mean, like a Pythondictionary?” you may ask.
Yes. Broadly speaking, there are many parallels you can draw between a Python dictionary (or generichash table) and what Redis is and does:
A Redis database holdskey:value pairs and supports commands such asGET
,SET
, andDEL
, as well asseveral hundred additional commands.
Rediskeys are alwaysstrings.
Redisvalues may be a number of different data types. We’ll cover some of the more essential value data types in this tutorial:string
,list
,hashes
, andsets
. Some advanced types includegeospatial items and the newstream type.
Many Redis commands operate in constant O(1) time, just like retrieving a value from a Pythondict
or any hash table.
Redis creator Salvatore Sanfilippo would probably not love the comparison of a Redis database to a plain-vanilla Pythondict
. He calls the project a “data structure server” (rather than a key-value store, such asmemcached) because, to its credit, Redis supports storing additional types ofkey:value data types besidesstring:string. But for our purposes here, it’s a useful comparison if you’re familiar with Python’s dictionary object.
Let’s jump in and learn by example. Our first toy database (with ID 0) will be a mapping ofcountry:capital city, where we useSET
to set key-value pairs:
127.0.0.1:6379>SET Bahamas NassauOK127.0.0.1:6379>SET Croatia ZagrebOK127.0.0.1:6379>GET Croatia"Zagreb"127.0.0.1:6379>GET Japan(nil)
The corresponding sequence of statements in pure Python would look like this:
>>>capitals={}>>>capitals["Bahamas"]="Nassau">>>capitals["Croatia"]="Zagreb">>>capitals.get("Croatia")'Zagreb'>>>capitals.get("Japan")# None
We usecapitals.get("Japan")
rather thancapitals["Japan"]
because Redis will returnnil
rather than an error when a key is not found, which is analogous to Python’sNone
.
Redis also allows you to set and get multiple key-value pairs in one command,MSET
andMGET
, respectively:
127.0.0.1:6379>MSET Lebanon Beirut Norway Oslo France ParisOK127.0.0.1:6379>MGET Lebanon Norway Bahamas1) "Beirut"2) "Oslo"3) "Nassau"
The closest thing in Python is withdict.update()
:
>>>capitals.update({..."Lebanon":"Beirut",..."Norway":"Oslo",..."France":"Paris",...})>>>[capitals.get(k)forkin("Lebanon","Norway","Bahamas")]['Beirut', 'Oslo', 'Nassau']
We use.get()
rather than.__getitem__()
to mimic Redis’ behavior of returning a null-like value when no key is found.
As a third example, theEXISTS
command does what it sounds like, which is to check if a key exists:
127.0.0.1:6379>EXISTS Norway(integer) 1127.0.0.1:6379>EXISTS Sweden(integer) 0
Python has thein
keyword to test the same thing, which routes todict.__contains__(key)
:
>>>"Norway"incapitalsTrue>>>"Sweden"incapitalsFalse
These few examples are meant to show, using native Python, what’s happening at a high level with a few common Redis commands. There’s no client-server component here to the Python examples, andredis-py
has not yet entered the picture. This is only meant to show Redis functionality by example.
Here’s a summary of the few Redis commands you’ve seen and their functional Python equivalents:
capitals["Bahamas"]="Nassau"
capitals.get("Croatia")
capitals.update({"Lebanon":"Beirut","Norway":"Oslo","France":"Paris",})
[capitals[k]forkin("Lebanon","Norway","Bahamas")]
"Norway"incapitals
The Python Redis client library,redis-py
, that you’ll dive into shortly in this article, does things differently. It encapsulates an actual TCP connection to a Redis server and sends raw commands, as bytes serialized using theREdis Serialization Protocol (RESP), to the server. It then takes the raw reply and parses it back into a Python object such asbytes
,int
, or evendatetime.datetime
.
Note: So far, you’ve been talking to the Redis server through the interactiveredis-cli
REPL. You can alsoissue commands directly, in the same way that you would pass the name of a script to thepython
executable, such aspython myscript.py
.
So far, you’ve seen a few of Redis’ fundamental data types, which is a mapping ofstring:string. While this key-value pair is common in most key-value stores, Redis offers a number of other possible value types, which you’ll see next.
Before you fire up theredis-py
Python client, it also helps to have a basic grasp on a few more Redis data types. To be clear, all Redis keys are strings. It’s the value that can take on data types (or structures) in addition to the string values used in the examples so far.
Ahash is a mapping ofstring:string, calledfield-value pairs, that sits under one top-level key:
127.0.0.1:6379>HSET realpython url "https://realpython.com/"(integer) 1127.0.0.1:6379>HSET realpython github realpython(integer) 1127.0.0.1:6379>HSET realpython fullname "Real Python"(integer) 1
This sets three field-value pairs for onekey,"realpython"
. If you’re used to Python’s terminology and objects, this can be confusing. A Redis hash is roughly analogous to a Pythondict
that is nested one level deep:
data={"realpython":{"url":"https://realpython.com/","github":"realpython","fullname":"Real Python",}}
Redis’ fields are akin to the Python keys of each nested key-value pair in the inner dictionary above. Redis reserves the termkey for the top-level database key that holds the hash structure itself.
Just like there’sMSET
for basicstring:string key-value pairs, there is alsoHMSET
for hashes to set multiple pairswithin the hash value object:
127.0.0.1:6379>HMSET pypa url "https://www.pypa.io/" github pypa fullname "Python Packaging Authority"OK127.0.0.1:6379>HGETALL pypa1) "url"2) "https://www.pypa.io/"3) "github"4) "pypa"5) "fullname"6) "Python Packaging Authority"
UsingHMSET
is probably a closer parallel for the way that we assigneddata
to a nested dictionary above, rather than setting each nested pair as is done withHSET
.
Two additional value types arelists andsets, which can take the place of a hash or string as a Redis value. They are largely what they sound like, so I won’t take up your time with additional examples. Hashes, lists, and sets each have some commands that are particular to that given data type, which are in some cases denoted by their initial letter:
Hashes: Commands to operate on hashes begin with anH
, such asHSET
,HGET
, orHMSET
.
Sets: Commands to operate on sets begin with anS
, such asSCARD
, which gets the number of elements at the set value corresponding to a given key.
Lists: Commands to operate on lists begin with anL
orR
. Examples includeLPOP
andRPUSH
. TheL
orR
refers to which side of the list is operated on. A few list commands are also prefaced with aB
, which meansblocking. A blocking operation doesn’t let other operations interrupt it while it’s executing. For instance,BLPOP
executes a blocking left-pop on a list structure.
Note: One noteworthy feature of Redis’ list type is that it is alinked list rather than an array. This means that appending is O(1) while indexing at an arbitrary index number is O(N).
Here is a quick listing of commands that are particular to the string, hash, list, and set data types in Redis:
Type | Commands |
---|---|
Sets | SADD ,SCARD ,SDIFF ,SDIFFSTORE ,SINTER ,SINTERSTORE ,SISMEMBER ,SMEMBERS ,SMOVE ,SPOP ,SRANDMEMBER ,SREM ,SSCAN ,SUNION ,SUNIONSTORE |
Hashes | HDEL ,HEXISTS ,HGET ,HGETALL ,HINCRBY ,HINCRBYFLOAT ,HKEYS ,HLEN ,HMGET ,HMSET ,HSCAN ,HSET ,HSETNX ,HSTRLEN ,HVALS |
Lists | BLPOP ,BRPOP ,BRPOPLPUSH ,LINDEX ,LINSERT ,LLEN ,LPOP ,LPUSH ,LPUSHX ,LRANGE ,LREM ,LSET ,LTRIM ,RPOP ,RPOPLPUSH ,RPUSH ,RPUSHX |
Strings | APPEND ,BITCOUNT ,BITFIELD ,BITOP ,BITPOS ,DECR ,DECRBY ,GET ,GETBIT ,GETRANGE ,GETSET ,INCR ,INCRBY ,INCRBYFLOAT ,MGET ,MSET ,MSETNX ,PSETEX ,SET ,SETBIT ,SETEX ,SETNX ,SETRANGE ,STRLEN |
This table isn’t a complete picture of Redis commands and types. There’s a smorgasbord of more advanced data types, such asgeospatial items,sorted sets, andHyperLogLog. At the Rediscommands page, you can filter by data-structure group. There is also thedata types summary andintroduction to Redis data types.
Since we’re going to be switching over to doing things in Python, you can now clear your toy database withFLUSHDB
and quit theredis-cli
REPL:
127.0.0.1:6379>FLUSHDBOK127.0.0.1:6379>QUIT
This will bring you back to your shell prompt. You can leaveredis-server
running in the background, since you’ll need it for the rest of the tutorial also.
redis-py
: Redis in PythonNow that you’ve mastered some basics of Redis, it’s time to jump intoredis-py
, the Python client that lets you talk to Redis from a user-friendly Python API.
redis-py
is a well-established Python client library that lets you talk to a Redis server directly through Python calls:
$python-mpipinstallredis
Next, make sure that your Redis server is still up and running in the background. You can check withpgrep redis-server
, and if you come up empty-handed, then restart a local server withredis-server /etc/redis/6379.conf
.
Now, let’s get into the Python-centric part of things. Here’s the “hello world” ofredis-py
:
1>>>importredis 2>>>r=redis.Redis() 3>>>r.mset({"Croatia":"Zagreb","Bahamas":"Nassau"}) 4True 5>>>r.get("Bahamas") 6b'Nassau'
Redis
, used in Line 2, is the central class of the package and the workhorse by which you execute (almost) any Redis command. The TCP socket connection and reuse is done for you behind the scenes, and you call Redis commands using methods on the class instancer
.
Notice also that the type of the returned object,b'Nassau'
in Line 6, is Python’sbytes
type, notstr
. It isbytes
rather thanstr
that is the most common return type acrossredis-py
, so you may need to callr.get("Bahamas").decode("utf-8")
depending on what you want to actually do with the returned bytestring.
Does the code above look familiar? The methods in almost all cases match the name of the Redis command that does the same thing. Here, you calledr.mset()
andr.get()
, which correspond toMSET
andGET
in the native Redis API.
This also means thatHGETALL
becomesr.hgetall()
,PING
becomesr.ping()
, and so on. There are afew exceptions, but the rule holds for the large majority of commands.
While the Redis command arguments usually translate into a similar-looking method signature, they take Python objects. For example, the call tor.mset()
in the example above uses a Pythondict
as its first argument, rather than a sequence of bytestrings.
We built theRedis
instancer
with no arguments, but it comes bundled with a number ofparameters if you need them:
# From redis/client.pyclassRedis(object):def__init__(self,host='localhost',port=6379,db=0,password=None,socket_timeout=None,# ...
You can see that the defaulthostname:port pair islocalhost:6379
, which is exactly what we need in the case of our locally keptredis-server
instance.
Thedb
parameter is the database number. You can manage multiple databases in Redis at once, and each is identified by an integer. The max number of databases is 16 by default.
When you run justredis-cli
from the command line, this starts you at database 0. Use the-n
flag to start a new database, as inredis-cli -n 5
.
One thing that’s worth knowing is thatredis-py
requires that you pass it keys that arebytes
,str
,int
, orfloat
. (It will convert the last 3 of these types tobytes
before sending them off to the server.)
Consider a case where you want to use calendar dates as keys:
>>>importdatetime>>>today=datetime.date.today()>>>visitors={"dan","jon","alex"}>>>r.sadd(today,*visitors)Traceback (most recent call last):# ...redis.exceptions.DataError:Invalid input of type: 'date'.Convert to a byte, string or number first.
You’ll need to explicitly convert the Pythondate
object tostr
, which you can do with.isoformat()
:
>>>stoday=today.isoformat()# Python 3.7+, or use str(today)>>>stoday'2019-03-10'>>>r.sadd(stoday,*visitors)# sadd: set-add3>>>r.smembers(stoday){b'dan', b'alex', b'jon'}>>>r.scard(today.isoformat())3
To recap, Redis itself only allows strings as keys.redis-py
is a bit more liberal in what Python types it will accept, although it ultimately converts everything to bytes before sending them off to a Redis server.
It’s time to break out a fuller example. Let’s pretend we’ve decided to start a lucrative website, PyHats.com, that sells outrageously overpriced hats to anyone who will buy them, and hired you to build the site.
You’ll use Redis to handle some of the product catalog, inventorying, and bot traffic detection for PyHats.com.
It’s day one for the site, and we’re going to be selling three limited-edition hats. Each hat gets held in a Redis hash of field-value pairs, and the hash has a key that is a prefixed random integer , such ashat:56854717
. Using thehat:
prefix is Redis convention for creating a sort ofnamespace within a Redis database:
importrandomrandom.seed(444)hats={f"hat:{random.getrandbits(32)}":iforiin({"color":"black","price":49.99,"style":"fitted","quantity":1000,"npurchased":0,},{"color":"maroon","price":59.99,"style":"hipster","quantity":500,"npurchased":0,},{"color":"green","price":99.99,"style":"baseball","quantity":200,"npurchased":0,})}
Let’s start with database1
since we used database0
in a previous example:
>>>r=redis.Redis(db=1)
To do an initial write of this data into Redis, we can use.hmset()
(hash multi-set), calling it for each dictionary. The “multi” is a reference to setting multiple field-value pairs, where “field” in this case corresponds to a key of any of the nested dictionaries inhats
:
1>>>withr.pipeline()aspipe: 2...forh_id,hatinhats.items(): 3...pipe.hmset(h_id,hat) 4...pipe.execute() 5Pipeline<ConnectionPool<Connection<host=localhost,port=6379,db=1>>> 6Pipeline<ConnectionPool<Connection<host=localhost,port=6379,db=1>>> 7Pipeline<ConnectionPool<Connection<host=localhost,port=6379,db=1>>> 8[True,True,True] 910>>>r.bgsave()11True
The code block above also introduces the concept of Redispipelining, which is a way to cut down the number of round-trip transactions that you need to write or read data from your Redis server. If you would have just calledr.hmset()
three times, then this would necessitate a back-and-forth round trip operation for each row written.
With a pipeline, all the commands are buffered on the client side and then sent at once, in one fell swoop, usingpipe.hmset()
in Line 3. This is why the threeTrue
responses are all returned at once, when you callpipe.execute()
in Line 4. You’ll see a more advanced use case for a pipeline shortly.
Note: The Redis docs provide anexample of doing this same thing with theredis-cli
, where you can pipe the contents of a local file to do mass insertion.
Let’s do a quick check that everything is there in our Redis database:
>>>pprint(r.hgetall("hat:56854717")){b'color': b'green', b'npurchased': b'0', b'price': b'99.99', b'quantity': b'200', b'style': b'baseball'}>>>r.keys()# Careful on a big DB. keys() is O(N)[b'56854717', b'1236154736', b'1326692461']
The first thing that we want to simulate is what happens when a user clicksPurchase. If the item is in stock, increase itsnpurchased
by 1 and decrease itsquantity
(inventory) by 1. You can use.hincrby()
to do this:
>>>r.hincrby("hat:56854717","quantity",-1)199>>>r.hget("hat:56854717","quantity")b'199'>>>r.hincrby("hat:56854717","npurchased",1)1
Note:HINCRBY
still operates on a hash value that is a string, but it tries to interpret the string as a base-10 64-bit signed integer to execute the operation.
This applies to other commands related to incrementing and decrementing for other data structures, namelyINCR
,INCRBY
,INCRBYFLOAT
,ZINCRBY
, andHINCRBYFLOAT
. You’ll get an error if the string at the value can’t be represented as an integer.
It isn’t really that simple, though. Changing thequantity
andnpurchased
in two lines of code hides the reality that a click, purchase, and payment entails more than this. We need to do a few more checks to make sure we don’t leave someone with a lighter wallet and no hat:
quantity
field, and increase thenpurchased
field.Step 1 is relatively straightforward: it consists of an.hget()
to check the available quantity.
Step 2 is a little bit more involved. The pair of increase and decrease operations need to be executedatomically: either both should be completed successfully, or neither should be (in the case that at least one fails).
With client-server frameworks, it’s always crucial to pay attention to atomicity and look out for what could go wrong in instances where multiple clients are trying to talk to the server at once. The answer to this in Redis is to use atransaction block, meaning that either both or neither of the commands get through.
Inredis-py
,Pipeline
is atransactional pipeline class by default. This means that, even though the class is actually named for something else (pipelining), it can be used to create a transaction block also.
In Redis, a transaction starts withMULTI
and ends withEXEC
:
1127.0.0.1:6379>MULTI 2127.0.0.1:6379>HINCRBY 56854717 quantity -1 3127.0.0.1:6379>HINCRBY 56854717 npurchased 1 4127.0.0.1:6379>EXEC
MULTI
(Line 1) marks the start of the transaction, andEXEC
(Line 4) marks the end. Everything in between is executed as one all-or-nothing buffered sequence of commands. This means that it will be impossible to decrementquantity
(Line 2) but then have the balancingnpurchased
increment operation fail (Line 3).
Let’s circle back to Step 3: we need to be aware of any changes that alter the inventory in between the first two steps.
Step 3 is the trickiest. Let’s say that there is one lone hat remaining in our inventory. In between the time that User A checks the quantity of hats remaining and actually processes their transaction, User B also checks the inventory and finds likewise that there is one hat listed in stock. Both users will be allowed to purchase the hat, but we have 1 hat to sell, not 2, so we’re on the hook and one user is out of their money. Not good.
Redis has a clever answer for the dilemma in Step 3: it’s calledoptimistic locking, and is different than how typical locking works in an RDBMS such as PostgreSQL. Optimistic locking, in a nutshell, means that the calling function (client) does not acquire a lock, but rather monitors for changes in the data it is writing toduring the time it would have held a lock. If there’s a conflict during that time, the calling function simply tries the whole process again.
You can effect optimistic locking by using theWATCH
command (.watch()
inredis-py
), which provides acheck-and-set behavior.
Let’s introduce a big chunk of code and walk through it afterwards step by step. You can picturebuyitem()
as being called any time a user clicks on aBuy Now orPurchase button. Its purpose is to confirm the item is in stock and take an action based on that result, all in a safe manner that looks out for race conditions and retries if one is detected:
1importlogging 2importredis 3 4logging.basicConfig() 5 6classOutOfStockError(Exception): 7"""Raised when PyHats.com is all out of today's hottest hat""" 8 9defbuyitem(r:redis.Redis,itemid:int)->None:10withr.pipeline()aspipe:11error_count=012whileTrue:13try:14# Get available inventory, watching for changes15# related to this itemid before the transaction16pipe.watch(itemid)17nleft:bytes=r.hget(itemid,"quantity")18ifnleft>b"0":19pipe.multi()20pipe.hincrby(itemid,"quantity",-1)21pipe.hincrby(itemid,"npurchased",1)22pipe.execute()23break24else:25# Stop watching the itemid and raise to break out26pipe.unwatch()27raiseOutOfStockError(28f"Sorry,{itemid} is out of stock!"29)30exceptredis.WatchError:31# Log total num. of errors by this user to buy this item,32# then try the same process again of WATCH/HGET/MULTI/EXEC33error_count+=134logging.warning(35"WatchError #%d:%s; retrying",36error_count,itemid37)38returnNone
The critical line occurs at Line 16 withpipe.watch(itemid)
, which tells Redis to monitor the givenitemid
for any changes to its value. The program checks the inventory through the call tor.hget(itemid, "quantity")
, in Line 17:
16pipe.watch(itemid)17nleft:bytes=r.hget(itemid,"quantity")18ifnleft>b"0":19# Item in stock. Proceed with transaction.
If the inventory gets touched during this short window between when the user checks the item stock and tries to purchase it, then Redis will return an error, andredis-py
will raise aWatchError
(Line 30). That is, if any of the hash pointed to byitemid
changes after the.hget()
call but before the subsequent.hincrby()
calls in Lines 20 and 21, then we’ll re-run the whole process in another iteration of thewhile True
loop as a result.
This is the “optimistic” part of the locking: rather than letting the client have a time-consuming total lock on the database through the getting and setting operations, we leave it up to Redis to notify the client and user only in the case that calls for a retry of the inventory check.
One key here is in understanding the difference betweenclient-side andserver-side operations:
nleft=r.hget(itemid,"quantity")
This Python assignment brings the result ofr.hget()
client-side. Conversely, methods that you call onpipe
effectively buffer all of the commands into one, and then send them to the server in a single request:
16pipe.multi()17pipe.hincrby(itemid,"quantity",-1)18pipe.hincrby(itemid,"npurchased",1)19pipe.execute()
No data comes back to the client side in the middle of the transactional pipeline. You need to call.execute()
(Line 19) to get the sequence of results back all at once.
Even though this block contains two commands, it consists of exactly one round-trip operation from client to server and back.
This means that the client can’t immediatelyuse the result ofpipe.hincrby(itemid, "quantity", -1)
, from Line 20, because methods on aPipeline
return just thepipe
instance itself. We haven’t asked anything from the server at this point. While normally.hincrby()
returns the resulting value, you can’t immediately reference it on the client side until the entire transaction is completed.
There’s a catch-22: this is also why you can’t put the call to.hget()
into the transaction block. If you did this, then you’d be unable to know if you want to increment thenpurchased
field yet, since you can’t get real-time results from commands that are inserted into a transactional pipeline.
Finally, if the inventory sits at zero, then weUNWATCH
the item ID and raise anOutOfStockError
(Line 27), ultimately displaying that covetedSold Out page that will make our hat buyers desperately want to buy even more of our hats at ever more outlandish prices:
24else:25# Stop watching the itemid and raise to break out26pipe.unwatch()27raiseOutOfStockError(28f"Sorry,{itemid} is out of stock!"29)
Here’s an illustration. Keep in mind that our starting quantity is199
for hat 56854717 since we called.hincrby()
above. Let’s mimic 3 purchases, which should modify thequantity
andnpurchased
fields:
>>>buyitem(r,"hat:56854717")>>>buyitem(r,"hat:56854717")>>>buyitem(r,"hat:56854717")>>>r.hmget("hat:56854717","quantity","npurchased")# Hash multi-get[b'196', b'4']
Now, we can fast-forward through more purchases, mimicking a stream of purchases until the stock depletes to zero. Again, picture these coming from a whole bunch of different clients rather than just oneRedis
instance:
>>># Buy remaining 196 hats for item 56854717 and deplete stock to 0>>>for_inrange(196):...buyitem(r,"hat:56854717")>>>r.hmget("hat:56854717","quantity","npurchased")[b'0', b'200']
Now, when some poor user is late to the game, they should be met with anOutOfStockError
that tells our application to render an error message page on the frontend:
>>>buyitem(r,"hat:56854717")Traceback (most recent call last): File"<stdin>", line1, in<module> File"<stdin>", line20, inbuyitem__main__.OutOfStockError:Sorry, hat:56854717 is out of stock!
Looks like it’s time to restock.
Let’s introducekey expiry, which is another distinguishing feature in Redis. When youexpire a key, that key and its corresponding value will be automatically deleted from the database after a certain number of seconds or at a certain timestamp.
Inredis-py
, one way that you can accomplish this is through.setex()
, which lets you set a basicstring:string key-value pair with an expiration:
1>>>fromdatetimeimporttimedelta 2 3>>># setex: "SET" with expiration 4>>>r.setex( 5..."runner", 6...timedelta(minutes=1), 7...value="now you see me, now you don't" 8...) 9True
You can specify the second argument as a number in seconds or atimedelta
object, as in Line 6 above. I like the latter because it seems less ambiguous and more deliberate.
There are also methods (and corresponding Redis commands, of course) to get the remaining lifetime (time-to-live) of a key that you’ve set to expire:
>>>r.ttl("runner")# "Time To Live", in seconds58>>>r.pttl("runner")# Like ttl, but milliseconds54368
Below, you can accelerate the window until expiration, and then watch the key expire, after whichr.get()
will returnNone
and.exists()
will return0
:
>>>r.get("runner")# Not expired yetb"now you see me, now you don't">>>r.expire("runner",timedelta(seconds=3))# Set new expire windowTrue>>># Pause for a few seconds>>>r.get("runner")>>>r.exists("runner")# Key & value are both gone (expired)0
The table below summarizes commands related to key-value expiration, including the ones covered above. The explanations are taken directly fromredis-py
methoddocstrings:
Signature | Purpose |
---|---|
r.setex(name, time, value) | Sets the value of keyname tovalue that expires intime seconds, wheretime can be represented by anint or a Pythontimedelta object |
r.psetex(name, time_ms, value) | Sets the value of keyname tovalue that expires intime_ms milliseconds, wheretime_ms can be represented by anint or a Pythontimedelta object |
r.expire(name, time) | Sets an expire flag on keyname fortime seconds, wheretime can be represented by anint or a Pythontimedelta object |
r.expireat(name, when) | Sets an expire flag on keyname , wherewhen can be represented as anint indicating Unix time or aPythondatetime object |
r.persist(name) | Removes an expiration onname |
r.pexpire(name, time) | Sets an expire flag on keyname fortime milliseconds, andtime can be represented by anint or a Pythontimedelta object |
r.pexpireat(name, when) | Sets an expire flag on keyname , wherewhen can be represented as anint representing Unix time in milliseconds (Unix time * 1000) or a Pythondatetime object |
r.pttl(name) | Returns the number of milliseconds until the keyname will expire |
r.ttl(name) | Returns the number of seconds until the keyname will expire |
A few days after its debut, PyHats.com has attracted so much hype that some enterprising users are creating bots to buy hundreds of items within seconds, which you’ve decided isn’t good for the long-term health of your hat business.
Now that you’ve seen how to expire keys, let’s put it to use on the backend of PyHats.com.
We’re going to create a new Redis client that acts as a consumer (or watcher) and processes a stream of incoming IP addresses, which in turn may come from multiple HTTPS connections to the website’s server.
The watcher’s goal is to monitor a stream of IP addresses from multiple sources, keeping an eye out for a flood of requests from a single address within a suspiciously short amount of time.
Some middleware on the website server pushes all incoming IP addresses into a Redis list with.lpush()
. Here’s a crude way of mimicking some incoming IPs, using a fresh Redis database:
>>>r=redis.Redis(db=5)>>>r.lpush("ips","51.218.112.236")1>>>r.lpush("ips","90.213.45.98")2>>>r.lpush("ips","115.215.230.176")3>>>r.lpush("ips","51.218.112.236")4
As you can see,.lpush()
returns the length of the list after the push operation succeeds. Each call of.lpush()
puts the IP at the beginning of the Redis list that is keyed by the string"ips"
.
In this simplified simulation, the requests are all technically from the same client, but you can think of them as potentially coming from many different clients and all being pushed to the same database on the same Redis server.
Now, open up a new shell tab or window and launch a newPython REPL. In this shell, you’ll create a new client that serves a very different purpose than the rest, which sits in an endlesswhile True
loop and does a blocking left-popBLPOP
call on theips
list, processing each address:
1# New shell window or tab 2 3importdatetime 4importipaddress 5 6importredis 7 8# Where we put all the bad egg IP addresses 9blacklist=set()10MAXVISITS=151112ipwatcher=redis.Redis(db=5)1314whileTrue:15_,addr=ipwatcher.blpop("ips")16addr=ipaddress.ip_address(addr.decode("utf-8"))17now=datetime.datetime.utcnow()18addrts=f"{addr}:{now.minute}"19n=ipwatcher.incrby(addrts,1)20ifn>=MAXVISITS:21print(f"Hat bot detected!:{addr}")22blacklist.add(addr)23else:24print(f"{now}: saw{addr}")25_=ipwatcher.expire(addrts,60)
Let’s walk through a few important concepts.
Theipwatcher
acts like aconsumer, sitting around and waiting for new IPs to be pushed on the"ips"
Redis list. It receives them asbytes
, such as b”51.218.112.236”, and makes them into a more properaddress object with theipaddress
module:
15_,addr=ipwatcher.blpop("ips")16addr=ipaddress.ip_address(addr.decode("utf-8"))
Then you form a Redis string key using the address and minute of the hour at which theipwatcher
saw the address, incrementing the corresponding count by1
and getting the new count in the process:
17now=datetime.datetime.utcnow()18addrts=f"{addr}:{now.minute}"19n=ipwatcher.incrby(addrts,1)
If the address has been seen more thanMAXVISITS
, then it looks as if we have a PyHats.com web scraper on our hands trying to create the nexttulip bubble. Alas, we have no choice but to give this user back something like a dreaded 403 status code.
We useipwatcher.expire(addrts, 60)
to expire the(address minute) combination 60 seconds from when it was last seen. This is to prevent our database from becoming clogged up with stale one-time page viewers.
If you execute this code block in a new shell, you should immediately see this output:
2019-03-11 15:10:41.489214: saw 51.218.112.2362019-03-11 15:10:41.490298: saw 115.215.230.1762019-03-11 15:10:41.490839: saw 90.213.45.982019-03-11 15:10:41.491387: saw 51.218.112.236
The output appears right away because those four IPs were sitting in the queue-like list keyed by"ips"
, waiting to be pulled out by ouripwatcher
. Using.blpop()
(or theBLPOP
command) will block until an item is available in the list, then pops it off. It behaves like Python’sQueue.get()
, which also blocks until an item is available.
Besides just spitting out IP addresses, ouripwatcher
has a second job. For a given minute of an hour (minute 1 through minute 60),ipwatcher
will classify an IP address as a hat-bot if it sends 15 or moreGET
requests in that minute.
Switch back to your first shell and mimic a page scraper that blasts the site with 20 requests in a few milliseconds:
for_inrange(20):r.lpush("ips","104.174.118.18")
Finally, toggle back to the second shell holdingipwatcher
, and you should see an output like this:
2019-03-11 15:15:43.041363: saw 104.174.118.182019-03-11 15:15:43.042027: saw 104.174.118.182019-03-11 15:15:43.042598: saw 104.174.118.182019-03-11 15:15:43.043143: saw 104.174.118.182019-03-11 15:15:43.043725: saw 104.174.118.182019-03-11 15:15:43.044244: saw 104.174.118.182019-03-11 15:15:43.044760: saw 104.174.118.182019-03-11 15:15:43.045288: saw 104.174.118.182019-03-11 15:15:43.045806: saw 104.174.118.182019-03-11 15:15:43.046318: saw 104.174.118.182019-03-11 15:15:43.046829: saw 104.174.118.182019-03-11 15:15:43.047392: saw 104.174.118.182019-03-11 15:15:43.047966: saw 104.174.118.182019-03-11 15:15:43.048479: saw 104.174.118.18Hat bot detected!: 104.174.118.18Hat bot detected!: 104.174.118.18Hat bot detected!: 104.174.118.18Hat bot detected!: 104.174.118.18Hat bot detected!: 104.174.118.18Hat bot detected!: 104.174.118.18
Now,Ctrl+C out of thewhile True
loop and you’ll see that the offending IP has been added to your blacklist:
>>>blacklist{IPv4Address('104.174.118.18')}
Can you find the defect in this detection system? The filter checks the minute as.minute
rather than thelast 60 seconds (a rolling minute). Implementing a rolling check to monitor how many times a user has been seen in the last 60 seconds would be trickier. There’s a crafty solution using using Redis’ sorted sets atClassDojo. Josiah Carlson’sRedis in Action also presents a more elaborate and general-purpose example of this section using an IP-to-location cache table.
One of the reasons that Redis is so fast in both read and write operations is that the database is held in memory (RAM) on the server. However, a Redis database can also be stored (persisted) to disk in a process calledsnapshotting. The point behind this is to keep a physical backup in binary format so that data can be reconstructed and put back into memory when needed, such as at server startup.
You already enabled snapshotting without knowing it when you set up basic configuration at the beginning of this tutorial with thesave
option:
# /etc/redis/6379.confport 6379daemonize yessave 60 1bind 127.0.0.1tcp-keepalive 300dbfilename dump.rdbdir ./rdbcompression yes
The format issave <seconds> <changes>
. This tells Redis to save the database to disk if both the given number of seconds and number of write operations against the database occurred. In this case, we’re telling Redis to save the database to disk every 60 seconds if at least one modifying write operation occurred in that 60-second timespan. This is a fairly aggressive setting versus thesample Redis config file, which uses these threesave
directives:
# Default redis/redis.confsave 900 1save 300 10save 60 10000
AnRDB snapshot is a full (rather than incremental) point-in-time capture of the database. (RDB refers to a Redis Database File.) We also specified the directory and file name of the resulting data file that gets written:
# /etc/redis/6379.confport 6379daemonize yessave 60 1bind 127.0.0.1tcp-keepalive 300dbfilename dump.rdbdir ./rdbcompression yes
This instructs Redis to save to a binary data file calleddump.rdb
in the current working directory of whereverredis-server
was executed from:
$file-bdump.rdbdata
You can also manually invoke a save with the Redis commandBGSAVE
:
127.0.0.1:6379>BGSAVEBackground saving started
The “BG” inBGSAVE
indicates that the save occurs in the background. This option is available in aredis-py
method as well:
>>>r.lastsave()# Redis command: LASTSAVEdatetime.datetime(2019, 3, 10, 21, 56, 50)>>>r.bgsave()True>>>r.lastsave()datetime.datetime(2019, 3, 10, 22, 4, 2)
This example introduces another new command and method,.lastsave()
. In Redis, it returns the Unix timestamp of the last DB save, which Python gives back to you as adatetime
object. Above, you can see that ther.lastsave()
result changes as a result ofr.bgsave()
.
r.lastsave()
will also change if you enable automatic snapshotting with thesave
configuration option.
To rephrase all of this, there are two ways to enable snapshotting:
BGSAVE
orredis-py
method.bgsave()
save
configuration option (which you can also set with.config_set()
inredis-py
)RDB snapshotting is fast because the parent process uses thefork()
system call to pass off the time-intensive write to disk to a child process, so that the parent process can continue on its way. This is what thebackground inBGSAVE
refers to.
There’s alsoSAVE
(.save()
inredis-py
), but this does a synchronous (blocking) save rather than usingfork()
, so you shouldn’t use it without a specific reason.
Even though.bgsave()
occurs in the background, it’s not without its costs. The time forfork()
itself to occur can actually be substantial if the Redis database is large enough in the first place.
If this is a concern, or if you can’t afford to miss even a tiny slice of data lost due to the periodic nature of RDB snapshotting, then you should look into theappend-only file (AOF) strategy that is an alternative to snapshotting. AOF copies Redis commands to disk in real time, allowing you to do a literal command-based reconstruction by replaying these commands.
Let’s get back to talking about Redis data structures. With its hash data structure, Redis in effect supports nesting one level deep:
127.0.0.1:6379>hset mykey field1 value1
The Python client equivalent would look like this:
r.hset("mykey","field1","value1")
Here, you can think of"field1": "value1"
as being the key-value pair of a Python dict,{"field1": "value1"}
, whilemykey
is the top-level key:
Redis Command | Pure-Python Equivalent |
---|---|
r.set("key", "value") | r = {"key": "value"} |
r.hset("key", "field", "value") | r = {"key": {"field": "value"}} |
But what if you want the value of this dictionary (the Redis hash) to contain something other than a string, such as alist
or nested dictionary with strings as values?
Here’s an example using someJSON-like data to make the distinction clearer:
restaurant_484272={"name":"Ravagh","type":"Persian","address":{"street":{"line1":"11 E 30th St","line2":"APT 1",},"city":"New York","state":"NY","zip":10016,}}
Say that we want to set a Redis hash with the key484272
and field-value pairs corresponding to the key-value pairs fromrestaurant_484272
. Redis does not support this directly, becauserestaurant_484272
is nested:
>>>r.hmset(484272,restaurant_484272)Traceback (most recent call last):# ...redis.exceptions.DataError:Invalid input of type: 'dict'.Convert to a byte, string or number first.
You can in fact make this work with Redis. There are two different ways to mimic nested data inredis-py
and Redis:
json.dumps()
Let’s take a look at an example of each.
Option 1: Serialize the Values Into a String
You can usejson.dumps()
to serialize thedict
into a JSON-formatted string:
>>>importjson>>>r.set(484272,json.dumps(restaurant_484272))True
If you call.get()
, the value you get back will be abytes
object, so don’t forget to deserialize it to get back the original object.json.dumps()
andjson.loads()
are inverses of each other, for serializing and deserializing data, respectively:
>>>frompprintimportpprint>>>pprint(json.loads(r.get(484272))){'address': {'city': 'New York', 'state': 'NY', 'street': '11 E 30th St', 'zip': 10016}, 'name': 'Ravagh', 'type': 'Persian'}
This applies to any serialization protocol, with another common choice beingyaml
:
>>>importyaml# python -m pip install PyYAML>>>yaml.dump(restaurant_484272)'address: {city: New York, state: NY, street: 11 E 30th St, zip: 10016}\nname: Ravagh\ntype: Persian\n'
No matter what serialization protocol you choose to go with, the concept is the same: you’re taking an object that is unique to Python and converting it to a bytestring that is recognized and exchangeable across multiple languages.
Option 2: Use a Delimiter in Key Strings
There’s a second option that involves mimicking “nestedness” by concatenating multiple levels of keys in a Pythondict
. This consists of flattening the nested dictionary throughrecursion, so that each key is a concatenated string of keys, and the values are the deepest-nested values from the original dictionary. Consider our dictionary objectrestaurant_484272
:
restaurant_484272={"name":"Ravagh","type":"Persian","address":{"street":{"line1":"11 E 30th St","line2":"APT 1",},"city":"New York","state":"NY","zip":10016,}}
We want to get it into this form:
{"484272:name":"Ravagh","484272:type":"Persian","484272:address:street:line1":"11 E 30th St","484272:address:street:line2":"APT 1","484272:address:city":"New York","484272:address:state":"NY","484272:address:zip":"10016",}
That’s whatsetflat_skeys()
below does, with the added feature that it does inplace.set()
operations on theRedis
instance itself rather than returning a copy of the input dictionary:
1fromcollections.abcimportMutableMapping 2 3defsetflat_skeys( 4r:redis.Redis, 5obj:dict, 6prefix:str, 7delim:str=":", 8*, 9_autopfix=""10)->None:11"""Flatten `obj` and set resulting field-value pairs into `r`.1213 Calls `.set()` to write to Redis instance inplace and returns None.1415 `prefix` is an optional str that prefixes all keys.16 `delim` is the delimiter that separates the joined, flattened keys.17 `_autopfix` is used in recursive calls to created de-nested keys.1819 The deepest-nested keys must be str, bytes, float, or int.20 Otherwise a TypeError is raised.21 """22allowed_vtypes=(str,bytes,float,int)23forkey,valueinobj.items():24key=_autopfix+key25ifisinstance(value,allowed_vtypes):26r.set(f"{prefix}{delim}{key}",value)27elifisinstance(value,MutableMapping):28setflat_skeys(29r,value,prefix,delim,_autopfix=f"{key}{delim}"30)31else:32raiseTypeError(f"Unsupported value type:{type(value)}")
The function iterates over the key-value pairs ofobj
, first checking the type of the value (Line 25) to see if it looks like it should stop recursing further and set that key-value pair. Otherwise, if the value looks like adict
(Line 27), then it recurses into that mapping, adding the previously seen keys as a key prefix (Line 28).
Let’s see it at work:
>>>r.flushdb()# Flush database: clear old entries>>>setflat_skeys(r,restaurant_484272,484272)>>>forkeyinsorted(r.keys("484272*")):# Filter to this pattern...print(f"{repr(key):35}{repr(r.get(key)):15}")...b'484272:address:city' b'New York'b'484272:address:state' b'NY'b'484272:address:street:line1' b'11 E 30th St'b'484272:address:street:line2' b'APT 1'b'484272:address:zip' b'10016'b'484272:name' b'Ravagh'b'484272:type' b'Persian'>>>r.get("484272:address:street:line1")b'11 E 30th St'
The final loop above usesr.keys("484272*")
, where"484272*"
is interpreted as a pattern and matches all keys in the database that begin with"484272"
.
Notice also howsetflat_skeys()
calls just.set()
rather than.hset()
, because we’re working with plainstring:string field-value pairs, and the 484272 ID key is prepended to each field string.
Another trick to help you sleep well at night is to add symmetric encryption before sending anything to a Redis server. Consider this as an add-on to the security that you should make sure is in place by setting proper values in yourRedis configuration. The example below uses thecryptography
package:
$python-mpipinstallcryptography
To illustrate, pretend that you have some sensitive cardholder data (CD) that you never want to have sitting around in plaintext on any server, no matter what. Before caching it in Redis, you can serialize the data and then encrypt the serialized string usingFernet:
>>>importjson>>>fromcryptography.fernetimportFernet>>>cipher=Fernet(Fernet.generate_key())>>>info={..."cardnum":2211849528391929,..."exp":[2020,9],..."cv2":842,...}>>>r.set(..."user:1000",...cipher.encrypt(json.dumps(info).encode("utf-8"))...)>>>r.get("user:1000")b'gAAAAABcg8-LfQw9TeFZ1eXbi' # ... [truncated]>>>cipher.decrypt(r.get("user:1000"))b'{"cardnum": 2211849528391929, "exp": [2020, 9], "cv2": 842}'>>>json.loads(cipher.decrypt(r.get("user:1000"))){'cardnum': 2211849528391929, 'exp': [2020, 9], 'cv2': 842}
Becauseinfo
contains a value that is alist
, you’ll need to serialize this into a string that’s acceptable by Redis. (You could usejson
,yaml
, or any other serialization for this.) Next, you encrypt and decrypt that string using thecipher
object. You need to deserialize the decrypted bytes usingjson.loads()
so that you can get the result back into the type of your initial input, adict
.
Note:Fernet uses AES 128 encryption in CBC mode. See thecryptography
docs for an example of using AES 256. Whatever you choose to do, usecryptography
, notpycrypto
(imported asCrypto
), which is no longer actively maintained.
If security is paramount, encrypting strings before they make their way across a network connection is never a bad idea.
One last quick optimization is compression. If bandwidth is a concern or you’re cost-conscious, you can implement a lossless compression and decompression scheme when you send and receive data from Redis. Here’s an example using the bzip2 compression algorithm, which in this extreme case cuts down on the number of bytes sent across the connection by a factor of over 2,000:
1>>>importbz2 2 3>>>blob="i have a lot to talk about"*10000 4>>>len(blob.encode("utf-8")) 5260000 6 7>>># Set the compressed string as value 8>>>r.set("msg:500",bz2.compress(blob.encode("utf-8"))) 9>>>r.get("msg:500")10b'BZh91AY&SY\xdaM\x1eu\x01\x11o\x91\x80@\x002l\x87\' # ... [truncated]11>>>len(r.get("msg:500"))1212213>>>260_000/122# Magnitude of savings142131.14754098360661516>>># Get and decompress the value, then confirm it's equal to the original17>>>rblob=bz2.decompress(r.get("msg:500")).decode("utf-8")18>>>rblob==blob19True
The way that serialization, encryption, and compression are related here is that they all occur client-side. You do some operation on the original object on the client-side that ends up making more efficient use of Redis once you send the string over to the server. The inverse operation then happens again on the client side when you request whatever it was that you sent to the server in the first place.
It’s common for a client library such asredis-py
to follow aprotocol in how it is built. In this case,redis-py
implements theREdis Serialization Protocol, or RESP.
Part of fulfilling this protocol consists of converting some Python object in a raw bytestring, sending it to the Redis server, and parsing the response back into an intelligible Python object.
For example, the string response “OK” would come back as"+OK\r\n"
, while the integer response 1000 would come back as":1000\r\n"
. This can get more complex with other data types such asRESP arrays.
Aparser is a tool in the request-response cycle that interprets this raw response and crafts it into something recognizable to the client.redis-py
ships with its own parser class,PythonParser
, which does the parsing in pure Python. (See.read_response()
if you’re curious.)
However, there’s also a C library,Hiredis, that contains a fast parser that can offer significant speedups for some Redis commands such asLRANGE
. You can think of Hiredis as an optional accelerator that it doesn’t hurt to have around in niche cases.
All that you have to do to enableredis-py
to use the Hiredis parser is to install its Python bindings in the same environment asredis-py
:
$python-mpipinstallhiredis
What you’re actually installing here ishiredis-py
, which is a Python wrapper for a portion of thehiredis
C library.
The nice thing is that you don’t really need to callhiredis
yourself. Justpip install
it, and this will letredis-py
see that it’s available and use itsHiredisParser
instead ofPythonParser
.
Internally,redis-py
will attempt toimporthiredis
, and use aHiredisParser
class to match it, but will fall back to itsPythonParser
instead, which may be slower in some cases:
# redis/utils.pytry:importhiredisHIREDIS_AVAILABLE=TrueexceptImportError:HIREDIS_AVAILABLE=False# redis/connection.pyifHIREDIS_AVAILABLE:DefaultParser=HiredisParserelse:DefaultParser=PythonParser
While Redis itself is open-source and free, several managed services have sprung up that offer a data store with Redis as the core and some additional features built on top of the open-source Redis server:
Amazon ElastiCache for Redis: This is a web service that lets you host a Redis server in the cloud, which you can connect to from an Amazon EC2 instance. For full setup instructions, you can walk through Amazon’sElastiCache for Redis launch page.
Microsoft’s Azure Cache for Redis: This is another capable enterprise-grade service that lets you set up a customizable, secure Redis instance in the cloud.
The designs of the two have some commonalities. You typically specify a custom name for your cache, which is embedded as part of a DNS name, such asdemo.abcdef.xz.0009.use1.cache.amazonaws.com
(AWS) ordemo.redis.cache.windows.net
(Azure).
Once you’re set up, here are a few quick tips on how to connect.
From the command line, it’s largely the same as in our earlier examples, but you’ll need to specify a host with theh
flag rather than using the default localhost. ForAmazon AWS, execute the following from your instance shell:
$exportREDIS_ENDPOINT="demo.abcdef.xz.0009.use1.cache.amazonaws.com"$redis-cli-h$REDIS_ENDPOINT
ForMicrosoft Azure, you can use a similar call. Azure Cache for Redisuses SSL (port 6380) by default rather than port 6379, allowing for encrypted communication to and from Redis, which can’t be said of TCP. All that you’ll need to supply in addition is a non-default port and access key:
$exportREDIS_ENDPOINT="demo.redis.cache.windows.net"$redis-cli-h$REDIS_ENDPOINT-p6380-a<primary-access-key>
The-h
flag specifies a host, which as you’ve seen is127.0.0.1
(localhost) by default.
When you’re usingredis-py
in Python, it’s always a good idea to keep sensitive variables out of Python scripts themselves, and to be careful about what read and write permissions you afford those files. The Python version would look like this:
>>>importos>>>importredis>>># Specify a DNS endpoint instead of the default localhost>>>os.environ["REDIS_ENDPOINT"]'demo.abcdef.xz.0009.use1.cache.amazonaws.com'>>>r=redis.Redis(host=os.environ["REDIS_ENDPOINT"])
That’s all there is to it. Besides specifying a differenthost
, you can now call command-related methods such asr.get()
as normal.
Note: If you want to use solely the combination ofredis-py
and an AWS or Azure Redis instance, then you don’t really need to install and make Redis itself locally on your machine, since you don’t need eitherredis-cli
orredis-server
.
If you’re deploying a medium- to large-scale production application where Redis plays a key role, then going with AWS or Azure’s service solutions can be a scalable, cost-effective, and security-conscious way to operate.
That concludes our whirlwind tour of accessing Redis through Python, including installing and using the Redis REPL connected to a Redis server and usingredis-py
in real-life examples. Here’s some of what you learned:
redis-py
lets you do (almost) everything that you can do with the Redis CLI through an intuitive Python API.Redis has an extensive set of features, some of which we didn’t really get to cover here, includingserver-side Lua scripting,sharding, andmaster-slave replication. If you think that Redis is up your alley, then make sure to follow developments as it implements anupdated protocol, RESP3.
Here are some resources that you can check out to learn more.
Books:
Redis in use:
Other:
🐍 Python Tricks 💌
Get a short & sweetPython Trick delivered to your inbox every couple of days. No spam ever. Unsubscribe any time. Curated by the Real Python team.
AboutBrad Solomon
Brad is a software engineer and a member of the Real Python Tutorial Team.
» More about BradMasterReal-World Python Skills With Unlimited Access to Real Python
Join us and get access to thousands of tutorials, hands-on video courses, and a community of expert Pythonistas:
MasterReal-World Python Skills
With Unlimited Access to Real Python
Join us and get access to thousands of tutorials, hands-on video courses, and a community of expert Pythonistas:
What Do You Think?
What’s your #1 takeaway or favorite thing you learned? How are you going to put your newfound skills to use? Leave a comment below and let us know.
Commenting Tips: The most useful comments are those written with the goal of learning from or helping out other students.Get tips for asking good questions andget answers to common questions in our support portal.
Keep Learning
Already have an account?Sign-In
Almost there! Complete this form and click the button below to gain instant access:
"Python Tricks: The Book" – Free Sample Chapter (PDF)