1. 程式人生 > >使用html2canvas.js實現頁面截圖並顯示或上傳

使用html2canvas.js實現頁面截圖並顯示或上傳

  最近寫專案有用到html2canvas.js,可以實現頁面的截圖功能,但遭遇了許多的坑,特此寫一篇隨筆記錄一下。

  在使用html2canvas時可能會遇到諸如只能擷取視覺化介面、截圖沒有背景色、svg標籤無法擷取等問題,下面詳細的說明一下。

一、匯入html2canvas.js

  這個不需要多說,可以從github上獲取:https://github.com/niklasvh/html2canvas

  也可以直接匯入連結: <script src="https://cdn.bootcss.com/html2canvas/0.5.0-beta4/html2canvas.js"></script>

 

 

  使用起來也非常簡單,具體的API可以去網上查詢,生成png圖片使用“image/png”即可。

  其中$("#xxx")為你想要擷取的div,外面可以通過jquery獲取它,當然document獲取也是可以的。

html2canvas($("#xxx"), {
         onrendered: function (canvas) {
             var url = canvas.toDataURL("image/png");
        window.location.href = url;
           }
   });

  其它型別的圖片如jpg,為image/jpeg等等,可自行查詢API。

  到這裡其實簡單的截圖已經完成了,如果介面稍微複雜一點的話,可能就會出現各種坑,下面一個一個解決。

二、svg無法擷取的問題

  當我們擷取一個div時,如果這個div中存在svg標籤,一般情況下是擷取不到的,比如擷取一個流程圖,得到的是下面這個樣子:

 

  

  可以看到,流程圖的線沒有擷取到,也就是svg沒有擷取到,這時的解決方法是把svg轉換成canvas再進行截圖即可,直接上程式碼。

  這裡的each迴圈是迴圈所有的svg標籤,將它們全部轉換為canvas

    if
(typeof html2canvas !== 'undefined') { //以下是對svg的處理 var nodesToRecover = []; var nodesToRemove = []; var svgElem = cloneDom.find('svg'); svgElem.each(function (index, node) { var parentNode = node.parentNode; var svg = node.outerHTML.trim(); var canvas = document.createElement('canvas'); canvas.width = 650; canvas.height = 798; canvg(canvas, svg); if (node.style.position) { canvas.style.position += node.style.position; canvas.style.left += node.style.left; canvas.style.top += node.style.top; } nodesToRecover.push({ parent: parentNode, child: node }); parentNode.removeChild(node); nodesToRemove.push({ parent: parentNode, child: canvas }); parentNode.appendChild(canvas); }); }

  這裡需要用到canvg.js,以及它的依賴檔案rgbcolor.js,網上可以直接下載,也可以直接匯入。

三、背景透明的問題

  這個其實很簡單,因為它預設是透明的,html2canvas中有一個引數background就可以新增背景色,如下:

html2canvas(cloneDom, {
      onrendered: function(canvas) {
           var url =canvas.toDataURL("image/png");
      },
      background:"#fafafa"
}); 

四、只能擷取可視部分的問題

  如果需要擷取的div超出了介面,可能會遇到擷取不全的問題,如上圖,只有一半的內容,這是因為看不到的部分被隱藏了,而html2canvas是無法擷取隱藏的dom的。

  所以此時的解決辦法是使用克隆,將需要擷取的部分克隆一份放在頁面底層,再使用html2canvas擷取這個完整的div,擷取完成後再remove這部分內容即可,完整程式碼如下:

function showQRCode() {
    scrollTo(0, 0);
    
    //克隆節點,預設為false,即不復制方法屬性,為true是全部複製。
    var cloneDom = $("#d1").clone(true);
    //設定克隆節點的z-index屬性,只要比被克隆的節點層級低即可。
    cloneDom.css({
        "background-color": "#fafafa",
        "position": "absolute",
        "top": "0px",
        "z-index": "-1",
        "height": 798,
        "width": 650
    });
   
    if (typeof html2canvas !== 'undefined') {
        //以下是對svg的處理
        var nodesToRecover = [];
        var nodesToRemove = [];
        var svgElem = cloneDom.find('svg');//divReport為需要擷取成圖片的dom的id
        svgElem.each(function (index, node) {
            var parentNode = node.parentNode;
            var svg = node.outerHTML.trim();

            var canvas = document.createElement('canvas');
            canvas.width = 650;
            canvas.height = 798;
            canvg(canvas, svg); 
            if (node.style.position) {
                canvas.style.position += node.style.position;
                canvas.style.left += node.style.left;
                canvas.style.top += node.style.top;
            }

            nodesToRecover.push({
                parent: parentNode,
                child: node
            });
            parentNode.removeChild(node);

            nodesToRemove.push({
                parent: parentNode,
                child: canvas
            });

            parentNode.appendChild(canvas);
        });
        
        //將克隆節點動態追加到body後面。
        $("body").append(cloneDom);

        html2canvas(cloneDom, {
            onrendered: function(canvas) {
                var url =canvas.toDataURL("image/png");
                window.location.href = url ;
                cloneDom.remove();    //清空克隆的內容
             },
             background:"#fafafa"
        }); 
        
   } 
}

  這裡外面首先將要擷取的div克隆一份,並將z-index設定為最小,避免引起介面的不美觀,然後是對svg進行的處理,上面已經分析過了,最後將克隆節點追加到body後面即可。

  在onrendered中,我們可以直接使用location.href跳轉檢視圖片,可以進行儲存操作,也可以將url寫入img的src中顯示在介面上,如 $('#imgId').attr('src',url); 

  最後可以在介面展示剛剛擷取到的圖片:

  

五、上傳圖片儲存到資料庫,並在介面中獲取該圖片顯示

  現在得到url了,需要上傳到後端,並存到資料庫中,再另一個展示的介面中載入該圖片。我一般習慣於使用url來儲存圖片路徑,而不是用blob儲存。

  因為需要在另一個介面中獲取圖片,所以我把圖片存在了與webapp同級的一個resource目錄下,程式碼如下:

        //儲存圖片並返回圖片路徑
        BASE64Decoder decoder = new BASE64Decoder();
        byte[] b = decoder.decodeBuffer(product.getProPic().substring("data:image/png;base64,".length()));
        ByteArrayInputStream bais = new ByteArrayInputStream(b);
        BufferedImage bi1 = ImageIO.read(bais);
        String url = "user_resource" + File.separator + "img" + File.separator + "product_"+UUID.randomUUID().toString().replace("-", "")+".png";
        String totalUrl = System.getProperty("root") + url;
        File w2 = new File(totalUrl);
        ImageIO.write(bi1, "png", w2);
        
        product.setProPic(url);    //將圖片的相對路徑儲存到資料庫中
        
        int res = productMapper.insertSelective(product);    //新增到資料庫

  這裡因為涉及到其它邏輯,所以只放一部分程式碼。

  這裡使用的是BASE64Decoder來儲存圖片,我們獲取到圖片後,需要使用substring將“data:image/png;base64,”的內容擷取掉,因為“,”後面才是圖片的url, url.substring("data:image/png;base64,".length()) 。

  對於路徑,上面程式碼中的url是我儲存到資料庫中的內容,而totalUrl就是實際進行ImageIO的write操作時儲存的真實路徑,getProperty()方法獲取的專案的根目錄,可以在web.xml中配置如下內容,然後 System.getProperty("root") 即可。

<!-- 配置系統獲得專案根目錄 -->
<context-param>
    <param-name>webAppRootKey</param-name>
    <param-value>root</param-value>
</context-param>
<listener>
    <listener-class>
        org.springframework.web.util.WebAppRootListener
    </listener-class>
</listener>

  現在圖片的url就存到資料庫裡了,而圖片本身就儲存在tomcat下該專案的這個目錄下。

  

  最後外面在介面上獲取,只需要在當前的url前面加上專案名即可 <img class="depot-img" src="<%=request.getContextPath()%>/`+e.proPic+`"> 。

  然後就可以看到介面上顯示的圖片了: