Cách sử dụng giao dịch cơ sở dữ liệu với MongoDB và Node.js

Sau khi thiết lập dự án Node.js và MongoDB đang hoạt động, bạn có thể thiết lập kết nối với cơ sở dữ liệu Mongo trong Node.js. Nếu bạn chưa có trước đây, hãy cài đặt mongoose bằng cách chạy npm cài đặt mongoose trong thiết bị đầu cuối của bạn.

import mongoose from 'mongoose'
let MONGO_URL = process.env.MONGO_URL || 'your-mongo-database-url';
let connection;
const connectDb = async () => {
try {
await mongoose.connect(MONGO_URL, {
useNewUrlParser: true,
useUnifiedTopology: true,
});
console.log("CONNECTED TO DATABASE");
connection = mongoose.connection;
} catch (err) {
console.error("DATABASE CONNECTION FAILED!");
console.error(err.message);
process.exit(1); 
}
};

Bạn nên lưu trữ kết nối trong một biến để bạn có thể sử dụng nó để bắt đầu một giao dịch sau này trong chương trình.

Bạn có thể triển khai bộ sưu tập người dùng và công việc như vậy:

const userSchema = new mongoose.Schema({
name: String,
email: String,
jobs: [mongoose.Schema.Types.ObjectId]
});
const jobSchema = new mongoose.Schema({
title: String,
location: String,
salary: String,
poster: mongoose.Schema.Types.ObjectId
});
const userCollection = mongoose.model('user', userSchema);
const jobCollection = mongoose.model('job', jobSchema);

Bạn có thể viết một hàm để thêm người dùng vào cơ sở dữ liệu như sau:

const createUser = async (user) => {
const newUser = await userCollection.create(user);
console.log("User added to database");
console.log(newUser);
}

Đoạn mã dưới đây trình bày chức năng tạo một công việc và thêm nó vào danh sách công việc của người đăng nó bằng cách sử dụng một giao dịch cơ sở dữ liệu.

const createJob = async (job) => {
const { userEmail, title, location, salary } = job;

const user = await userCollection.findOne({ email: userEmail });

const session = await connection.startSession();

try {
await session.startTransaction();

const newJob = await jobCollection.create(
[
{
title,
location,
salary,
poster: user._id,
},
],
{ session }
);
console.log("Created new job successfully!");
console.log(newJob[0]);

const newJobId = newJob[0]._id;
const addedToUser = await userCollection.findByIdAndUpdate(
user._id,
{ $addToSet: { jobs: newJobId } },
{ session }
);
console.log("Successfully added job to user's jobs list");
console.log(addedToUser);
await session.commitTransaction();
console.log("Successfully carried out DB transaction");
} catch (e) {
console.error(e);
console.log("Failed to complete database operations");
await session.abortTransaction();
} finally {
await session.endSession();
console.log("Ended transaction session");
}
};

Dưới đây là minh họa về cách hoạt động của các hàm trên:

const mockUser = {
name: "Timmy Omolana",
email: "jobposter@example.com",
};
const mockJob = {
title: "Sales Manager",
location: "Lagos, Nigeria",
salary: "$40,000",
userEmail: "jobposter@example.com", 
};
const startServer = async () => {
await connectDb();
await createUser(mockUser);
await createJob(mockJob);
};
startServer()
.then()
.catch((err) => console.log(err));

Nếu bạn lưu mã này và chạy nó bằng cách sử dụng npm bắt đầu hoặc là nút nó sẽ tạo ra một đầu ra như thế này:

Một cách khác để thực hiện các giao dịch ACID trong MongoDB bằng cách sử dụng Mongoose là bằng cách sử dụng withTransaction () hàm số. Cách tiếp cận này cung cấp ít tính linh hoạt vì nó chạy tất cả các truy vấn bên trong một hàm gọi lại mà bạn chuyển làm đối số cho hàm.

Bạn có thể cấu trúc lại giao dịch cơ sở dữ liệu ở trên để sử dụng withTransaction () như thế này:

const createJob = async (job) => {
const { userEmail, title, location, salary } = job;

const user = await userCollection.findOne({ email: userEmail });

const session = await connection.startSession();

try {
const transactionSuccess = await session.withTransaction(async () => {
const newJob = await jobCollection.create(
[
{
title,
location,
salary,
poster: user._id,
},
],
{ session }
);
console.log("Created new job successfully!");
console.log(newJob[0]);

const newJobId = newJob[0]._id;
const addedToUser = await userCollection.findByIdAndUpdate(
user._id,
{ $addToSet: { jobs: newJobId } },
{ session }
);
console.log("Successfully added job to user's jobs list");
console.log(addedToUser);
});
if (transactionSuccess) {
console.log("Successfully carried out DB transaction");
} else {
console.log("Transaction failed");
}
} catch (e) {
console.error(e);
console.log("Failed to complete database operations");
} finally {
await session.endSession();
console.log("Ended transaction session");
}
};

Điều này sẽ tạo ra kết quả tương tự như việc triển khai trước đó. Bạn có thể tự do chọn kiểu nào sẽ sử dụng khi thực hiện các giao dịch cơ sở dữ liệu trong MongoDB.

Thực hiện các giao dịch cơ sở dữ liệu ACID trong MongoDB

Các giao dịch cơ sở dữ liệu rất dễ sử dụng khi được thực hiện một cách chính xác. Bây giờ bạn sẽ hiểu cách các giao dịch Cơ sở dữ liệu hoạt động trong MongoDB và cách bạn có thể triển khai chúng trong các ứng dụng Node.js.

Để khám phá thêm ý tưởng về các giao dịch ACID và cách chúng hoạt động trong MongoDB, hãy xem xét việc xây dựng ví fintech hoặc ứng dụng viết blog.