Có hai tùy chọn:
- Sử dụng
MERGE
câu lệnh thay vìINSERT ... ON CONFLICT
. - Sử dụng
UPDATE
câu lệnh vớiJOIN
, theo sau làINSERT
có điều kiện .
Tài liệu T-SQL cho MERGE cho biết:
Mẹo về hiệu suất:Hành vi có điều kiện được mô tả cho câu lệnh MERGE hoạt động tốt nhất khi hai bảng có một hỗn hợp phức tạp của các đặc điểm phù hợp. Ví dụ:chèn một hàng nếu nó không tồn tại hoặc cập nhật một hàng nếu nó khớp. Khi chỉ cần cập nhật một bảng dựa trên các hàng của bảng khác, hãy cải thiện hiệu suất và khả năng mở rộng bằng các câu lệnh INSERT, UPDATE và DELETE cơ bản.
Trong nhiều trường hợp, việc sử dụng UPDATE
riêng biệt sẽ nhanh hơn và ít phức tạp hơn và INSERT
tuyên bố.
engine = sa.create_engine(
connection_uri, fast_executemany=True, isolation_level="SERIALIZABLE"
)
with engine.begin() as conn:
# step 0.0 - create test environment
conn.execute(sa.text("DROP TABLE IF EXISTS main_table"))
conn.execute(
sa.text(
"CREATE TABLE main_table (id int primary key, txt varchar(50))"
)
)
conn.execute(
sa.text(
"INSERT INTO main_table (id, txt) VALUES (1, 'row 1 old text')"
)
)
# step 0.1 - create DataFrame to UPSERT
df = pd.DataFrame(
[(2, "new row 2 text"), (1, "row 1 new text")], columns=["id", "txt"]
)
# step 1 - upload DataFrame to temporary table
df.to_sql("#temp_table", conn, index=False, if_exists="replace")
# step 2 - merge temp_table into main_table
conn.execute(
sa.text("""\
UPDATE main SET main.txt = temp.txt
FROM main_table main INNER JOIN #temp_table temp
ON main.id = temp.id
"""
)
)
conn.execute(
sa.text("""\
INSERT INTO main_table (id, txt)
SELECT id, txt FROM #temp_table
WHERE id NOT IN (SELECT id FROM main_table)
"""
)
)
# step 3 - confirm results
result = conn.execute(sa.text("SELECT * FROM main_table ORDER BY id")).fetchall()
print(result) # [(1, 'row 1 new text'), (2, 'new row 2 text')]