circlecloud/cc-server-parent
1
0
Fork 0
Code Releases Activity

feat: optimiz db client

Browse Source
This commit is contained in:
MiaoWoo
2019-06-11 18:49:06 +08:00
parent 507fa20623
commit bae9995ba7
3 changed files with 43 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
packages/cc-server-db-mongo/src/client/index.ts
View File

@@ -1,11 +1,12 @@
import { Db, ObjectID } from 'mongodb';
import { Db, ObjectID, Collection } from 'mongodb';
import { MongoDBConnection } from './connection';
import { provide } from 'cc-server-ioc'
import { DBClient } from 'cc-server-db'
@provide(DBClient)
export class MongoDBClient<T = any> implements DBClient {
public db: Db;
private db: Db;
private collection: Collection;
constructor() {
MongoDBConnection.getConnection((connection) => {
@@ -17,39 +18,43 @@ export class MongoDBClient<T = any> implements DBClient {
return this.db;
}
public async find(collection: string, filter: object): Promise<T[]> {
return await this.db.collection(collection).find(filter).toArray();
public setTable(table: string): void {
this.collection = this.db.collection(table);
}
public async findOne(collection: string, filter: Object): Promise<T> {
let result = await this.db.collection(collection).find(filter).limit(1).toArray();
public async find(filter: object): Promise<T[]> {
return await this.collection.find(filter).toArray();
}
public async findOne(filter: Object): Promise<T> {
let result = await this.collection.find(filter).limit(1).toArray();
return result[0];
}
public async findOneById(collection: string, objectId: string): Promise<T> {
return await this.findOne(collection, { _id: new ObjectID(objectId) })
public async findOneById(objectId: string): Promise<T> {
return await this.findOne({ _id: new ObjectID(objectId) })
}
public async insertOne(collection: string, model: T): Promise<T> {
var insert = await this.db.collection(collection).insertOne(model);
public async insertOne(model: T): Promise<T> {
var insert = await this.collection.insertOne(model);
return insert.ops[0];
}
public async updateOne(collection: string, where: any, model: any): Promise<boolean> {
let result = await this.db.collection(collection).updateOne(where, { $set: model });
public async updateOne(where: any, model: any): Promise<boolean> {
let result = await this.collection.updateOne(where, { $set: model });
return result.result.ok == 1 && result.result.n > 0;
}
public async updateById(collection: string, objectId: string, model: any): Promise<boolean> {
return await this.updateOne(collection, { _id: new ObjectID(objectId) }, { $set: model })
public async updateById(objectId: string, model: any): Promise<boolean> {
return await this.updateOne({ _id: new ObjectID(objectId) }, { $set: model })
}
public async deleteOne(collection: string, where: any): Promise<boolean> {
let result = await this.db.collection(collection).deleteOne(where);
public async deleteOne(where: any): Promise<boolean> {
let result = await this.collection.deleteOne(where);
return result.result.ok === 1 && result.result.n > 0
}
public async deleteById(collection: string, objectId: string): Promise<boolean> {
return this.deleteOne(collection, { _id: new ObjectID(objectId) });
public async deleteById(objectId: string): Promise<boolean> {
return this.deleteOne({ _id: new ObjectID(objectId) });
}
}