98 lines
3.5 KiB
TypeScript
98 lines
3.5 KiB
TypeScript
|
import axios from "axios";
|
|||
|
|
import { getRandomOffset } from "@/utils/others/quillutils";
|
|||
|
|
const xml2js = require("xml2js");
|
|||
|
|
|
|||
|
|
// 定义文章详细信息的 TypeScript 接口
|
|||
|
|
interface ArticleDetail {
|
|||
|
|
id: string;
|
|||
|
|
title: string;
|
|||
|
|
abstract: string;
|
|||
|
|
authors: string[];
|
|||
|
|
publishedDate: string;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 定义PubMed文章ID为字符串类型
|
|||
|
|
type PubMedID = string;
|
|||
|
|
|
|||
|
|
// 定义idList为PubMedID数组
|
|||
|
|
type IDList = PubMedID[];
|
|||
|
|
|
|||
|
|
async function getPubMedPapers(query: string, year: number, limit = 2) {
|
|||
|
|
try {
|
|||
|
|
const baseURL =
|
|||
|
|
"https://eutils.ncbi.nlm.nih.gov/entrez/eutils/esearch.fcgi";
|
|||
|
|
const db = "pubmed"; // 设定搜索的数据库为PubMed
|
|||
|
|
const retMax = limit; // 检索的最大记录数
|
|||
|
|
const retStart = getRandomOffset(10 - limit); // 假设每页最多10条，根据需要随机偏移
|
|||
|
|
|
|||
|
|
const url = `${baseURL}?db=${db}&term=${query}[Title/Abstract]+AND+${year}[Date+-+Publication]&retMax=${retMax}&retStart=${retStart}&api_key=${process.env.NEXT_PUBLIC_PUBMED_API_KEY}`;
|
|||
|
|
const response = await axios.get(url, { responseType: "text" });
|
|||
|
|
console.log(response.data);
|
|||
|
|
// 解析XML数据
|
|||
|
|
const parser = new xml2js.Parser();
|
|||
|
|
const result = await parser.parseStringPromise(response.data);
|
|||
|
|
|
|||
|
|
// 提取PubMed文章ID
|
|||
|
|
const idList = result.eSearchResult.IdList[0].Id.map((id) => id);
|
|||
|
|
console.log(idList);
|
|||
|
|
// 可以进一步使用这些ID来获取文章详情，例如使用esummary或efetch
|
|||
|
|
// 这里只返回了ID列表，你可能需要根据实际需要进行调整
|
|||
|
|
return idList;
|
|||
|
|
} catch (error) {
|
|||
|
|
console.error("Error fetching data from PubMed API:", error);
|
|||
|
|
return null; // 或根据需要处理错误
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 根据文章ID列表获取详细信息
|
|||
|
|
async function getPubMedPaperDetails(idList: IDList) {
|
|||
|
|
try {
|
|||
|
|
const ids = idList.join(","); // 将ID列表转换为逗号分隔的字符串
|
|||
|
|
const baseURL = "https://eutils.ncbi.nlm.nih.gov/entrez/eutils/efetch.fcgi";
|
|||
|
|
const url = `${baseURL}?db=pubmed&id=${ids}&rettype=abstract&retmode=xml`;
|
|||
|
|
|
|||
|
|
const response = await axios.get(url);
|
|||
|
|
const data = response.data; // 这里获取的数据是XML格式，需要解析
|
|||
|
|
// 解析XML数据
|
|||
|
|
const parser = new xml2js.Parser({ explicitArray: false });
|
|||
|
|
let result = await parser.parseStringPromise(data);
|
|||
|
|
console.log(result);
|
|||
|
|
|
|||
|
|
// 提取并处理文章详细信息
|
|||
|
|
const articles = result.PubmedArticleSet.PubmedArticle.map((article) => {
|
|||
|
|
const articleDetails = article.MedlineCitation.Article;
|
|||
|
|
return {
|
|||
|
|
id: article.MedlineCitation.PMID,
|
|||
|
|
title: articleDetails.ArticleTitle,
|
|||
|
|
abstract: articleDetails.Abstract
|
|||
|
|
? articleDetails.Abstract.AbstractText
|
|||
|
|
: "",
|
|||
|
|
authors: articleDetails.AuthorList
|
|||
|
|
? articleDetails.AuthorList.Author.map(
|
|||
|
|
(author) => `${author.ForeName} ${author.LastName}`
|
|||
|
|
)
|
|||
|
|
: [],
|
|||
|
|
publishedDate: articleDetails.ArticleDate
|
|||
|
|
? articleDetails.ArticleDate.Date
|
|||
|
|
: "",
|
|||
|
|
};
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
return articles;
|
|||
|
|
} catch (error) {
|
|||
|
|
console.error("Error fetching paper details from PubMed:", error);
|
|||
|
|
return null;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 示例：使用这些函数
|
|||
|
|
async function fetchPubMedData(query: string, year: number, limit: number) {
|
|||
|
|
const idList = await getPubMedPapers(query, year, limit);
|
|||
|
|
if (idList && idList.length > 0) {
|
|||
|
|
const paperDetails = await getPubMedPaperDetails(idList);
|
|||
|
|
console.log(paperDetails); // 处理或显示文章详情
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
export { fetchPubMedData, getPubMedPapers, getPubMedPaperDetails };
|