【React】为img标签的src请求添加自定义头部,添加token,实现前端带身份向后端请求图片

应用场景

问题:前端写入指定的img标签和src属性,请求不了后端图片。
原因:401,UnauthorizedError,因为前端中的src属于普通的get请求,不带自定义header,而后端有对每个请求都进行身份验证,最终返回

{
    
    
    status: 1,
    msg: "TOKEN ERROR"
}

前端请求头中找不到

Authorization: Bearer token....

那请求的接口都有对axios进行拦截,但是img标签的src不经过此请求

axios.interceptors.request.use(config => {
    
    
    const passURL = ['/api/login', 'api/register'];
    if (passURL.includes(config.url)) return config;
    const tk = localStorage.getItem('TOKEN');
    if (tk) {
    
    
        config.headers.Authorization = 'Bearer ' + tk;
    } else {
    
    
        delete config.headers.Authorization;
    }
    return config;
})

尝试解决一:

后台使用CORS,允许跨域及携带cookie,后端服务端口号是:9999

const cors = require('cors');
app.use(cors({
    
    credentials: true}));

而后端一开始使用解析token白名单

const expJWT = require('express-jwt'); //解析token
// 解析token,unless表示不希望解析的路径
const unPath = ['/api/login','/api/register',/^\/public\/.*/];
app.use(expJWT({
    
     secret: config.jwtKey, algorithms: ['HS256']}).unless({
    
    path: unPath}));

注意:/^\/public\/.*/
这里使用的是正则匹配,如果直接写 ‘/public/*’则达不到效果
此时可以通过http://localhost:9999/public/images/bg1.jpg直接访问后端图片,但是我们都希望后端的图片不能完全开放,也是需要身份验证的。最终还是需要走axios或者fetch等请求,为请求带上自定义头部。

尝试解决二:

既然已经保证了图片是可以访问的,那我们就要在返回中加一道防线,防止其他不相关的用户随意访问。
重置img的src。既然直接带上src访问最终都是错误,那我们可以利用onError事件重置一下图片。
util/axios.js

import axios from 'axios';
axios.interceptors.request.use(config => {
    
    
    const passURL = ['/api/login', 'api/register'];
    if (passURL.includes(config.url)) return config;
    const tk = localStorage.getItem('TOKEN');
    if (tk) {
    
    
        config.headers.Authorization = 'Bearer ' + tk;
    } else {
    
    
        delete config.headers.Authorization;
    }
    return config;
})

axios.interceptors.response.use(response=>{
    
    
    //....
    return response;
})
export default axios;

在util中写一个方法去异步请求图片
util/utils.js

import axios from './axios';
/**
 * 图片拦截请求
 * @param url
 * @constructor
 */
export async function ImageInterceptor(url) {
    
    
    // 请求图片,axios是已经实现拦截的,如上面导入,responseType为arraybuffer
    let response = await axios(url, {
    
     responseType: 'arraybuffer' });
    /**
     * 利用data转化图片格式
     */
    // response是后台返回二进制流
    // btoa表示将字符串转化成编码字符串
    // Uint8Array 数组类型表示一个8位无符号整型数组
    // reduce方法为Uint8Array.prototype.reduce(),对累加器和数组的每个值应用函数(从左到右),使其归约为单一的值,默认值为'',实现累加
    // String.fromCharCode(),将 Unicode 编码转为一个字符:
    let res = "data:image/png;base64," + btoa(new Uint8Array(response.data).reduce((data, byte) => data + String.fromCharCode(byte), ""));
    return res;
}

showImg.js

import React, {
    
     Component } from 'react';
import {
    
     withRouter } from 'react-router-dom';
import {
    
     connect } from 'react-redux';
import {
    
     bindActionCreators } from 'redux';
import {
    
     actionCreators } from '@store';
import {
    
    ImageInterceptor} from '@utils/util'

class showImage extends Component {
    
    
    getImage = async (e)=>{
    
    
    	// 获得base64格式的图片
        let baseImage = await ImageInterceptor(e.target.src);
        e.target.src = baseImage;
    }
    render() {
    
    
        return (
            <>
                {
    
    /* 此处的process.env.REACT_APP_BASE_URL我在.env中设置的是REACT_APP_BASE_URL = http://localhost:9999,对应后端服务,当然也可以手动设置http://localhost:9999/public/images/bg1.jpg */}
              <img ref='myimage' src={
    
    `${
      
      process.env.REACT_APP_BASE_URL}/public/images/bg1.jpg`} onError={
    
    this.getImage}></img>
            </>
        )
    }
}
const mapStateToProps = state => {
    
    
    return {
    
    
        loginData: state.login
    }
}
const mapDispatchToProps = dispatch => {
    
    
    return {
    
    
        actionFn: bindActionCreators(actionCreators, dispatch)
    }
}
export default withRouter(connect(mapStateToProps, mapDispatchToProps)(showImage));

核心代码:

// 请求图片
let response = await axios(url, {
    
     responseType: 'arraybuffer' });
let res = "data:image/png;base64,"
	+ btoa(new Uint8Array(response.data).reduce((data, byte) => 
		data + String.fromCharCode(byte), ""
	));


//重置src
getImage = async (e)=>{
    
    
	// 获得base64格式的图片
    let baseImage = await ImageInterceptor(e.target.src);
    e.target.src = baseImage;
}

//设置onError事件
<img ref='myimage' src={
    
    `....`} onError={
    
    this.getImage}></img>

最后不需要解析的路径中可以删除了

const expJWT = require('express-jwt'); //解析token
// 解析token,unless表示不希望解析的路径
const unPath = ['/api/login','/api/register'];
app.use(expJWT({
    
     secret: config.jwtKey, algorithms: ['HS256']}).unless({
    
    path: unPath}));

总结:

这个问题纠结了我很久,总想着在哪里拦截图片的网络请求,尝试了在代理的header上添加,但是createProxyMiddleware上并没有走通,可能是我配置错了,没有找到合适的方案,就采用了以上的方式,不知道前端的小伙伴们有没有其他更好的方法,欢迎指导~

猜你喜欢

转载自blog.csdn.net/weixin_42339197/article/details/110526216