im.doctor/src/server/models/search/object.searcher.js

"use strict";

let RedisModel = require('./../redis.model.js');
let ModelUtil = require('../../util/model.util');
let RedisClient = require('../../repository/redis/redis.client.js');
let SearchRepo = require('../../repository/mysql/search.repo');

let log = require('../../util/log.js');
let async = require("async");
let ObjectUtil = require("../../util/object.util.js");
let redis = RedisClient.redisClient().connection;

const REDIS_KEYS = require('../../include/commons').REDIS_KEYS;
const DB_TABLES = require('../../include/commons').DB_TABLES;

class ObjectSearcher extends RedisModel {
    constructor() {
        super();
    }

    search(userId, keyword, targetRole, entityType, excludeTopicEndedSessions, page, size) {
        let self = this;

        async.waterfall([
                // 获取会话ID列表，根据参数过滤应该议题已结束的会话
                function (callback) {
                    if (excludeTopicEndedSessions) {
                        SearchRepo.findTopicEndedSessionIdList(userId, function (err, sessionIdList) {
                            if (err) return callback(err, null);

                            callback(null, sessionIdList, keyword, targetRole, entityType);
                        })
                    } else {
                        redis.zrangeAsync(RedisModel.makeRedisKey(REDIS_KEYS.UserSessions, userId), 0, -1)
                            .then(function (sessionIdList) {
                                callback(null, sessionIdList, keyword, targetRole, entityType);
                            })
                            .catch(function (err) {
                                return callback(err, null);
                            });
                    }
                },
                // 搜索
                function (sessionIdList, keyword, targetRole, entityType, callback) {
                    if (targetRole !== 'doctor' && targetRole !== 'patient') {
                        ModelUtil.emitError(self.eventEmitter, "Unknown target role: " + targetRole);
                    }

                    let userTable = targetRole === 'doctor' ? DB_TABLES.Doctors : DB_TABLES.Patients;

                    if (entityType === 'all') {
                        SearchRepo.searchAll(sessionIdList, keyword, userTable, function (err, res) {
                            if (err) return callback(err, null);

                            callback(null, res);
                        });
                    } else if (entityType === 'user') {
                        SearchRepo.searchUser(sessionIdList, keyword, userTable, page, size, function (err, res) {
                            if (err) return callback(err, null);

                            res.forEach(function (user) {
                                if(!user.avatar) user.avatar = "";

                                user.birthdate = user.birthdate ? ObjectUtil.timestampToLong(user.birthdate) : "";
                            });

                            if (userTable == DB_TABLES.Patients) {
                                callback(null, {patients: res});
                            } else {
                                callback(null, {doctors: res});
                            }
                        });
                    } else if (entityType === 'session') {
                        SearchRepo.searchSessions(sessionIdList, keyword, userTable, page, size, function (err, res) {
                            if (err) return callback(err, null);

                            callback(null, {sessions: res});
                        });
                    } else if (entityType === 'message') {
                        SearchRepo.searchMessages(sessionIdList, keyword, page, size, function (err, res) {
                            if (err) return callback(err, null);

                            res.forEach(function (message) {
                                message.timestamp = ObjectUtil.timestampToLong(message.timestamp);
                            });

                            callback(null, {messages: res});
                        });
                    } else {
                        callback(new Error("Unknown entity type: " + entityType), null);
                    }
                }
            ],
            function (err, res) {
                if (err) {
                    return ModelUtil.emitError(self.eventEmitter, "Search failed", err);
                }

                ModelUtil.emitOK(self.eventEmitter, res);
            });
    }
}

module.exports = ObjectSearcher;