yeyinly/tvbox-mv
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

MOD:添加首页展示

Browse Source
This commit is contained in:
leosam1024
2023-06-17 21:18:28 +08:00
parent d78c34d18a
commit 4c25797dda
8 changed files with 177 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/leosam/tvbox/mv/MainVerticle.java
+31 -6
View File
@@ -13,6 +13,7 @@ import io.vertx.core.http.HttpServer;
import io.vertx.core.json.Json;
import io.vertx.ext.web.Router;
import io.vertx.ext.web.RoutingContext;
import io.vertx.ext.web.handler.StaticHandler;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
@@ -52,7 +53,8 @@ public class MainVerticle extends AbstractVerticle {
Router router = Router.router(vertx);
router.route("/mv/search").handler(this::searchMv);
router.route("/mv/vod").handler(this::searchMvVod);
router.route().handler(req -> req.response().putHeader("content-type", "text/plain").end("Hello from Vert.x!"));
router.route("/*").handler(StaticHandler.create("static"));
router.route().handler(this::home);
httpServer.requestHandler(router);
httpServer.listen(7777, http -> {
@@ -66,21 +68,36 @@ public class MainVerticle extends AbstractVerticle {
port = httpServer.actualPort();
}
/**
* 参考 <a href="https://github.com/FongMi/TV/blob/release/app/src/main/java/com/fongmi/android/tv/model/SiteViewModel.java#L32">...</a>
* @param req
*/
private void searchMvVod(RoutingContext req) {
// 获取参数
String wd = VertxUtils.queryParam(req, "wd");
String ac = VertxUtils.queryParam(req, "ac");
String ids = VertxUtils.queryParam(req, "ids");
String maxCount = VertxUtils.queryParam(req, "maxCount");
int max = Math.min(Math.max(NumberUtils.toInt(maxCount, 100), 10), 1000);
String query = StringUtils.isNotEmpty(wd) ? wd : ids;
String ac = VertxUtils.queryParam(req, "ac");
String type = VertxUtils.queryParam(req, "t");
//query = StringUtils.isNotEmpty(query) ? query : "我";
String maxCount = VertxUtils.queryParam(req, "maxCount");
int max = Math.min(Math.max(NumberUtils.toInt(maxCount, 200), 10), 1000);
String pg = VertxUtils.queryParam(req, "pg");
int page = NumberUtils.toInt(pg, 0);
// 处理结果
VodResult vodResult = null;
try {
if (mvService != null) {
// 搜索
if (mvService != null && StringUtils.isNotEmpty(query) && page <= 0) {
vodResult = mvService.searchVod(query, max);
}
// 首页
if (mvService != null && StringUtils.isEmpty(query)) {
vodResult = mvService.searchVodHome(type, page);
}
} catch (Exception e) {
throw new RuntimeException(e);
}
@@ -95,8 +112,10 @@ public class MainVerticle extends AbstractVerticle {
private void searchMv(RoutingContext req) {
// 获取参数
String query = VertxUtils.queryParam(req, "query");
String wd = VertxUtils.queryParam(req, "wd");
query = StringUtils.isNotEmpty(query) ? query : wd;
String maxCount = VertxUtils.queryParam(req, "maxCount");
int max = Math.min(Math.max(NumberUtils.toInt(maxCount, 100), 10), 1000);
int max = Math.min(Math.max(NumberUtils.toInt(maxCount, 200), 10), 1000);
// 处理结果
MvResult search = null;
@@ -115,6 +134,12 @@ public class MainVerticle extends AbstractVerticle {
.end(jsonString);
}
private void home(RoutingContext req) {
req.response()
.putHeader("content-type", "text/plain;charset=utf-8")
.end("MV搜索服务\n源码网址：https://github.com/leosam1024/tvbox-mv");
}
public static void main(String[] args) throws Exception {
Vertx vertx = Vertx.vertx();
src/main/java/com/leosam/tvbox/mv/data/VodClass.java
+30
View File
@@ -0,0 +1,30 @@
package com.leosam.tvbox.mv.data;
import com.fasterxml.jackson.annotation.JsonProperty;
public class VodClass {
@JsonProperty("type_id")
private String typeId;
@JsonProperty("type_name")
private String typeName;
public String getTypeId() {
return typeId;
}
public VodClass setTypeId(String typeId) {
this.typeId = typeId;
return this;
}
public String getTypeName() {
return typeName;
}
public VodClass setTypeName(String typeName) {
this.typeName = typeName;
return this;
}
}
src/main/java/com/leosam/tvbox/mv/data/VodResult.java
+13
View File
@@ -1,5 +1,7 @@
package com.leosam.tvbox.mv.data;
import com.fasterxml.jackson.annotation.JsonProperty;
import java.util.List;
/**
@@ -14,6 +16,8 @@ public class VodResult {
private int limit;
private int total;
private List<Vod> list;
@JsonProperty("class")
private List<VodClass> vodClassList;
public void init() {
this.code = 1;
@@ -76,4 +80,13 @@ public class VodResult {
this.list = list;
return this;
}
public List<VodClass> getVodClassList() {
return vodClassList;
}
public VodResult setVodClassList(List<VodClass> vodClassList) {
this.vodClassList = vodClassList;
return this;
}
}
src/main/java/com/leosam/tvbox/mv/service/MvService.java
+71 -2
View File
@@ -1,8 +1,11 @@
package com.leosam.tvbox.mv.service;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.leosam.tvbox.mv.data.MvContent;
import com.leosam.tvbox.mv.data.MvResult;
import com.leosam.tvbox.mv.data.Vod;
import com.leosam.tvbox.mv.data.VodClass;
import com.leosam.tvbox.mv.data.VodResult;
import com.leosam.tvbox.mv.lucene.MvIndex;
import com.leosam.tvbox.mv.lucene.MvSearcher;
@@ -23,9 +26,12 @@ import java.io.File;
import java.io.IOException;
import java.nio.file.Path;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.LinkedHashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.atomic.AtomicInteger;
@@ -35,9 +41,14 @@ import java.util.concurrent.atomic.AtomicInteger;
*/
public class MvService {
private static final Logger logger = LoggerFactory.getLogger(MvService.class);
private static final String indexDirectoryPath = "index";
private static final String MV_FILE = "tvbox/16wMV.txt";
private static final Map<String, List<String>> homeConfigMap = new LinkedHashMap<>();
private static final String HOME_KEY = "HOME";
/**
* 最低相似分数
*/
private static final float MIN_QUERY_SCORE = 3.0F;
private MvSearcher mvSearcher;
@@ -56,6 +67,10 @@ public class MvService {
result.setTotalHits(topDocs.totalHits.value);
result.setList(new ArrayList<>(topDocs.scoreDocs.length));
for (ScoreDoc scoreDoc : topDocs.scoreDocs) {
// 分数太低 忽略
if(scoreDoc.score < MIN_QUERY_SCORE){
break;
}
Document document = mvSearcher.getDocument(scoreDoc.doc);
if (document == null) {
continue;
@@ -97,7 +112,6 @@ public class MvService {
vod.setVodName(wd);
vod.setVodPlayFrom("mv");
vod.setVodPic("http://yanxuan.nosdn.127.net/b6fb987ce79f308949e44f5129a4b51c.jpeg");
// vod.setVodPic("http://m4.auto.itc.cn/auto/content/20230611/45d65d8a001f0c5aa008030f41c98666.jpeg");
List<String> playUrlList = new LinkedList<>();
Set<String> vodActorList = new LinkedHashSet<>();
for (MvContent content : search.getList()) {
@@ -116,7 +130,51 @@ public class MvService {
return vodResult;
}
public VodResult searchVodHome(String type, int page) throws Exception {
if (page > 1) {
return new VodResult();
}
StopWatch stopWatch = new StopWatch();
stopWatch.start();
VodResult vodResult = new VodResult();
vodResult.init();
vodResult.setList(new ArrayList<>()).getList();
// 分类
if (StringUtils.isEmpty(type)) {
vodResult.setVodClassList(new ArrayList<>(homeConfigMap.size()));
Set<String> keySet = homeConfigMap.keySet();
for (String key : keySet) {
if (HOME_KEY.equalsIgnoreCase(key)) {
continue;
}
VodClass vodClass = new VodClass().setTypeId(key).setTypeName(key);
vodResult.getVodClassList().add(vodClass);
}
}
// 首页mv推荐
String thisType = StringUtils.isNotEmpty(type) ? type : HOME_KEY;
List<String> geshouList = homeConfigMap.getOrDefault(thisType, new ArrayList<>());
for (String geshou : geshouList) {
Vod vod = new Vod();
vod.setVodId(geshou);
vod.setVodName(geshou);
vod.setVodPlayFrom("mv");
vod.setVodPic("http://yanxuan.nosdn.127.net/b6fb987ce79f308949e44f5129a4b51c.jpeg");
vodResult.getList().add(vod);
}
stopWatch.stop();
logger.info("加载首页成功，类型={}, 返回={}条, 耗时{}毫秒", thisType, vodResult.getList().size(), stopWatch.getTotalTimeMillis());
return vodResult;
}
public void initIndex() throws Exception {
// 加载首页配置
buildHomeConfig();
// 重建索引
String indexAbsolutePath = new File(indexDirectoryPath).getAbsoluteFile().getAbsolutePath();
@@ -129,6 +187,17 @@ public class MvService {
search("五月天", 10);
}
private void buildHomeConfig() {
try {
String content = ClassPathReaderUtils.getContent("tvbox/home.json");
Map<String, List<String>> map = new ObjectMapper().readValue(content, new TypeReference<HashMap<String, List<String>>>() {
});
homeConfigMap.putAll(map);
logger.info("加载首页配置成功");
} catch (Exception e) {
logger.error("buildHomeConfig error", e);
}
}
private static void reBuildIndex(String indexDirectoryPath) throws IOException {
File file = new File(indexDirectoryPath).getAbsoluteFile();