Method handle_stash_request

gunicorn/dirty/arbiter.py:791–906

Handle a stash operation directly in the arbiter. All stash tables are stored in arbiter memory for simplicity and fast access. Args: message: Stash operation message client_writer: StreamWriter to send response to client

(self, message, client_writer)

            await DirtyProtocol.write_message_async(client_writer, response)

    async def handle_stash_request(self, message, client_writer):
        """
        Handle a stash operation directly in the arbiter.

        All stash tables are stored in arbiter memory for simplicity
        and fast access.

        Args:
            message: Stash operation message
            client_writer: StreamWriter to send response to client
        """
        request_id = message.get("id", "unknown")
        op = message.get("op")
        table = message.get("table", "")
        key = message.get("key")
        value = message.get("value")
        pattern = message.get("pattern")

        try:
            result = None

            if op == STASH_OP_PUT:
                # Auto-create table if needed
                if table not in self.stash_tables:
                    self.stash_tables[table] = {}
                self.stash_tables[table][key] = value
                result = True

            elif op == STASH_OP_GET:
                if table not in self.stash_tables:
                    result = {"error": "key_not_found"}
                elif key not in self.stash_tables[table]:
                    result = {"error": "key_not_found"}
                else:
                    result = self.stash_tables[table][key]

            elif op == STASH_OP_DELETE:
                if table in self.stash_tables and key in self.stash_tables[table]:
                    del self.stash_tables[table][key]
                    result = True
                else:
                    result = False

            elif op == STASH_OP_KEYS:
                if table not in self.stash_tables:
                    result = []
                else:
                    all_keys = list(self.stash_tables[table].keys())
                    if pattern:
                        all_keys = [k for k in all_keys
                                    if fnmatch.fnmatch(str(k), pattern)]
                    result = all_keys

            elif op == STASH_OP_CLEAR:
                if table in self.stash_tables:
                    self.stash_tables[table].clear()
                result = True