Decorator là gì và bạn viết một decorator như thế nào?

Question

Accepted Answer

Một **decorator** là một hàm **bao bọc một hàm khác** để thêm hành vi, mà không sửa đổi code của hàm gốc. Nó được áp dụng bằng cú pháp `@decorator` phía trên một hàm — một cách gọn gàng để thêm các mối quan tâm xuyên suốt (cross-cutting concern) như logging, đo thời gian, caching hoặc kiểm soát truy cập.

## Decorator hoạt động như thế nào

```python
def my_decorator(func):
    def wrapper(*args, **kwargs):       # bao bọc hàm gốc
        print("before")
        result = func(*args, **kwargs)  # gọi hàm gốc
        print("after")
        return result
    return wrapper                       # trả về phiên bản đã bao bọc

@my_decorator
def greet(name):
    print(f"Hello {name}")

greet("Ann")
# Output: before / Hello Ann / after
```

Cú pháp `@my_decorator` là đường cú pháp (sugar) cho `greet = my_decorator(greet)` — nó thay thế `greet` bằng `wrapper`, hàm này chạy code bổ sung xung quanh lời gọi gốc. Phần `*args, **kwargs` khiến wrapper hoạt động với bất kỳ chữ ký hàm nào.

## Một ví dụ thực tế: đo thời gian

```python
import time, functools

def timed(func):
    @functools.wraps(func)        # giữ lại tên/docstring của func (quan trọng!)
    def wrapper(*args, **kwargs):
        start = time.perf_counter()
        result = func(*args, **kwargs)
        print(f"{func.__name__} took {time.perf_counter() - start:.3f}s")
        return result
    return wrapper

@timed
def slow_task():
    time.sleep(1)
```

`@functools.wraps(func)` rất quan trọng — không có nó, hàm được bao bọc sẽ mất tên gốc và docstring.

## Decorator có đối số

```python
def repeat(times):               # lớp ngoài nhận đối số của decorator
    def decorator(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            for _ in range(times):
                func(*args, **kwargs)
        return wrapper
    return decorator

@repeat(times=3)
def hello(): print("hi")  # in "hi" ba lần
```

Một decorator có tham số cần thêm một lớp (một hàm trả về một decorator).

## Các decorator dựng sẵn / thư viện phổ biến

```text
@property            → biến một method thành một thuộc tính được tính toán
@staticmethod / @classmethod → các loại method
@functools.lru_cache → ghi nhớ (memoize) kết quả
@app.route("/")      → định tuyến của web framework (Flask)
@pytest.fixture      → fixture cho test
```

## Tại sao điều này quan trọng

Decorator là một tính năng mạnh mẽ, đặc trưng-Python để **thêm hành vi có thể tái sử dụng** (logging, đo thời gian, caching, xác thực, validation, định tuyến) vào các hàm một cách gọn gàng và khai báo — tách các mối quan tâm xuyên suốt khỏi logic cốt lõi.

Chúng có mặt khắp nơi trong Python thực tế: web framework (`@app.route`), caching (`@lru_cache`), class (`@property`) và testing.

Hiểu cách viết chúng (một hàm wrapper, `*args/**kwargs`, `functools.wraps`) và dạng có tham số cho phép bạn vừa dùng đúng các decorator của hệ sinh thái vừa xây dựng các abstraction tái sử dụng của riêng mình.