Method batchGetLibrariesByIDs

api/service/rag/search.go:3267–3296

(eid int64, libraryIDs []int64)

}

func (s *SearchService) batchGetLibrariesByIDs(eid int64, libraryIDs []int64) (map[int64]*model.Library, error) {
	if len(libraryIDs) == 0 {
		return make(map[int64]*model.Library), nil
	}

	uniqueLibraryIDs := uniqueInt64IDsInOrder(libraryIDs)
	if len(uniqueLibraryIDs) == 0 {
		return make(map[int64]*model.Library), nil
	}

	libraries, err := model.GetLibrariesByEidCached(eid)
	if err != nil {
		return nil, err
	}

	requested := make(map[int64]struct{}, len(uniqueLibraryIDs))
	for _, libraryID := range uniqueLibraryIDs {
		requested[libraryID] = struct{}{}
	}

	result := make(map[int64]*model.Library, len(uniqueLibraryIDs))
	for i := range libraries {
		if _, ok := requested[libraries[i].ID]; !ok {
			continue
		}
		library := libraries[i]
		result[library.ID] = &library
	}
	return result, nil
}

// saveQueryRecord 保存查询记录
func (s *SearchService) saveQueryRecord(eid int64, userID *int64, req *SearchRequest, totalResults int, searchTimeMs int64) (int64, error) {