mirror of
https://github.com/d0zingcat/NotionNext.git
synced 2026-05-14 15:09:22 +00:00
99 lines
2.9 KiB
JavaScript
99 lines
2.9 KiB
JavaScript
import BLOG from '@/blog.config'
|
|
import { NotionAPI } from 'notion-client'
|
|
import { getDataFromCache, setDataToCache } from '@/lib/cache/cache_manager'
|
|
import { deepClone, delay } from '../utils'
|
|
|
|
export async function getPostBlocks(id, from, slice) {
|
|
const cacheKey = 'page_block_' + id
|
|
let pageBlock = await getDataFromCache(cacheKey)
|
|
if (pageBlock) {
|
|
console.log('[命中缓存]:', `from:${from}`, cacheKey)
|
|
return filterPostBlocks(id, pageBlock, slice)
|
|
}
|
|
|
|
const start = new Date().getTime()
|
|
pageBlock = await getPageWithRetry(id, from)
|
|
const end = new Date().getTime()
|
|
console.log('[API耗时]', `${end - start}ms`)
|
|
|
|
if (pageBlock) {
|
|
await setDataToCache(cacheKey, pageBlock)
|
|
return filterPostBlocks(id, pageBlock, slice)
|
|
}
|
|
return pageBlock
|
|
}
|
|
|
|
/**
|
|
* 调用接口，失败会重试
|
|
* @param {*} id
|
|
* @param {*} retryAttempts
|
|
*/
|
|
async function getPageWithRetry(id, from, retryAttempts = 3) {
|
|
if (retryAttempts && retryAttempts > 0) {
|
|
console.log('[请求API]', `from:${from}`, `id:${id}`, retryAttempts < 3 ? `剩余重试次数:${retryAttempts}` : '')
|
|
try {
|
|
const authToken = BLOG.NOTION_ACCESS_TOKEN || null
|
|
const api = new NotionAPI({ authToken })
|
|
const pageData = await api.getPage(id)
|
|
console.info('[响应成功]:', `from:${from}`, `id:${id}`)
|
|
return pageData
|
|
} catch (e) {
|
|
await delay(1000)
|
|
const cacheKey = 'page_block_' + id
|
|
const pageBlock = await getDataFromCache(cacheKey)
|
|
if (pageBlock) {
|
|
console.log('[重试缓存]', `from:${from}`, `id:${id}`)
|
|
return pageBlock
|
|
}
|
|
return await getPageWithRetry(id, from, retryAttempts - 1)
|
|
}
|
|
} else {
|
|
console.error('[请求失败]:', `from:${from}`, `id:${id}`)
|
|
return null
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 获取到的blockMap删除不需要的字段
|
|
* @param {*} id 页面ID
|
|
* @param {*} pageBlock 页面元素
|
|
* @param {*} slice 截取数量
|
|
* @returns
|
|
*/
|
|
function filterPostBlocks(id, pageBlock, slice) {
|
|
const clonePageBlock = deepClone(pageBlock)
|
|
let count = 0
|
|
|
|
for (const i in clonePageBlock?.block) {
|
|
const b = clonePageBlock?.block[i]
|
|
if (slice && slice > 0 && count > slice) {
|
|
delete clonePageBlock?.block[i]
|
|
continue
|
|
}
|
|
count++
|
|
// 处理 c++ 和 c#两种语言
|
|
if (b?.value?.type === 'code') {
|
|
if (b?.value?.properties?.language?.[0][0] === 'C++') {
|
|
b.value.properties.language[0][0] = 'cpp'
|
|
}
|
|
if (b?.value?.properties?.language?.[0][0] === 'C#') {
|
|
b.value.properties.language[0][0] = 'csharp'
|
|
}
|
|
}
|
|
|
|
delete b?.role
|
|
delete b?.value?.version
|
|
delete b?.value?.created_by_table
|
|
delete b?.value?.created_by_id
|
|
delete b?.value?.last_edited_by_table
|
|
delete b?.value?.last_edited_by_id
|
|
delete b?.value?.space_id
|
|
}
|
|
|
|
// 去掉不用的字段
|
|
if (id === BLOG.NOTION_PAGE_ID) {
|
|
return clonePageBlock
|
|
}
|
|
return clonePageBlock
|
|
}
|