无聊啊，写一写，微信聊天机器人

1.sql数据表建立：

一开始使用varchar型的，后来感觉不行，只有重新写成int 自增的，自增在数据库里设置吧

CREATE TABLE CHATMAN_KNOWLEDGE(
      K_ID INTEGER NOT NULL,
      QUESTION VARCHAR(2000),
      ANSWER VARCHAR(8000),
      CATEGORY INT,
      PRIMARY KEY(K_ID)
);
alter tablle CHATMAN_KNOWLEDGE
 add id INTEGER ;
CREATE TABLE CHATMAN_KNSUB(
KN_ID INTEGER  identity(1,1) primary key,    --主键自增长
K_ID INTEGER,
ANSWER VARCHAR(8000)
/*PRIMARY KEY(KN_ID)*/
)


CREATE TABLE CHATMAN_JOKE(
J_ID VARCHAR(32),
CONTENT VARCHAR(5000),
PRIMARY KEY(J_ID)
)


CREATE TABLE CHATMAN_CHAT_LOG(
CL_ID VARCHAR(32) NOT NULL,
W_ID VARCHAR(100) NOT NULL,
CREATE_TIME DATETIME,
REG_MSG VARCHAR(2000),
RESP_MSG VARCHAR(2000),
CHAT_CATEGORY INT,
PRIMARY KEY(CL_ID)
)

2.数据库crud

3.服务chatService

package com.dm.wx.service;


import java.io.File;
import java.util.Date;
import java.util.List;
import java.util.Random;


import com.dm.chatman.domain.ChatLog;
import com.dm.chatman.domain.Knowledge;
import com.dm.chatman.service.ChatLogService;
import com.dm.chatman.service.JokeService;
import com.dm.chatman.service.KnowledgeService;
import com.dm.chatman.service.KnsubService;
import com.dm.core.service.BaseService;
import org.apache.lucene.document.Document;
import org.apache.lucene.document.IntField;
import org.apache.lucene.document.StringField;
import org.apache.lucene.document.TextField;
import org.apache.lucene.document.Field.Store;
import org.apache.lucene.index.IndexReader;
import org.apache.lucene.index.IndexWriter;
import org.apache.lucene.index.IndexWriterConfig;
import org.apache.lucene.queryparser.classic.QueryParser;
import org.apache.lucene.search.IndexSearcher;
import org.apache.lucene.search.Query;
import org.apache.lucene.search.ScoreDoc;
import org.apache.lucene.search.TopDocs;
import org.apache.lucene.store.Directory;
import org.apache.lucene.store.FSDirectory;
import org.apache.lucene.util.Version;


import org.wltea.analyzer.lucene.IKAnalyzer;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Isolation;
import org.springframework.transaction.annotation.Transactional;
import org.wltea.analyzer.lucene.IKAnalyzer;
import org.apache.lucene.util.Version;


import java.io.File;
import java.util.List;


/**
 * Created by IntelliJ IDEA.
 * User: Administrator
 * Date: 15-5-25
 * Time: 上午6:21
 * To change this template use File | Settings | File Templates.
 */
@Service("wxChatService")
@Transactional(isolation = Isolation.READ_COMMITTED, readOnly = true, rollbackFor = Throwable.class)
public class ChatService extends BaseService {
        @Autowired
    @Qualifier("chatmanKnowledgeService")
    private KnowledgeService knowledgeService;
     @Autowired
    @Qualifier("chatmanJokeService")
    private JokeService jokeService;
        @Autowired
    @Qualifier("chatmanChatLogService")
    private ChatLogService chatLogService;


        @Autowired
    @Qualifier("chatmanKnsubService")
    private KnsubService knsubService;
    public static String getIndexDir() {
// 得到.class文件所在路径（WEB-INF/classes/）
String classpath = ChatService.class.getResource("/").getPath();
// 将classpath中的%20替换为空格
classpath = classpath.replaceAll("%20", " ");
// 索引存储位置：WEB-INF/classes/index/
return classpath + "index/";
}
    /**
* 创建索引
*/
public  void createIndex() throws Exception{
// 取得问答知识库中的所有记录
List<Knowledge> knowledgeList = knowledgeService.list();
Directory directory = null;
IndexWriter indexWriter = null;
try {
directory = FSDirectory.open(new File(getIndexDir()));
IndexWriterConfig iwConfig = new IndexWriterConfig(Version.LUCENE_46, new IKAnalyzer(true));
indexWriter = new IndexWriter(directory, iwConfig);
Document doc = null;
// 遍历问答知识库创建索引
for (Knowledge knowledge : knowledgeList) {
doc = new Document();
// 对question进行分词存储
                doc = new Document();
// 对question进行分词存储
doc.add(new TextField("question", knowledge.getQuestion(), Store.YES));
// 对id、answer和category不分词存储
doc.add(new IntField("id", knowledge.getK_id(), Store.YES));
doc.add(new StringField("answer", knowledge.getAnswer(), Store.YES));
doc.add(new IntField("category", knowledge.getCategory(), Store.YES));
indexWriter.addDocument(doc);
}
indexWriter.close();
directory.close();
} catch (Exception e) {
e.printStackTrace();
}
}
    /**
* 从索引文件中根据问题检索答案
*
* @param content
* @return Knowledge
*/
@SuppressWarnings("deprecation")
private static Knowledge searchIndex(String content) {
Knowledge knowledge = null;
try {
Directory directory = FSDirectory.open(new File(getIndexDir()));
IndexReader reader = IndexReader.open(directory);
IndexSearcher searcher = new IndexSearcher(reader);
// 使用查询解析器创建Query
QueryParser questParser = new QueryParser(Version.LUCENE_46, "question", new IKAnalyzer(true));
Query query = questParser.parse(QueryParser.escape(content));
// 检索得分最高的文档
TopDocs topDocs = searcher.search(query, 1);
if (topDocs.totalHits > 0) {
knowledge = new Knowledge();
ScoreDoc[] scoreDoc = topDocs.scoreDocs;
for (ScoreDoc sd : scoreDoc) {
Document doc = searcher.doc(sd.doc);
knowledge.setK_id(doc.getField("id").numericValue().intValue());
knowledge.setQuestion(doc.get("question"));
knowledge.setAnswer(doc.get("answer"));
knowledge.setCategory(doc.getField("category").numericValue().intValue());
}
}
reader.close();
directory.close();
} catch (Exception e) {
knowledge = null;
e.printStackTrace();
}
return knowledge;
}


/**
* 聊天方法（根据question返回answer）
*
* @param openId 用户的OpenID
* @param createTime 消息创建时间
* @param question 用户上行的问题
* @return answer
*/
public  String chat(String openId, String createTime, String question) throws Exception{
String answer = null;
int chatCategory = 0;
Knowledge knowledge = searchIndex(question);
// 找到匹配项
if (null != knowledge) {
// 笑话
if (2 == knowledge.getCategory()) {
answer = jokeService.querymyJoke().getContent();
chatCategory = 2;
}
// 上下文
else if (3 == knowledge.getCategory()) {
// 判断上一次的聊天类别
int category = chatLogService.getLastCategory(openId).getChat_category();
// 如果是笑话，本次继续回复笑话给用户
if (2 == category) {
answer = jokeService.querymyJoke().getContent();
chatCategory = 2;
} else {
answer = knowledge.getAnswer();
chatCategory = knowledge.getCategory();
}
}
// 普通对话
else {
answer = knowledge.getAnswer();
// 如果答案为空，根据知识id从问答知识分表中随机获取一条
if ("".equals(answer))
answer = knsubService.getKnowledSub(knowledge.getK_id()).getAnswer();
chatCategory = 1;
}
}
// 未找到匹配项
else {
answer = getDefaultAnswer();
chatCategory = 0;
}
// 保存聊天记录
        ChatLog chatLog = new ChatLog();
        chatLog.setW_id(openId);
        chatLog.setCreate_time(new Date());
        chatLog.setReg_msg(question);
        chatLog.setResp_msg(answer);
        chatLog.setChat_category(chatCategory);
        chatLogService.create(chatLog);


return answer;
}


/**
* 随机获取一个默认的答案
*
* @return
*/
private static String getDefaultAnswer() {
String []answer = {
"要不我们聊点别的？",
"恩？你到底在说什么呢？",
"没有听懂你说的，能否换个说法？",
"虽然不明白你的意思，但我却能用心去感受",
"听的我一头雾水，阁下的知识真是渊博呀，膜拜~",
"真心听不懂你在说什么，要不你换种表达方式如何？",
"哎，我小学语文是体育老师教的，理解起来有点困难哦",
"是世界变化太快，还是我不够有才？为何你说话我不明白？"
};
return answer[getRandomNumber(answer.length)];
}


/**
* 随机生成 0~length-1 之间的某个值
*
* @return int
*/
private static int getRandomNumber(int length) {
Random random = new Random();
return random.nextInt(length);
}
}

4.配置索引

   private ChatService chatService;
    public void init(ServletConfig config) throws ServletException {
        System.out.print("初始化servelt成功");
        File indexDir = new File(ChatService.getIndexDir());
// 如果索引目录不存在则创建索引
if (!indexDir.exists()) {
             try{
chatService.createIndex();
             }catch (Exception e){
                 System.out.print("创建索引异常");
             }
        }
        this.config = config;
    }

5.设置web.xnl每次首先加载

    <servlet>
        <servlet-name>coreServlet</servlet-name>
        <servlet-class>com.dm.wx.HttpServletProxy</servlet-class>
        <load-on-startup>0</load-on-startup>
    </servlet>
        <!--<servlet>-->
        <!--<servlet-name>coreServlet</servlet-name>-->
        <!--<servlet-class>com.dm.wx.controller.CoreServlet</servlet-class>-->
    <!--</servlet>-->
    <servlet-mapping>
        <servlet-name>coreServlet</servlet-name>
        <url-pattern>/coreServlet</url-pattern>
    </servlet-mapping>

6.添加和修改数据表，删除索引，重新启动就可以加载添加的数据索引了，自动化查询时挺有用的。

大家可参考我的博客：

我的博客：115.159.110.224

我的微信公众号：my528xx

公众号二维码：