快捷導(dǎo)航

使用Python操作Redis所有數(shù)據(jù)類型的方法

更新時(shí)間：2023年11月05日 16:07:29 作者：pycode

當(dāng)今互聯(lián)網(wǎng)時(shí)代,數(shù)據(jù)處理已經(jīng)成為了一個(gè)非常重要的任務(wù),而Redis作為一款高性能的NoSQL數(shù)據(jù)庫(kù),越來(lái)越受到了廣大開(kāi)發(fā)者的喜愛(ài),本篇博客將介紹如何使用Python操作Redis的所有類型,以及一些高級(jí)用法,需要的朋友可以參考下

Redis的數(shù)據(jù)類型

Redis支持五種不同的數(shù)據(jù)類型，分別是：

字符串（String）
哈希（Hash）
列表（List）
集合（Set）
有序集合（Sorted Set）

下面我們將詳細(xì)介紹如何在Python中操作這五種數(shù)據(jù)類型。

字符串（String）

字符串是Redis中最基本的數(shù)據(jù)類型。在Python中，我們可以使用redis-py庫(kù)來(lái)操作字符串。

import redis

r = redis.Redis(host='localhost', port=6379, db=0)

# 設(shè)置字符串
r.set('name', 'Tom')

# 獲取字符串
print(r.get('name'))

輸出：

b'Tom'

在Redis中，字符串可以是任何形式的數(shù)據(jù)，比如數(shù)字、布爾值、甚至是序列化后的對(duì)象。而在Python中，我們可以使用pickle庫(kù)來(lái)序列化對(duì)象。

import pickle

class User:
    def __init__(self, name, age):
        self.name = name
        self.age = age

user = User('Tom', 18)
r.set('user', pickle.dumps(user))

print(pickle.loads(r.get('user')))

輸出：

<__main__.User object at 0x0000025E5C5D5C10>

哈希（Hash）

哈希是Redis中的一種鍵值對(duì)存儲(chǔ)結(jié)構(gòu)，可以看做是一張表。在Python中，我們同樣可以使用redis-py庫(kù)來(lái)操作哈希。

import redis

r = redis.Redis(host='localhost', port=6379, db=0)

# 設(shè)置哈希
r.hset('user:1', 'name', 'Tom')
r.hset('user:1', 'age', 18)

# 獲取哈希
print(r.hgetall('user:1'))

輸出：

{b'name': b'Tom', b'age': b'18'}

在Redis中，哈?？梢杂脕?lái)存儲(chǔ)對(duì)象，每個(gè)屬性對(duì)應(yīng)一個(gè)鍵值對(duì)。

import pickle

class User:
    def __init__(self, name, age):
        self.name = name
        self.age = age

user = User('Tom', 18)

r.hset('user:1', 'data', pickle.dumps(user))

data = pickle.loads(r.hget('user:1', 'data'))
print(data.name, data.age)

輸出：

Tom 18

列表（List）

列表是Redis中的一種有序集合，可以用來(lái)存儲(chǔ)一組有序的數(shù)據(jù)。在Python中，我們同樣可以使用redis-py庫(kù)來(lái)操作列表。

import redis

r = redis.Redis(host='localhost', port=6379, db=0)

# 設(shè)置列表
r.lpush('mylist', 'a')
r.lpush('mylist', 'b')
r.lpush('mylist', 'c')

# 獲取列表
print(r.lrange('mylist', 0, -1))

輸出：

[b'c', b'b', b'a']

在Redis中，列表可以用來(lái)實(shí)現(xiàn)簡(jiǎn)單的消息隊(duì)列，比如生產(chǎn)者消費(fèi)者模式。

import time
import threading

def producer():
    for i in range(10):
        r.lpush('queue', str(i))
        time.sleep(0.1)

def consumer():
    while True:
        item = r.brpop('queue', timeout=1)
        if item:
            print(item[1])

t1 = threading.Thread(target=producer)
t2 = threading.Thread(target=consumer)

t1.start()
t2.start()

t1.join()
t2.join()

輸出：

b'0'
b'1'
b'2'
b'3'
b'4'
b'5'
b'6'
b'7'
b'8'
b'9'

集合（Set）

集合是Redis中的一種無(wú)序集合，可以用來(lái)存儲(chǔ)一組唯一的數(shù)據(jù)。在Python中，我們同樣可以使用redis-py庫(kù)來(lái)操作集合。

import redis

r = redis.Redis(host='localhost', port=6379, db=0)

# 設(shè)置集合
r.sadd('myset', 'a')
r.sadd('myset', 'b')
r.sadd('myset', 'c')

# 獲取集合
print(r.smembers('myset'))

輸出：

{b'c', b'b', b'a'}

在Redis中，集合可以用來(lái)實(shí)現(xiàn)簡(jiǎn)單的去重操作，也可以用來(lái)實(shí)現(xiàn)交集、并集、差集等操作。

r.sadd('set1', 1, 2, 3, 4)
r.sadd('set2', 3, 4, 5, 6)

# 求交集
print(r.sinter('set1', 'set2'))

# 求并集
print(r.sunion('set1', 'set2'))

# 求差集
print(r.sdiff('set1', 'set2'))

輸出：

{b'3', b'4'}
{b'1', b'2', b'3', b'4', b'5', b'6'}
{b'1', b'2'}

有序集合（Sorted Set）

有序集合是Redis中的一種有序集合，可以用來(lái)存儲(chǔ)一組有序的數(shù)據(jù)，并且每個(gè)數(shù)據(jù)都有一個(gè)分?jǐn)?shù)。在Python中，我們同樣可以使用redis-py庫(kù)來(lái)操作有序集合。

import redis

r = redis.Redis(host='localhost', port=6379, db=0)

# 設(shè)置有序集合
r.zadd('myzset', {'a': 1, 'b': 2, 'c': 3})

# 獲取有序集合
print(r.zrange('myzset', 0, -1, withscores=True))

輸出：

[(b'a', 1.0), (b'b', 2.0), (b'c', 3.0)]

在Redis中，有序集合可以用來(lái)實(shí)現(xiàn)排行榜等功能。

r.zadd('rank', {'Tom': 100, 'Jerry': 200, 'Alice': 300})

# 獲取排行榜前三名
print(r.zrevrange('rank', 0, 2, withscores=True))

輸出：

[(b'Alice', 300.0), (b'Jerry', 200.0), (b'Tom', 100.0)]

Redis的高級(jí)用法

除了以上介紹的五種數(shù)據(jù)類型，Redis還有一些高級(jí)用法，比如事務(wù)、發(fā)布訂閱等。下面我們將介紹其中的一些用法。

事務(wù)

Redis的事務(wù)是通過(guò)MULTI、EXEC、WATCH等命令實(shí)現(xiàn)的。在Python中，我們同樣可以使用redis-py庫(kù)來(lái)執(zhí)行事務(wù)。

import redis

r = redis.Redis(host='localhost', port=6379, db=0)

with r.pipeline() as pipe:
    while True:
        try:
            pipe.watch('balance')
            balance = int(pipe.get('balance'))
            if balance < 10:
                raise Exception('Not enough balance')
            pipe.multi()
            pipe.decrby('balance', 10)
            pipe.incr('income', 10)
            pipe.execute()
            break
        except Exception as e:
            print(e)
            break

以上代碼中，我們使用了一個(gè)簡(jiǎn)單的轉(zhuǎn)賬例子來(lái)演示事務(wù)的使用。在事務(wù)中，我們首先使用WATCH命令監(jiān)視了balance鍵，然后獲取了其當(dāng)前的值。如果余額不足，則拋出異常，事務(wù)結(jié)束。否則，我們使用MULTI命令開(kāi)始事務(wù)，執(zhí)行減少余額和增加收入的操作，最后使用EXEC命令提交事務(wù)。

發(fā)布訂閱

Redis的發(fā)布訂閱是通過(guò)PUBLISH、SUBSCRIBE、UNSUBSCRIBE等命令實(shí)現(xiàn)的。在Python中，我們同樣可以使用redis-py庫(kù)來(lái)實(shí)現(xiàn)發(fā)布訂閱功能。

import redis
import threading

r = redis.Redis(host='localhost', port=6379, db=0)

def publisher():
    for i in range(10):
        r.publish('channel', i)
        time.sleep(0.1)

def subscriber():
    pubsub = r.pubsub()
    pubsub.subscribe('channel')
    for item in pubsub.listen():
        print(item['data'])

t1 = threading.Thread(target=publisher)
t2 = threading.Thread(target=subscriber)

t1.start()
t2.start()

t1.join()
t2.join()

以上代碼中，我們使用了一個(gè)簡(jiǎn)單的發(fā)布訂閱例子來(lái)演示發(fā)布訂閱的使用。在這個(gè)例子中，我們使用了兩個(gè)線程，一個(gè)線程用來(lái)發(fā)布消息，一個(gè)線程用來(lái)訂閱消息。在訂閱線程中，我們使用pubsub對(duì)象來(lái)訂閱channel頻道，并通過(guò)listen方法來(lái)獲取消息。