修改insert为upset

src/main/java/com/zhiwei/manage/dao/MessageDaoImpl.java

@@ -69,7 +69,7 @@ public class MessageDaoImpl implements MessageDao {
 				e.printStackTrace();
 			}
 		}
-		WriteResult result = mongo.updateFirst(Query.query(Criteria.where("_id").is(message.getMessageId())), update,
+		WriteResult result = mongo.upsert(Query.query(Criteria.where("_id").is(message.getMessageId())), update,
 				Message.class);
 		if (result.getN() == 1) {
 			return "修改成功";
@@ -80,11 +80,9 @@ public class MessageDaoImpl implements MessageDao {
 
 	@Override
 	public List<Message> findAll(int pageNo, int pageSize, String pt) {
-		return mongo.find(
-				Query.query(Criteria.where("pt").is(pt).andOperator(Criteria.where("handle").is(false)))
+		return mongo.find(Query.query(Criteria.where("pt").is(pt).andOperator(Criteria.where("handle").is(false)))
 				.skip((pageNo - 1) * pageSize).with(new Sort(Sort.Direction.ASC, "templateLv"))
-						.with(new Sort(Sort.Direction.DESC, "createDate")).limit(pageSize),
-				Message.class);
+				.with(new Sort(Sort.Direction.DESC, "createDate")).limit(pageSize), Message.class);
 	}
 
 	@Override
@@ -118,11 +116,12 @@ public class MessageDaoImpl implements MessageDao {
 
 	@Override
 	public List<Message> findByHandle(int pageNo, int pageSize, boolean handle, String tempName, String pt) {
-		return mongo.find(Query
-				.query(Criteria.where("tempName").regex(tempName)
+		return mongo.find(
+				Query.query(Criteria.where("tempName").regex(tempName)
 						.andOperator(Criteria.where("handle").is(handle).andOperator(Criteria.where("pt").is(pt))))
 						.skip((pageNo - 1) * pageSize).limit(pageSize)
-				.with(new Sort(Sort.Direction.DESC, "handleDate")), Message.class);
+						.with(new Sort(Sort.Direction.DESC, "handleDate")),
+				Message.class);
 	}
 
 	@Override
@@ -134,8 +133,10 @@ public class MessageDaoImpl implements MessageDao {
 	}
 
 	@Override
-	public Message findMsg(String tempName,String errorType) {
-		return mongo.findOne(Query.query(Criteria.where("tempName").is(tempName)
-				.andOperator(Criteria.where("errotType").is(errorType))), Message.class);
+	public Message findMsg(String tempName, String errorType) {
+		return mongo.findOne(
+				Query.query(
+						Criteria.where("tempName").is(tempName).andOperator(Criteria.where("errotType").is(errorType))),
+				Message.class);
 	}
 }

src/main/java/com/zhiwei/manage/handle/Task.java

@@ -58,7 +58,8 @@ public class Task implements Runnable {
 						msg.setTempName(ping.getTempName());
 						msg.setTemplateLv(tp.getTemplateLv());
 						msg.setErrorType("url");
-						messageService.insert(msg);
+						msg.setPt(ping.getPt());
+						messageService.update(msg,ping.getPt());
 					} else {
 						Message msg = messageService.findMsg(ping.getTempName(), "url");
 						if (msg != null) {
@@ -78,7 +79,7 @@ public class Task implements Runnable {
 					msg.setTemplateLv(1);
 					msg.setErrorType("url");
 					msg.setPt(ping.getPt());
-					messageService.insert(msg);
+					messageService.update(msg,ping.getPt());
 				}
 				if (null != ping.getPingUrlTwo() && !ping.getPingUrlTwo().equals(""))
 					try {
@@ -99,7 +100,7 @@ public class Task implements Runnable {
 							msg.setTemplateLv(tp.getTemplateLv());
 							msg.setErrorType("url");
 							msg.setPt(ping.getPt());
-							messageService.insert(msg);
+							messageService.update(msg,ping.getPt());
 						} else {
 							Message msg = messageService.findMsg(ping.getTempName(), "url");
 							if (msg != null) {
@@ -119,7 +120,7 @@ public class Task implements Runnable {
 						msg.setPt(ping.getPt());
 						msg.setErrorType("url");
 						msg.setTemplateLv(tp.getTemplateLv());
-						messageService.insert(msg);
+						messageService.update(msg,ping.getPt());
 					}
 			}
 		}
@@ -139,7 +140,7 @@ public class Task implements Runnable {
 					msg.setTemplateLv(tp.getTemplateLv());
 					msg.setErrorType("count2");
 					msg.setTempName(field.getTempName());
-					messageService.insert(msg);
+					messageService.update(msg,field.getPt());
 				} else {
 					Message msg = messageService.findMsg(field.getTempName(), "count2");
 					if (msg != null) {
@@ -173,7 +174,7 @@ public class Task implements Runnable {
 						msg.setTemplateCreator(tp.getCreator());
 						msg.setTemplateLv(tp.getTemplateLv());
 						msg.setTempName(field.getTempName());
-						messageService.insert(msg);
+						messageService.update(msg,field.getPt());
 					} else {
 						Message msg = messageService.findMsg(field.getTempName(), "field");
 						if (msg != null) {
@@ -233,7 +234,7 @@ public class Task implements Runnable {
 						msg.setTemplateCreator(tp.getCreator());
 						msg.setTemplateLv(tp.getTemplateLv());
 						msg.setTempName(delayed.getTempName());
-						messageService.insert(msg);
+						messageService.update(msg,delayed.getPt());
 					}
 				}
 				Collections.sort(dNum);
@@ -253,7 +254,7 @@ public class Task implements Runnable {
 					msg.setTemplateLv(tp.getTemplateLv());
 					msg.setPt(delayed.getPt());
 					msg.setTempName(delayed.getTempName());
-					messageService.insert(msg);
+					messageService.update(msg,delayed.getPt());
 				} else {
 					Message msg = messageService.findMsg(delayed.getTempName(), "delayed");
 					if (msg != null) {