fix: 修复水印重复叠加问题 (#2622)

Co-authored-by: whong <whif@foxmail.com>
This commit is contained in:
予ヾ 2023-03-14 17:06:25 +08:00 committed by GitHub
parent 3de5b53bcd
commit 3b0077dc73
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions

View File

@ -4,10 +4,15 @@ import { addResizeListener, removeResizeListener } from '/@/utils/event';
import { isDef } from '/@/utils/is';
const domSymbol = Symbol('watermark-dom');
const sourceMap = new WeakMap<HTMLElement, {}>();
export function useWatermark(
appendEl: Ref<HTMLElement | null> = ref(document.body) as Ref<HTMLElement>,
) {
const appendElRaw = unref(appendEl);
if (appendElRaw && sourceMap.has(appendElRaw)) {
return sourceMap.get(appendElRaw);
}
const func = useRafThrottle(function () {
const el = unref(appendEl);
if (!el) return;
@ -82,6 +87,7 @@ export function useWatermark(
const { clientHeight: height, clientWidth: width } = el;
updateWatermark({ str, width, height });
el.appendChild(div);
sourceMap.set(el, { setWatermark, clear });
return id;
};