一日一技:如何通过迭代器精简你的代码

本文涉及的产品
云数据库 Redis 版,社区版 2GB
推荐场景:
搭建游戏排行榜
简介: 一日一技:如何通过迭代器精简你的代码

截图:产品经理游玩:kingname & 产品经理

假设有一个 Redis 集合,里面有 N 条数据,你不停从里面lpop数据,直到某一条数据的值为'Stop'字符串为止(已知里面必有一条数据为'Stop'字符串,但其位置不知道)。

这个需求看起来很简单,于是你立刻就着手写出了代码:

import redis
client = redis.Redis()
def read_data():
    datas = []
    whileTrue:
        data = client.lpop().decode()
        if data == 'Stop':
            break
        datas.append(data)
    return datas

现在问题来了,如果 Redis 里面的数据非常多,已经超过了你的内存容量怎么办?数据全部放在datas列表里面再返回显然是不可取的做法。

好在,这些数据读取出来以后,会传给一个parse函数,并且这个函数是一条一条处理数据的,它处理完成以后,就可以把数据丢弃了。

于是你可能会这样改写代码:

import redis
client = redis.Redis()
def read_data():
    whileTrue:
        data = client.lpop().decode()
        if data == 'Stop':
            break
        parse(data)

但我们知道,在编码规范和软件工程里面,建议一个函数,它应该只做一件事情,而现在read_data()函数却做了两件事情:1. 从 Redis 里面读取数据。2.调用parse()函数。

那么我们有没有办法把他们区分开来呢?如何让read_data能返回数据,但是又不会把内存撑爆呢?

这个时候,我们就可以使用生成器来解决问题:

import redis
client = redis.Redis()
def read_data():
    whileTrue:
        data = client.lpop().decode()
        if data == 'Stop':
            break
        yield data
def parse_data():
    for data in read_data():
        parse(data)

在这个代码里面,read_data变成了生成器函数,它返回一个生成器,对生成器进行迭代的时候,每次返回一条数据,这一条数据立即传给parse()函数。整个过程源源不断,生生不息。不需要额外创建一个列表用来存放数据。

那么代码还能不能继续简化呢?此时我们就可以使用iter关键字了。

使用了iter关键字的效果如下图所示:

import redis
client = redis.Redis()
def read_data():
    data = client.lpop().decode()
    return data
def parse_data():
    for data in iter(read_data, 'Stop'):
        parse(data)

其中,read_data现在每运行一次只会返回列表最左边的数据。但是当我们直接使用iter(read_data, 'Stop')的时候,就会得到一个迭代器。对这个迭代器进行迭代,相当于在while True里面不停运行read_data函数,直到某一次迭代的时候,read_data函数返回了Stop,就停止。

如果你想炫技的话,还可以进一步简化:

import redis
client = redis.Redis()
def parse_data():
    for data in iter(lambda: client.lpop().decode(), 'Stop'):
        parse(data)

当然,我是不推荐你这样写的^_^。

相关实践学习
基于Redis实现在线游戏积分排行榜
本场景将介绍如何基于Redis数据库实现在线游戏中的游戏玩家积分排行榜功能。
云数据库 Redis 版使用教程
云数据库Redis版是兼容Redis协议标准的、提供持久化的内存数据库服务,基于高可靠双机热备架构及可无缝扩展的集群架构,满足高读写性能场景及容量需弹性变配的业务需求。 产品详情:https://www.aliyun.com/product/kvstore     ------------------------------------------------------------------------- 阿里云数据库体验:数据库上云实战 开发者云会免费提供一台带自建MySQL的源数据库 ECS 实例和一台目标数据库 RDS实例。跟着指引,您可以一步步实现将ECS自建数据库迁移到目标数据库RDS。 点击下方链接,领取免费ECS&RDS资源,30分钟完成数据库上云实战!https://developer.aliyun.com/adc/scenario/51eefbd1894e42f6bb9acacadd3f9121?spm=a2c6h.13788135.J_3257954370.9.4ba85f24utseFl
目录
相关文章
|
2月前
|
JSON 算法 前端开发
2705. 精简对象
2705. 精简对象
19 0
|
3月前
|
Python
Python中的生成器函数与迭代器——提升代码效率的利器
生成器函数和迭代器是Python中强大的编程工具,它们能够提升代码的效率和可读性。本文将介绍生成器函数和迭代器的概念,并通过示例代码展示它们在实际开发中的应用,帮助读者更好地理解和运用这些技术。
26 1
|
8月前
|
设计模式 Java 数据库连接
手写自定义迭代器,秒懂迭代器底层原理
迭代器模式的UML类图如下图所示。
47 0
|
29天前
|
算法 程序员 C语言
【C++ 迭代器实现细节 】深入探索C++迭代器:从实现到整合
【C++ 迭代器实现细节 】深入探索C++迭代器:从实现到整合
78 0
|
1月前
|
存储 算法 安全
深入理解C++迭代器:让你的C++代码更加灵活
深入理解C++迭代器:让你的C++代码更加灵活
69 0
|
1月前
|
存储 C++
C++STL模板之——list(简化源码,模拟源码)
C++STL模板之——list(简化源码,模拟源码)
|
8月前
|
算法 测试技术 C++
10.1 C++ STL 模板适配与迭代器
STL(Standard Template Library)标准模板库提供了模板适配器和迭代器等重要概念,为开发者提供了高效、灵活和方便的编程工具。模板适配器是指一组模板类或函数,它们提供一种适配机制,使得现有的模板能够适应新的需求。而迭代器则是STL中的令一种重要的概念,它是一个抽象化的数据访问机制,通过迭代器可以遍历STL容器中的元素。适配器与迭代器两者的紧密配合,使得开发者能够高效地处理容器中的元素,提高了代码的复用性和可维护性。
35 0
|
8月前
JDK1.8新特性Lambda表达式简化if-else里都有for循环的优化方式
JDK1.8新特性Lambda表达式简化if-else里都有for循环的优化方式
52 0
|
6月前
|
JavaScript 前端开发
带你读《现代Javascript高级教程》十五、Iterator 迭代器:简化集合遍历的利器(2)
带你读《现代Javascript高级教程》十五、Iterator 迭代器:简化集合遍历的利器(2)
|
6月前
|
JavaScript 前端开发
带你读《现代Javascript高级教程》十五、Iterator 迭代器:简化集合遍历的利器(1)
带你读《现代Javascript高级教程》十五、Iterator 迭代器:简化集合遍历的利器(1)