Call api in python with multi-thread

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
import threading
import requests
import queue
import random

def check_result_is_ok(result):
    # có thể đọc kết quả từ result để kiểm tra, result1.get('attribute_sample')
    # trong ví dụ này dùng random
    return random.choice([True, False])


# Sử dụng queue.Queue để chia sẻ kết quả giữa các luồng
result_queue = queue.Queue()

def call_api1():
    # Gọi API đầu tiên
    response = requests.get('https://catfact.ninja/fact')
    # Xử lý kết quả API đầu tiên
    result = response.json()
    if check_result_is_ok(result) == True:
        # Nếu kết quả tốt, gửi kết quả vào hàng đợi và không cần chờ nữa
        result_queue.put(result)
    else:
        # Nếu kết quả không tốt, chờ kết quả của luồng 2 và không dừng luồng 2 nếu nó đang chạy
        result_queue.put(None)

def call_api2():
    # Gọi API thứ hai
    response = requests.get('https://cat-fact.herokuapp.com/facts')
    # Xử lý kết quả API thứ hai
    result = response.json()
    # Gửi kết quả vào hàng đợi và không cần chờ nữa
    result_queue.put(result)

def process_result():
    # Chờ cho tất cả các luồng kết thúc
    thread1.join()
    thread2.join()

    # Lấy kết quả từ hàng đợi
    result1 = result_queue.get()
    result2 = result_queue.get()

    # Kiểm tra kết quả từ luồng 1
    if result1 and check_result_is_ok(result1) == True:
        # Nếu kết quả tốt từ luồng 1, in kết quả và kết thúc luồng 2 (nếu đang chạy)
        print("Result from Thread 1:", result1)
        if thread2.is_alive():
            thread2.join()
        return result1
    else:
        # Nếu kết quả không tốt từ luồng 1, in kết quả từ luồng 2 và trả về kết quả của luồng 2
        print("Result from Thread 2:", result2)
        return result2

if __name__ == "__main__":
    # Tạo và bắt đầu luồng 1
    thread1 = threading.Thread(target=call_api1)
    thread1.start()

    # Tạo và bắt đầu luồng 2
    thread2 = threading.Thread(target=call_api2)
    thread2.start()

    # Chờ cho kết quả từ cả hai luồng
    thread1.join()
    thread2.join()

    # Xử lý kết quả từ cả hai luồng và trả về kết quả cho client
    final_result = process_result()
    print("Final Result:", final_result)