82 lines
4.7 KiB
ReStructuredText
82 lines
4.7 KiB
ReStructuredText
mindspore.mint.distributed.TCPStore
|
||
=====================================
|
||
|
||
.. py:class:: mindspore.mint.distributed.TCPStore(host_name, port, world_size=None, is_master=False, timeout=timedelta(seconds=300), wait_for_workers=True, multi_tenant=False, master_listen_fd=None, use_libuv=True)
|
||
|
||
一种基于传输控制协议（TCP）的分布式键值存储实现方法。
|
||
|
||
.. note::
|
||
- 该功能通过CPU实现，不涉及任何与Ascend相关的硬件操作。
|
||
- 目前TCPStore类构造函数除了 `host_name` 、 `port` 、 `world_size` 、 `is_master` 、 `timeout` 和 `wait_for_workers` ，其他提供的所有参数均不支持，为预留参数，设置无效。
|
||
- 当前TCPStore功能受限，仅支持key小于4k、value小于1G场景，复杂场景待支持。
|
||
- 当前仅支持PyNative模式，不支持Graph模式。
|
||
|
||
参数：
|
||
- **host_name** (str) - 服务器存储应运行的主机名或 IP 地址。当前仅支持用户传入 IP 地址。
|
||
- **port** (int) - 服务器存储应侦听传入请求的端口。
|
||
- **world_size** (int，可选) - 存储用户总数（客户端数量 + 服务器数量 ``1`` ）。默认值为 ``None`` ，表示存储用户数量不固定。该参数只对服务器有效。
|
||
- **is_master** (bool，可选) - 初始化服务器存储时为 ``True`` ，初始化客户端存储时为 ``False`` 。默认值为 ``False`` 。
|
||
- **timeout** (timedelta，可选) - 初始化期间存储使用的超时时长，默认值为 ``timedelta(seconds=300)`` 。
|
||
- **wait_for_workers** (bool，可选) - 是否等待所有工作进程连接到服务器存储。这仅在 `world_size` 为固定值时适用。默认值为 ``True`` 。该参数只对服务器有效。
|
||
- **multi_tenant** (bool，无效参数，可选) - 如果为 True，则当前进程中具有相同主机/端口的所有 ``TCPStore`` 实例将使用相同的底层 ``TCPServer`` 。默认值为 ``False`` 。
|
||
- **master_listen_fd** (int，无效参数，可选) - 如果指定，则底层 ``TCPServer`` 将侦听此文件描述符，该文件描述符必须是已绑定到端口的套接字。在某些情况下有助于避免端口分配竞争。默认值为 ``None`` （意味着服务器创建一个新的套接字并尝试将其绑定到端口）。
|
||
- **use_libuv** (bool，无效参数，可选) - 如果为 ``True`` ，则将 libuv 用于 ``TCPServer`` 后端。默认值为 ``True`` 。
|
||
|
||
返回：
|
||
``TCPStore`` 对象。
|
||
|
||
.. py:method:: add(key, amount)
|
||
|
||
第一次调用 `add` 函数时，给定一个键值，会在存储中创建一个带有该键值对应的计数器，计数器的初始值为 `amount` 。后续使用相同键调用 `add` 函数会进行增量操作按数量进行计数。
|
||
|
||
参数：
|
||
- **key** (str) - 计数器值将被增加的键。
|
||
- **amount** (int) - 计数器将增加的数值。
|
||
|
||
返回：
|
||
int，返回 `key` 对应的计数器的值。
|
||
|
||
异常：
|
||
- **TypeError** - 当 `key` 不是字符串。
|
||
- **TypeError** - 当 `amount` 不是整数。
|
||
- **RuntimeError** - 如果 `add` 和 `set` 传入相同的 `key` ，且 `set` 传入的 `value` 无法正确转换成数值，此时调用 `add` 会报错。
|
||
|
||
.. py:method:: delete_key(key)
|
||
|
||
从存储中删除与 `key` 关联的键值对。
|
||
|
||
参数：
|
||
- **key** (str) - 要从存储中删除的 `key` 。
|
||
|
||
返回：
|
||
bool，如果成功删除了该键，则返回 ``True`` ，否则返回 ``False`` 。
|
||
|
||
异常：
|
||
- **TypeError** - 当 `key` 不是字符串。
|
||
|
||
.. py:method:: get(key)
|
||
|
||
检索与存储中给定 `key` 关联的值。如果存储中不存在 `key` ，此函数将等待类初始化设置的超时时间 `timeout` ，然后抛出异常。
|
||
|
||
参数：
|
||
- **key** (str) - 该函数将返回与此键关联的值。
|
||
|
||
返回：
|
||
bytes，如果 `key` 在存储中，则返回与 `key` 关联的值。
|
||
|
||
异常：
|
||
- **TypeError** - 当 `key` 不是字符串。
|
||
- **RuntimeError** - 当 `get` 执行超时。
|
||
|
||
.. py:method:: set(key, value)
|
||
|
||
根据提供的 `key` 和 `value` 将键值对插入到存储中。如果存储中已经存在 `key` ，它将使用新提供的 `value` 覆盖旧值。
|
||
|
||
参数：
|
||
- **key** (str) - 要添加到存储中的键。
|
||
- **value** (Union[bytes, str]) - 要添加到存储中的键值对的value。
|
||
|
||
异常：
|
||
- **TypeError** - 当 `key` 不是字符串。
|
||
- **TypeError** - 当 `value` 不是字符串或者字节类型。
|