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

Question

IT Interview Practice · 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.

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

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

Decorator có đối số

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

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

Câu hỏi liên quan