mirror of
https://github.com/yanyiwu/cppjieba.git
synced 2025-07-18 00:00:12 +08:00
91 lines
2.2 KiB
C++
91 lines
2.2 KiB
C++
#include <unistd.h>
|
|
#include <algorithm>
|
|
#include <string>
|
|
#include <ctype.h>
|
|
#include <string.h>
|
|
#include "Limonp/Config.hpp"
|
|
#include "Husky/ThreadPoolServer.hpp"
|
|
#include "MixSegment.hpp"
|
|
#include "QuerySegment.hpp"
|
|
#include "FullSegment.hpp"
|
|
|
|
using namespace Husky;
|
|
using namespace CppJieba;
|
|
|
|
class ReqHandler: public IRequestHandler {
|
|
public:
|
|
ReqHandler(const ISegment& segment): _segment(segment) {
|
|
}
|
|
virtual ~ReqHandler() {};
|
|
|
|
virtual bool do_GET(const HttpReqInfo& httpReq, string& strSnd) const {
|
|
string sentence, tmp;
|
|
vector<string> words;
|
|
httpReq.GET("key", tmp);
|
|
URLDecode(tmp, sentence);
|
|
_segment.cut(sentence, words);
|
|
if(httpReq.GET("format", tmp) && tmp == "simple") {
|
|
join(words.begin(), words.end(), strSnd, " ");
|
|
return true;
|
|
}
|
|
strSnd << words;
|
|
return true;
|
|
}
|
|
virtual bool do_POST(const HttpReqInfo& httpReq, string& strSnd) const {
|
|
vector<string> words;
|
|
_segment.cut(httpReq.getBody(), words);
|
|
strSnd << words;
|
|
return true;
|
|
}
|
|
private:
|
|
const ISegment& _segment;
|
|
};
|
|
|
|
bool run(int argc, char** argv) {
|
|
if(argc < 2) {
|
|
return false;
|
|
}
|
|
Config conf(argv[1]);
|
|
if(!conf) {
|
|
return false;
|
|
}
|
|
int port = 0;
|
|
int threadNumber = 0;
|
|
int queueMaxSize = 0;
|
|
string dictPath;
|
|
string modelPath;
|
|
string userDictPath;
|
|
LIMONP_CHECK(conf.get("port", port));
|
|
LIMONP_CHECK(conf.get("thread_number", threadNumber));
|
|
LIMONP_CHECK(conf.get("queue_max_size", queueMaxSize));
|
|
LIMONP_CHECK(conf.get("dict_path", dictPath));
|
|
LIMONP_CHECK(conf.get("model_path", modelPath));
|
|
if(!conf.get("user_dict_path", userDictPath)) { //optional
|
|
userDictPath = "";
|
|
}
|
|
|
|
LogInfo("config info: %s", conf.getConfigInfo().c_str());
|
|
|
|
/*
|
|
* segment can be one of (MPSegment, HMMSegment, MixSegment, QuerySegment ...)
|
|
*/
|
|
//MPSegment segment(dictPath, userDictPath);
|
|
//HMMSegment segment(modelPath);
|
|
MixSegment segment(dictPath, modelPath, userDictPath);
|
|
//QuerySegment segment(dictPath, modelPath);
|
|
|
|
ReqHandler reqHandler(segment);
|
|
ThreadPoolServer sf(threadNumber, queueMaxSize, port, reqHandler);
|
|
return sf.start();
|
|
|
|
}
|
|
|
|
int main(int argc, char* argv[]) {
|
|
if(!run(argc, argv)) {
|
|
printf("usage: %s <config_file>\n", argv[0]);
|
|
return EXIT_FAILURE;
|
|
}
|
|
return EXIT_SUCCESS;
|
|
}
|
|
|