增加城市接口

controller/v1/cities.js

@@ -18,11 +18,16 @@ class CityHandle {
 		let cityInfo;
 		switch (type){
 			case 'guess': 
-				cityInfo = await Cities.cityGuess('SH');
-
-
+				cityInfo = await Cities.cityGuess('shanghai');
+				break;
+			case 'hot': 
+				cityInfo = await Cities.cityHot();
+				break;
+			case 'group': 
+				cityInfo = await Cities.cityGroup();
+				break;
 		}
-		res.json(cityInfo)
+		res.send(cityInfo)
 	}
 }
 

models/v1/cities.js

@@ -14,18 +14,50 @@ const citySchema = new mongoose.Schema({
 	pinyin: String,
 });
 
-citySchema.statics.cityGuess = async (name) => {
-	try{
-		let city = await this.find();
-		city.entries.forEach(item => {
-			
-		})
-	}catch(err){
-		console.error(err);
-	}
+citySchema.statics.cityGuess = function(name){
+	return new Promise(async (resolve, reject) => {
+		const firtWord = name.substr(0,1).toUpperCase();
+		try{
+			const city = await this.findOne();
+			Object.entries(city._doc).forEach(item => {
+				if(item[0] == firtWord){
+					item[1].forEach(cityItem => {
+						if (cityItem.pinyin == name) {
+							resolve(cityItem)
+						}
+					})
+				}
+			})
+		}catch(err){
+			console.error(err);
+		}
+	})
+}
 
+citySchema.statics.cityHot = function (){
+	return new Promise(async (resolve, reject) => {
+		try{
+			const city = await this.findOne();
+			resolve(city._doc.hotCities)
+		}catch(err){
+			console.error(err);
+		}
+	})
 }
 
+citySchema.statics.cityGroup = function (){
+	return new Promise(async (resolve, reject) => {
+		try{
+			const city = await this.findOne();
+			let cityObj = city._doc;
+			delete(cityObj._id)
+			delete(cityObj.hotCities)
+			resolve(cityObj)
+		}catch(err){
+			console.error(err);
+		}
+	})
+}
 
 const Cities = mongoose.model('Cities', citySchema);