forked from fanruan/design
XiaXiang
6 years ago
14 changed files with 760 additions and 499 deletions
@ -1,303 +0,0 @@
|
||||
package com.fr.design.mainframe.alphafine.search.manager; |
||||
|
||||
import com.fr.base.FRContext; |
||||
import com.fr.base.Utils; |
||||
import com.fr.design.mainframe.alphafine.AlphaFineConstants; |
||||
import com.fr.design.mainframe.alphafine.AlphaFineHelper; |
||||
import com.fr.design.mainframe.alphafine.CellType; |
||||
import com.fr.design.mainframe.alphafine.cell.CellModelHelper; |
||||
import com.fr.design.mainframe.alphafine.cell.model.ActionModel; |
||||
import com.fr.design.mainframe.alphafine.cell.model.AlphaCellModel; |
||||
import com.fr.design.mainframe.alphafine.cell.model.MoreModel; |
||||
import com.fr.design.mainframe.alphafine.model.SearchResult; |
||||
import com.fr.design.mainframe.toolbar.UpdateActionManager; |
||||
import com.fr.file.XMLFileManager; |
||||
import com.fr.general.ComparatorUtils; |
||||
import com.fr.general.FRLogger; |
||||
import com.fr.general.IOUtils; |
||||
import com.fr.general.Inter; |
||||
import com.fr.json.JSONException; |
||||
import com.fr.json.JSONObject; |
||||
import com.fr.stable.ProductConstants; |
||||
import com.fr.stable.StableUtils; |
||||
import com.fr.stable.StringUtils; |
||||
import com.fr.stable.project.ProjectConstants; |
||||
import com.fr.stable.xml.XMLPrintWriter; |
||||
import com.fr.stable.xml.XMLReadable; |
||||
import com.fr.stable.xml.XMLTools; |
||||
import com.fr.stable.xml.XMLableReader; |
||||
|
||||
import java.io.File; |
||||
import java.io.FileOutputStream; |
||||
import java.io.FileWriter; |
||||
import java.io.IOException; |
||||
import java.io.OutputStream; |
||||
import java.io.StringReader; |
||||
import java.util.Collections; |
||||
import java.util.HashMap; |
||||
import java.util.Iterator; |
||||
import java.util.List; |
||||
import java.util.Map; |
||||
|
||||
/** |
||||
* Created by XiaXiang on 2017/5/15. |
||||
*/ |
||||
public class RecentSearchManager extends XMLFileManager implements AlphaFineSearchProcessor { |
||||
|
||||
private static final String XML_TAG = "AlphaFineRecent"; |
||||
private static final int MAX_SIZE = 3; |
||||
private static RecentSearchManager recentSearchManager = null; |
||||
private static File recentFile = null; |
||||
private SearchResult modelList; |
||||
private SearchResult recentModelList; |
||||
private Map<String, SearchResult> recentKVModelMap = new HashMap<>(); |
||||
|
||||
public synchronized static RecentSearchManager getRecentSearchManger() { |
||||
if (recentSearchManager == null) { |
||||
recentSearchManager = new RecentSearchManager(); |
||||
try { |
||||
XMLTools.readFileXML(recentSearchManager, recentSearchManager.getRecentEnvFile()); |
||||
} catch (Exception e) { |
||||
FRContext.getLogger().error(e.getMessage(), e); |
||||
} |
||||
} |
||||
return recentSearchManager; |
||||
|
||||
} |
||||
|
||||
@Override |
||||
public void readXML(XMLableReader reader) { |
||||
if (reader.isAttr()) { |
||||
reader.readXMLObject(new XMLReadable() { |
||||
public void readXML(XMLableReader reader) { |
||||
if (reader.isChildNode()) { |
||||
String nodeName = reader.getTagName(); |
||||
if (nodeName.equals("RecentModelList")) { |
||||
String key = reader.getAttrAsString("searchKey", StringUtils.EMPTY); |
||||
final SearchResult list = new SearchResult(); |
||||
reader.readXMLObject(new XMLReadable() { |
||||
@Override |
||||
public void readXML(XMLableReader reader) { |
||||
readCellModel(reader, list); |
||||
} |
||||
} |
||||
); |
||||
recentKVModelMap.put(key, list); |
||||
} |
||||
} |
||||
} |
||||
}); |
||||
} |
||||
|
||||
} |
||||
|
||||
private void readCellModel(XMLableReader reader, List<AlphaCellModel> list) { |
||||
if (reader.isChildNode()) { |
||||
String nodeName = reader.getTagName(); |
||||
if (nodeName.equals("model")) { |
||||
String modelValue = reader.getAttrAsString("cellModel", StringUtils.EMPTY); |
||||
addModelToList(list, modelValue); |
||||
} |
||||
} |
||||
} |
||||
|
||||
|
||||
private void addModelToList(List<AlphaCellModel> list, String modelValue) { |
||||
try { |
||||
AlphaCellModel model = CellModelHelper.getModelFromJson(new JSONObject(modelValue)); |
||||
if (model != null) { |
||||
list.add(model); |
||||
} |
||||
} catch (JSONException e) { |
||||
FRLogger.getLogger().error(e.getMessage()); |
||||
} |
||||
} |
||||
|
||||
|
||||
@Override |
||||
public void writeXML(XMLPrintWriter writer) { |
||||
writer.startTAG(XML_TAG); |
||||
if (!recentKVModelMap.isEmpty()) { |
||||
for (String key : recentKVModelMap.keySet()) { |
||||
writer.startTAG("RecentModelList"); |
||||
writer.attr("searchKey", key); |
||||
for (AlphaCellModel model : recentKVModelMap.get(key)) { |
||||
try { |
||||
String modelValue = model.ModelToJson().toString(); |
||||
writer.startTAG("model"); |
||||
writer.attr("cellModel", modelValue); |
||||
writer.end(); |
||||
} catch (JSONException e) { |
||||
FRLogger.getLogger().error(e.getMessage()); |
||||
} |
||||
} |
||||
writer.end(); |
||||
} |
||||
} |
||||
writer.end(); |
||||
|
||||
} |
||||
|
||||
@Override |
||||
public String fileName() { |
||||
return "AlphaFine_Recent.xml"; |
||||
} |
||||
|
||||
|
||||
/** |
||||
* 获取xml |
||||
* |
||||
* @return |
||||
*/ |
||||
private File getRecentFile() { |
||||
if (recentFile == null) { |
||||
recentFile = new File(ProductConstants.getEnvHome() + File.separator + fileName()); |
||||
} |
||||
return recentFile; |
||||
} |
||||
|
||||
private File getRecentEnvFile() { |
||||
File envFile = getRecentFile(); |
||||
if (!envFile.exists()) { |
||||
createRecentFile(envFile); |
||||
} |
||||
return envFile; |
||||
} |
||||
|
||||
/** |
||||
* 创建XML |
||||
* |
||||
* @param envFile |
||||
*/ |
||||
private void createRecentFile(File envFile) { |
||||
try { |
||||
FileWriter fileWriter = new FileWriter(envFile); |
||||
StringReader stringReader = new StringReader("<?xml version=\"1.0\" encoding=\"UTF-8\" ?><AlphaFineRecent></AlphaFineRecent>"); |
||||
Utils.copyCharTo(stringReader, fileWriter); |
||||
stringReader.close(); |
||||
fileWriter.close(); |
||||
} catch (IOException e) { |
||||
FRContext.getLogger().error(e.getMessage(), e); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* 保存XML到设计器安装目录 |
||||
*/ |
||||
public void saveXMLFile() { |
||||
File xmlFile = this.getRecentEnvFile(); |
||||
if (xmlFile == null) { |
||||
return; |
||||
} |
||||
if (!xmlFile.getParentFile().exists()) {//建立目录.
|
||||
StableUtils.mkdirs(xmlFile.getParentFile()); |
||||
} |
||||
|
||||
String tempName = xmlFile.getName() + ProjectConstants.TEMP_SUFFIX; |
||||
File tempFile = new File(xmlFile.getParentFile(), tempName); |
||||
|
||||
writeTempFile(tempFile); |
||||
IOUtils.renameTo(tempFile, xmlFile); |
||||
} |
||||
|
||||
private void writeTempFile(File tempFile) { |
||||
try { |
||||
OutputStream fout = new FileOutputStream(tempFile); |
||||
XMLTools.writeOutputStreamXML(this, fout); |
||||
fout.flush(); |
||||
fout.close(); |
||||
} catch (Exception e) { |
||||
FRContext.getLogger().error(e.getMessage()); |
||||
} |
||||
} |
||||
|
||||
|
||||
public List<AlphaCellModel> getRecentModelList() { |
||||
return recentModelList; |
||||
} |
||||
|
||||
/** |
||||
* 根据搜索字段获取对应的model列表 |
||||
* |
||||
* @param searchText |
||||
* @return |
||||
*/ |
||||
private synchronized SearchResult getRecentModelList(String searchText) { |
||||
recentModelList = new SearchResult(); |
||||
for (String key : recentKVModelMap.keySet()) { |
||||
AlphaFineHelper.checkCancel(); |
||||
if (ComparatorUtils.equals(key, searchText)) { |
||||
recentModelList = recentKVModelMap.get(searchText); |
||||
SearchResult resultModelList = recentModelList; |
||||
Iterator<AlphaCellModel> modelIterator = resultModelList.iterator(); |
||||
SearchResult searchResult = new SearchResult(); |
||||
while (modelIterator.hasNext()) { |
||||
AlphaCellModel model = modelIterator.next(); |
||||
if (model.getType() == CellType.ACTION && !UpdateActionManager.getUpdateActionManager().isEnable(((ActionModel) model).getAction())) { |
||||
continue; |
||||
} else { |
||||
searchResult.add(model); |
||||
} |
||||
|
||||
} |
||||
Collections.sort(searchResult); |
||||
int size = searchResult.size(); |
||||
if (size > MAX_SIZE) { |
||||
SearchResult result = new SearchResult(); |
||||
result.addAll(searchResult.subList(0, MAX_SIZE)); |
||||
return result; |
||||
} |
||||
return searchResult; |
||||
} |
||||
} |
||||
return recentModelList; |
||||
} |
||||
|
||||
/** |
||||
* 将搜索结果加入到当前MAP中 |
||||
* |
||||
* @param searchKey |
||||
* @param cellModel |
||||
*/ |
||||
public void addRecentModel(String searchKey, AlphaCellModel cellModel) { |
||||
if (recentKVModelMap.keySet().contains(searchKey)) { |
||||
List<AlphaCellModel> cellModels = recentKVModelMap.get(searchKey); |
||||
int index = cellModels.indexOf(cellModel); |
||||
if (index >= 0) { |
||||
cellModels.get(index).addSearchCount(); |
||||
} else { |
||||
cellModels.add(cellModel); |
||||
} |
||||
//trimToSize(cellModels);
|
||||
} else { |
||||
SearchResult list = new SearchResult(); |
||||
list.add(cellModel); |
||||
recentKVModelMap.put(searchKey, list); |
||||
} |
||||
} |
||||
|
||||
|
||||
private synchronized void trimToSize(List<AlphaCellModel> cellModels) { |
||||
if (cellModels.size() > AlphaFineConstants.MAX_FILE_SIZE) { |
||||
cellModels.remove(0); |
||||
} |
||||
} |
||||
|
||||
|
||||
@Override |
||||
public synchronized SearchResult getLessSearchResult(String searchText) { |
||||
this.modelList = new SearchResult(); |
||||
recentModelList = getRecentModelList(searchText); |
||||
if (recentModelList != null && recentModelList.size() > 0) { |
||||
modelList.add(new MoreModel(Inter.getLocText("FR-Designer_AlphaFine_Latest"))); |
||||
} |
||||
modelList.addAll(recentModelList); |
||||
return modelList; |
||||
} |
||||
|
||||
@Override |
||||
public SearchResult getMoreSearchResult() { |
||||
return new SearchResult(); |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,317 @@
|
||||
//package com.fr.design.mainframe.alphafine.search.manager.impl;
|
||||
//
|
||||
//import com.fr.base.FRContext;
|
||||
//import com.fr.base.Utils;
|
||||
//import com.fr.design.actions.UpdateAction;
|
||||
//import com.fr.design.mainframe.alphafine.AlphaFineConstants;
|
||||
//import com.fr.design.mainframe.alphafine.AlphaFineHelper;
|
||||
//import com.fr.design.mainframe.alphafine.CellType;
|
||||
//import com.fr.design.mainframe.alphafine.cell.CellModelHelper;
|
||||
//import com.fr.design.mainframe.alphafine.cell.model.ActionModel;
|
||||
//import com.fr.design.mainframe.alphafine.cell.model.AlphaCellModel;
|
||||
//import com.fr.design.mainframe.alphafine.cell.model.MoreModel;
|
||||
//import com.fr.design.mainframe.alphafine.model.SearchResult;
|
||||
//import com.fr.design.mainframe.alphafine.search.manager.fun.AlphaFineSearchProvider;
|
||||
//import com.fr.design.mainframe.toolbar.UpdateActionManager;
|
||||
//import com.fr.file.XMLFileManager;
|
||||
//import com.fr.general.ComparatorUtils;
|
||||
//import com.fr.general.FRLogger;
|
||||
//import com.fr.general.IOUtils;
|
||||
//import com.fr.general.Inter;
|
||||
//import com.fr.json.JSONException;
|
||||
//import com.fr.json.JSONObject;
|
||||
//import com.fr.stable.ProductConstants;
|
||||
//import com.fr.stable.StableUtils;
|
||||
//import com.fr.stable.StringUtils;
|
||||
//import com.fr.stable.project.ProjectConstants;
|
||||
//import com.fr.stable.xml.XMLPrintWriter;
|
||||
//import com.fr.stable.xml.XMLReadable;
|
||||
//import com.fr.stable.xml.XMLTools;
|
||||
//import com.fr.stable.xml.XMLableReader;
|
||||
//
|
||||
//import java.io.File;
|
||||
//import java.io.FileOutputStream;
|
||||
//import java.io.FileWriter;
|
||||
//import java.io.IOException;
|
||||
//import java.io.OutputStream;
|
||||
//import java.io.StringReader;
|
||||
//import java.util.Collections;
|
||||
//import java.util.HashMap;
|
||||
//import java.util.Iterator;
|
||||
//import java.util.List;
|
||||
//import java.util.Map;
|
||||
//
|
||||
///**
|
||||
// * Created by XiaXiang on 2017/5/15.
|
||||
// */
|
||||
//public class RecentSearchManager extends XMLFileManager implements AlphaFineSearchProvider {
|
||||
//
|
||||
// private static final String XML_TAG = "AFSearch_Recent";
|
||||
// private static final int MAX_SIZE = 3;
|
||||
// private static RecentSearchManager recentSearchManager = null;
|
||||
// private static File recentFile = null;
|
||||
// private SearchResult modelList;
|
||||
// private SearchResult recentModelList;
|
||||
// private Map<String, SearchResult> recentKVModelMap = new HashMap<>();
|
||||
//
|
||||
// public synchronized static RecentSearchManager getInstance() {
|
||||
// if (recentSearchManager == null) {
|
||||
// recentSearchManager = new RecentSearchManager();
|
||||
// try {
|
||||
// XMLTools.readFileXML(recentSearchManager, recentSearchManager.getRecentEnvFile());
|
||||
// } catch (Exception e) {
|
||||
// FRContext.getLogger().error(e.getMessage(), e);
|
||||
// }
|
||||
// }
|
||||
// return recentSearchManager;
|
||||
//
|
||||
// }
|
||||
//
|
||||
// @Override
|
||||
// public void readXML(XMLableReader reader) {
|
||||
// if (reader.isAttr()) {
|
||||
// reader.readXMLObject(new XMLReadable() {
|
||||
// public void readXML(XMLableReader reader) {
|
||||
// if (reader.isChildNode()) {
|
||||
// String nodeName = reader.getTagName();
|
||||
// if (nodeName.equals("RecentModelList")) {
|
||||
// String key = reader.getAttrAsString("searchKey", StringUtils.EMPTY);
|
||||
// final SearchResult list = new SearchResult();
|
||||
// reader.readXMLObject(new XMLReadable() {
|
||||
// @Override
|
||||
// public void readXML(XMLableReader reader) {
|
||||
// readCellModel(reader, list);
|
||||
// }
|
||||
// }
|
||||
// );
|
||||
// recentKVModelMap.put(key, list);
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
// });
|
||||
// }
|
||||
//
|
||||
// }
|
||||
//
|
||||
// private void readCellModel(XMLableReader reader, List<AlphaCellModel> list) {
|
||||
// if (reader.isChildNode()) {
|
||||
// String nodeName = reader.getTagName();
|
||||
// if (nodeName.equals("model")) {
|
||||
// String modelValue = reader.getAttrAsString("cellModel", StringUtils.EMPTY);
|
||||
// addModelToList(list, modelValue);
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
//
|
||||
//
|
||||
// private void addModelToList(List<AlphaCellModel> list, String modelValue) {
|
||||
// try {
|
||||
// AlphaCellModel model = CellModelHelper.getModelFromJson(new JSONObject(modelValue));
|
||||
// if (model != null) {
|
||||
// list.add(model);
|
||||
// }
|
||||
// } catch (JSONException e) {
|
||||
// FRLogger.getLogger().error(e.getMessage());
|
||||
// }
|
||||
// }
|
||||
//
|
||||
//
|
||||
// @Override
|
||||
// public void writeXML(XMLPrintWriter writer) {
|
||||
// writer.startTAG(XML_TAG);
|
||||
// if (!recentKVModelMap.isEmpty()) {
|
||||
// for (String key : recentKVModelMap.keySet()) {
|
||||
// writer.startTAG("RecentModelList");
|
||||
// writer.attr("searchKey", key);
|
||||
// for (AlphaCellModel model : recentKVModelMap.get(key)) {
|
||||
// try {
|
||||
// String modelValue = model.ModelToJson().toString();
|
||||
// writer.startTAG("model");
|
||||
// writer.attr("cellModel", modelValue);
|
||||
// writer.end();
|
||||
// } catch (JSONException e) {
|
||||
// FRLogger.getLogger().error(e.getMessage());
|
||||
// }
|
||||
// }
|
||||
// writer.end();
|
||||
// }
|
||||
// }
|
||||
// writer.end();
|
||||
//
|
||||
// }
|
||||
//
|
||||
// @Override
|
||||
// public String fileName() {
|
||||
// return "AFSearch_Recent.xml";
|
||||
// }
|
||||
//
|
||||
//
|
||||
// /**
|
||||
// * 获取xml
|
||||
// *
|
||||
// * @return
|
||||
// */
|
||||
// private File getRecentFile() {
|
||||
// if (recentFile == null) {
|
||||
// recentFile = new File(ProductConstants.getEnvHome() + File.separator + fileName());
|
||||
// }
|
||||
// return recentFile;
|
||||
// }
|
||||
//
|
||||
// private File getOldFile() {
|
||||
// return new File(ProductConstants.getEnvHome() + File.separator + "AlphaFine_Recent.xml");
|
||||
// }
|
||||
//
|
||||
// private File getRecentEnvFile() {
|
||||
// File envFile = getRecentFile();
|
||||
// File oldFile = getOldFile();
|
||||
// if (oldFile.exists()) {
|
||||
// StableUtils.deleteFile(oldFile);
|
||||
// }
|
||||
// if (!envFile.exists()) {
|
||||
// createRecentFile(envFile);
|
||||
// }
|
||||
// return envFile;
|
||||
// }
|
||||
//
|
||||
// /**
|
||||
// * 创建XML
|
||||
// *
|
||||
// * @param envFile
|
||||
// */
|
||||
// private void createRecentFile(File envFile) {
|
||||
// try {
|
||||
// FileWriter fileWriter = new FileWriter(envFile);
|
||||
// StringReader stringReader = new StringReader("<?xml version=\"1.0\" encoding=\"UTF-8\" ?><AFSearch_Recent></AFSearch_Recent>");
|
||||
// Utils.copyCharTo(stringReader, fileWriter);
|
||||
// stringReader.close();
|
||||
// fileWriter.close();
|
||||
// } catch (IOException e) {
|
||||
// FRContext.getLogger().error(e.getMessage(), e);
|
||||
// }
|
||||
// }
|
||||
//
|
||||
// /**
|
||||
// * 保存XML到设计器安装目录
|
||||
// */
|
||||
// public void saveXMLFile() {
|
||||
// File xmlFile = this.getRecentEnvFile();
|
||||
// if (xmlFile == null) {
|
||||
// return;
|
||||
// }
|
||||
// if (!xmlFile.getParentFile().exists()) {//建立目录.
|
||||
// StableUtils.mkdirs(xmlFile.getParentFile());
|
||||
// }
|
||||
//
|
||||
// String tempName = xmlFile.getName() + ProjectConstants.TEMP_SUFFIX;
|
||||
// File tempFile = new File(xmlFile.getParentFile(), tempName);
|
||||
//
|
||||
// writeTempFile(tempFile);
|
||||
// IOUtils.renameTo(tempFile, xmlFile);
|
||||
// }
|
||||
//
|
||||
// private void writeTempFile(File tempFile) {
|
||||
// try {
|
||||
// OutputStream fout = new FileOutputStream(tempFile);
|
||||
// XMLTools.writeOutputStreamXML(this, fout);
|
||||
// fout.flush();
|
||||
// fout.close();
|
||||
// } catch (Exception e) {
|
||||
// FRContext.getLogger().error(e.getMessage());
|
||||
// }
|
||||
// }
|
||||
//
|
||||
//
|
||||
// public List<AlphaCellModel> getRecentModelList() {
|
||||
// return recentModelList;
|
||||
// }
|
||||
//
|
||||
// /**
|
||||
// * 根据搜索字段获取对应的model列表
|
||||
// *
|
||||
// * @param searchText
|
||||
// * @return
|
||||
// */
|
||||
// private synchronized SearchResult getRecentModelList(String searchText) {
|
||||
// recentModelList = new SearchResult();
|
||||
// for (String key : recentKVModelMap.keySet()) {
|
||||
// AlphaFineHelper.checkCancel();
|
||||
// if (ComparatorUtils.equals(key, searchText)) {
|
||||
// recentModelList = recentKVModelMap.get(searchText);
|
||||
// SearchResult resultModelList = recentModelList;
|
||||
// Iterator<AlphaCellModel> modelIterator = resultModelList.iterator();
|
||||
// SearchResult searchResult = new SearchResult();
|
||||
// while (modelIterator.hasNext()) {
|
||||
// AlphaCellModel model = modelIterator.next();
|
||||
// if (model.getType() == CellType.ACTION) {
|
||||
// UpdateAction action = UpdateActionManager.getUpdateActionManager().getActionByName(model.getName());
|
||||
// if (action != null) {
|
||||
// ((ActionModel) model).setAction(action);
|
||||
// searchResult.add(model);
|
||||
// }
|
||||
// } else {
|
||||
// searchResult.add(model);
|
||||
// }
|
||||
//
|
||||
// }
|
||||
// Collections.sort(searchResult);
|
||||
// int size = searchResult.size();
|
||||
// if (size > MAX_SIZE) {
|
||||
// SearchResult result = new SearchResult();
|
||||
// result.addAll(searchResult.subList(0, MAX_SIZE));
|
||||
// return result;
|
||||
// }
|
||||
// return searchResult;
|
||||
// }
|
||||
// }
|
||||
// return recentModelList;
|
||||
// }
|
||||
//
|
||||
// /**
|
||||
// * 将搜索结果加入到当前MAP中
|
||||
// *
|
||||
// * @param searchKey
|
||||
// * @param cellModel
|
||||
// */
|
||||
// public void addRecentModel(String searchKey, AlphaCellModel cellModel) {
|
||||
// if (recentKVModelMap.keySet().contains(searchKey)) {
|
||||
// List<AlphaCellModel> cellModels = recentKVModelMap.get(searchKey);
|
||||
// int index = cellModels.indexOf(cellModel);
|
||||
// if (index >= 0) {
|
||||
// cellModels.get(index).addSearchCount();
|
||||
// } else {
|
||||
// cellModels.add(cellModel);
|
||||
// }
|
||||
// //trimToSize(cellModels);
|
||||
// } else {
|
||||
// SearchResult list = new SearchResult();
|
||||
// list.add(cellModel);
|
||||
// recentKVModelMap.put(searchKey, list);
|
||||
// }
|
||||
// }
|
||||
//
|
||||
//
|
||||
// private synchronized void trimToSize(List<AlphaCellModel> cellModels) {
|
||||
// if (cellModels.size() > AlphaFineConstants.MAX_FILE_SIZE) {
|
||||
// cellModels.remove(0);
|
||||
// }
|
||||
// }
|
||||
//
|
||||
//
|
||||
// @Override
|
||||
// public synchronized SearchResult getLessSearchResult(String searchText) {
|
||||
// this.modelList = new SearchResult();
|
||||
// recentModelList = getRecentModelList(searchText);
|
||||
// if (recentModelList != null && recentModelList.size() > 0) {
|
||||
// modelList.add(new MoreModel(Inter.getLocText("FR-Designer_AlphaFine_Latest")));
|
||||
// }
|
||||
// modelList.addAll(recentModelList);
|
||||
// return modelList;
|
||||
// }
|
||||
//
|
||||
// @Override
|
||||
// public SearchResult getMoreSearchResult(String searchText) {
|
||||
// return new SearchResult();
|
||||
// }
|
||||
//
|
||||
//}
|
||||
@ -0,0 +1,193 @@
|
||||
package com.fr.design.mainframe.alphafine.search.manager.impl; |
||||
|
||||
import com.fr.design.mainframe.alphafine.cell.CellModelHelper; |
||||
import com.fr.design.mainframe.alphafine.cell.model.AlphaCellModel; |
||||
import com.fr.design.mainframe.alphafine.cell.model.MoreModel; |
||||
import com.fr.design.mainframe.alphafine.model.SearchResult; |
||||
import com.fr.design.mainframe.alphafine.search.manager.fun.AlphaFineSearchProvider; |
||||
import com.fr.general.Inter; |
||||
import com.fr.json.JSONException; |
||||
import com.fr.json.JSONObject; |
||||
import com.fr.log.FRLoggerFactory; |
||||
import com.fr.stable.ProductConstants; |
||||
import com.fr.third.org.apache.lucene.analysis.Analyzer; |
||||
import com.fr.third.org.apache.lucene.analysis.standard.StandardAnalyzer; |
||||
import com.fr.third.org.apache.lucene.document.Document; |
||||
import com.fr.third.org.apache.lucene.document.Field; |
||||
import com.fr.third.org.apache.lucene.document.IntField; |
||||
import com.fr.third.org.apache.lucene.document.StringField; |
||||
import com.fr.third.org.apache.lucene.index.DirectoryReader; |
||||
import com.fr.third.org.apache.lucene.index.IndexReader; |
||||
import com.fr.third.org.apache.lucene.index.IndexWriter; |
||||
import com.fr.third.org.apache.lucene.index.IndexWriterConfig; |
||||
import com.fr.third.org.apache.lucene.index.Term; |
||||
import com.fr.third.org.apache.lucene.search.IndexSearcher; |
||||
import com.fr.third.org.apache.lucene.search.Query; |
||||
import com.fr.third.org.apache.lucene.search.ScoreDoc; |
||||
import com.fr.third.org.apache.lucene.search.Sort; |
||||
import com.fr.third.org.apache.lucene.search.SortField; |
||||
import com.fr.third.org.apache.lucene.search.TermQuery; |
||||
import com.fr.third.org.apache.lucene.search.TopFieldDocs; |
||||
import com.fr.third.org.apache.lucene.store.Directory; |
||||
import com.fr.third.org.apache.lucene.store.FSDirectory; |
||||
import com.fr.third.org.apache.lucene.util.Version; |
||||
|
||||
import java.io.File; |
||||
import java.io.IOException; |
||||
import java.util.List; |
||||
|
||||
/** |
||||
* Created by XiaXiang on 2018/1/22. |
||||
*/ |
||||
public class RecentSearchManager implements AlphaFineSearchProvider { |
||||
private static final int MAX_SIZE = 3; |
||||
private static RecentSearchManager recentSearchManager = null; |
||||
IndexReader indexReader = null; |
||||
IndexSearcher indexSearcher = null; |
||||
//索引存储路径
|
||||
private String path = ProductConstants.getEnvHome() + File.separator + "searchIndex"; |
||||
//分词器，暂时先用这个
|
||||
private Analyzer analyzer = new StandardAnalyzer(Version.LUCENE_40); |
||||
// 存储目录
|
||||
private Directory directory = null; |
||||
private IndexWriterConfig config = new IndexWriterConfig(Version.LUCENE_40, analyzer); |
||||
private IndexWriter indexWriter = null; |
||||
private SearchResult recentModelList; |
||||
private SearchResult modelList; |
||||
|
||||
public synchronized static RecentSearchManager getInstance() { |
||||
if (recentSearchManager == null) { |
||||
return new RecentSearchManager(); |
||||
} |
||||
return recentSearchManager; |
||||
} |
||||
|
||||
@Override |
||||
public SearchResult getLessSearchResult(String searchText) { |
||||
this.modelList = new SearchResult(); |
||||
recentModelList = getRecentModelList(searchText); |
||||
if (recentModelList != null && recentModelList.size() > 0) { |
||||
modelList.add(new MoreModel(Inter.getLocText("FR-Designer_AlphaFine_Latest"))); |
||||
} |
||||
modelList.addAll(recentModelList); |
||||
return modelList; |
||||
} |
||||
|
||||
@Override |
||||
public SearchResult getMoreSearchResult(String searchText) { |
||||
return new SearchResult(); |
||||
} |
||||
|
||||
public synchronized SearchResult getRecentModelList(String searchText) { |
||||
return searchBySort(searchText); |
||||
} |
||||
|
||||
public List<AlphaCellModel> getRecentModelList() { |
||||
return recentModelList; |
||||
} |
||||
|
||||
/** |
||||
* 初始化indexWriter |
||||
*/ |
||||
private void initWriter() { |
||||
if (indexWriter == null) { |
||||
try { |
||||
directory = FSDirectory.open(new File(path)); |
||||
} catch (IOException e) { |
||||
FRLoggerFactory.getLogger().error("cannot open directory " + path); |
||||
} |
||||
try { |
||||
indexWriter = new IndexWriter(directory, config); |
||||
} catch (IOException e) { |
||||
FRLoggerFactory.getLogger().error("not privilege to write to" + path); |
||||
} |
||||
} |
||||
|
||||
} |
||||
|
||||
/** |
||||
* 初始化indexReader |
||||
*/ |
||||
private void initReader() { |
||||
if (indexReader == null) { |
||||
try { |
||||
directory = FSDirectory.open(new File(path)); |
||||
indexReader = DirectoryReader.open(directory); |
||||
} catch (IOException e) { |
||||
FRLoggerFactory.getLogger().error("not privilege to read " + path); |
||||
} |
||||
indexSearcher = new IndexSearcher(indexReader); |
||||
} |
||||
|
||||
} |
||||
|
||||
/** |
||||
* 添加模型 |
||||
* @param searchKey |
||||
* @param cellModel |
||||
* @param searchCount |
||||
*/ |
||||
public void addModel(String searchKey, AlphaCellModel cellModel, int searchCount) { |
||||
try { |
||||
initWriter(); |
||||
Document doc = new Document(); |
||||
doc.add(new StringField("searchKey", searchKey, Field.Store.YES)); |
||||
doc.add(new StringField("cellModel", cellModel.ModelToJson().toString(), Field.Store.YES)); |
||||
doc.add(new IntField("searchCount", searchCount, Field.Store.YES)); |
||||
writeDoc(doc); |
||||
} catch (JSONException e) { |
||||
FRLoggerFactory.getLogger().error("add document error: " + e.getMessage()); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* 写文档，建立索引 |
||||
* @param doc |
||||
*/ |
||||
private void writeDoc(Document doc) { |
||||
try { |
||||
indexWriter.addDocument(doc); |
||||
indexWriter.commit(); |
||||
indexWriter.close(); |
||||
} catch (IOException e) { |
||||
FRLoggerFactory.getLogger().error("write document error: " + e.getMessage()); |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* 按序搜索 |
||||
* @param key |
||||
* @return |
||||
*/ |
||||
private synchronized SearchResult searchBySort(String key) { |
||||
recentModelList = new SearchResult(); |
||||
try { |
||||
initReader(); |
||||
IndexSearcher searcher = new IndexSearcher(indexReader); |
||||
//构建排序字段
|
||||
SortField[] sortField = new SortField[1]; |
||||
sortField[0] = new SortField("searchCount", SortField.Type.INT, true); |
||||
Sort sortKey = new Sort(sortField); |
||||
String searchField = "searchKey"; |
||||
Term term = new Term(searchField, key); |
||||
Query query = new TermQuery(term); |
||||
TopFieldDocs docs = searcher.search(query, MAX_SIZE, sortKey); |
||||
ScoreDoc[] scores = docs.scoreDocs; |
||||
this.recentModelList = new SearchResult(); |
||||
//遍历结果
|
||||
for (ScoreDoc scoreDoc : scores) { |
||||
Document document = searcher.doc(scoreDoc.doc); |
||||
System.out.println(document.get("cellModel") + "\n" + document.get("searchCount") + "\n"); |
||||
AlphaCellModel model = CellModelHelper.getModelFromJson(new JSONObject(document.get("cellModel"))); |
||||
this.recentModelList.add(model); |
||||
|
||||
} |
||||
} catch (Exception e) { |
||||
FRLoggerFactory.getLogger().error("local search error: " + e.getMessage()); |
||||
return recentModelList; |
||||
} |
||||
return recentModelList; |
||||
} |
||||
|
||||
|
||||
} |
||||
Loading…
Reference in new issue