mirror of
https://github.com/wgh136/nysoure.git
synced 2025-12-15 07:31:14 +00:00
API for rebuilding search index
This commit is contained in:
@@ -32,4 +32,7 @@ BACKUP_SCHEDULE=0 2 * * *
|
|||||||
BACKUP_RETENTION_DAYS=30
|
BACKUP_RETENTION_DAYS=30
|
||||||
|
|
||||||
# Download Configuration
|
# Download Configuration
|
||||||
DOWNLOAD_SECRET_KEY=your_download_secret_key_here
|
DOWNLOAD_SECRET_KEY=your_download_secret_key_here
|
||||||
|
|
||||||
|
# Access Key for Development API
|
||||||
|
DEV_ACCESS_KEY=your_dev_access_key_here
|
||||||
@@ -3,12 +3,27 @@ package api
|
|||||||
import (
|
import (
|
||||||
"nysoure/server/middleware"
|
"nysoure/server/middleware"
|
||||||
|
|
||||||
|
"nysoure/server/search"
|
||||||
|
|
||||||
"github.com/gofiber/fiber/v3"
|
"github.com/gofiber/fiber/v3"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
func rebuildSearchIndex(c fiber.Ctx) error {
|
||||||
|
err := search.RebuildSearchIndex()
|
||||||
|
if err != nil {
|
||||||
|
return c.Status(fiber.StatusInternalServerError).JSON(fiber.Map{
|
||||||
|
"error": "Failed to rebuild search index: " + err.Error(),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
return c.JSON(fiber.Map{
|
||||||
|
"message": "Search index rebuilt successfully",
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
func AddDevAPI(router fiber.Router) {
|
func AddDevAPI(router fiber.Router) {
|
||||||
devGroup := router.Group("/dev")
|
devGroup := router.Group("/dev")
|
||||||
|
devGroup.Use(middleware.DevMiddleware())
|
||||||
{
|
{
|
||||||
devGroup.Use(middleware.DevMiddleware())
|
devGroup.Post("/rebuild_search_index", rebuildSearchIndex)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -287,7 +287,11 @@ func handleGetCharactersFromVndb(c fiber.Ctx) error {
|
|||||||
if vnID == "" {
|
if vnID == "" {
|
||||||
return model.NewRequestError("VNDB ID is required")
|
return model.NewRequestError("VNDB ID is required")
|
||||||
}
|
}
|
||||||
characters, err := service.GetCharactersFromVndb(vnID)
|
uid, ok := c.Locals("uid").(uint)
|
||||||
|
if !ok {
|
||||||
|
return model.NewUnAuthorizedError("You must be logged in to get characters from VNDB")
|
||||||
|
}
|
||||||
|
characters, err := service.GetCharactersFromVndb(vnID, uid)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -6,12 +6,19 @@ import (
|
|||||||
"nysoure/server/dao"
|
"nysoure/server/dao"
|
||||||
"nysoure/server/model"
|
"nysoure/server/model"
|
||||||
"nysoure/server/utils"
|
"nysoure/server/utils"
|
||||||
|
"os"
|
||||||
"strconv"
|
"strconv"
|
||||||
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/blevesearch/bleve"
|
"github.com/blevesearch/bleve"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
index bleve.Index
|
||||||
|
mu = sync.RWMutex{}
|
||||||
|
)
|
||||||
|
|
||||||
type ResourceParams struct {
|
type ResourceParams struct {
|
||||||
Id uint
|
Id uint
|
||||||
Title string
|
Title string
|
||||||
@@ -26,9 +33,9 @@ type ResourceCharacter struct {
|
|||||||
CV string
|
CV string
|
||||||
}
|
}
|
||||||
|
|
||||||
var index bleve.Index
|
|
||||||
|
|
||||||
func AddResourceToIndex(r model.Resource) error {
|
func AddResourceToIndex(r model.Resource) error {
|
||||||
|
mu.RLock()
|
||||||
|
defer mu.RUnlock()
|
||||||
cs := make([]ResourceCharacter, 0, len(r.Characters))
|
cs := make([]ResourceCharacter, 0, len(r.Characters))
|
||||||
for _, c := range r.Characters {
|
for _, c := range r.Characters {
|
||||||
cs = append(cs, ResourceCharacter{
|
cs = append(cs, ResourceCharacter{
|
||||||
@@ -96,6 +103,8 @@ func init() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func SearchResource(keyword string) ([]uint, error) {
|
func SearchResource(keyword string) ([]uint, error) {
|
||||||
|
mu.RLock()
|
||||||
|
defer mu.RUnlock()
|
||||||
query := bleve.NewMatchQuery(keyword)
|
query := bleve.NewMatchQuery(keyword)
|
||||||
searchRequest := bleve.NewSearchRequest(query)
|
searchRequest := bleve.NewSearchRequest(query)
|
||||||
searchResults, err := index.Search(searchRequest)
|
searchResults, err := index.Search(searchRequest)
|
||||||
@@ -128,3 +137,17 @@ func IsStopWord(word string) bool {
|
|||||||
tokens := analyzer.Analyze([]byte(word))
|
tokens := analyzer.Analyze([]byte(word))
|
||||||
return len(tokens) == 0
|
return len(tokens) == 0
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func RebuildSearchIndex() error {
|
||||||
|
mu.Lock()
|
||||||
|
defer mu.Unlock()
|
||||||
|
err := index.Close()
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to close search index: %w", err)
|
||||||
|
}
|
||||||
|
err = os.Remove(utils.GetStoragePath() + "/resource_index.bleve")
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to remove search index: %w", err)
|
||||||
|
}
|
||||||
|
return createIndex()
|
||||||
|
}
|
||||||
|
|||||||
@@ -615,7 +615,15 @@ func GetPinnedResources() ([]model.ResourceView, error) {
|
|||||||
return views, nil
|
return views, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func GetCharactersFromVndb(vnID string) ([]CharacterParams, error) {
|
func GetCharactersFromVndb(vnID string, uid uint) ([]CharacterParams, error) {
|
||||||
|
canUpload, err := checkUserCanUpload(uid)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if !canUpload {
|
||||||
|
return nil, model.NewUnAuthorizedError("You have not permission to fetch characters from VNDB")
|
||||||
|
}
|
||||||
|
|
||||||
client := http.Client{}
|
client := http.Client{}
|
||||||
jsonStr := fmt.Sprintf(`
|
jsonStr := fmt.Sprintf(`
|
||||||
{
|
{
|
||||||
|
|||||||
Reference in New Issue
Block a user