java实现二维码登录功能

本文采用Springboot工程进行开发，使用Google的zxing生成二维码，直接放代码：

xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
    xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0modelVersion>
    <parent>
        <groupId>org.springframework.bootgroupId>
        <artifactId>spring-boot-starter-parentartifactId>
        <version>2.1.12.RELEASEversion>
        <relativePath /> 
    parent>
    <groupId>com.hcxgroupId>
    <artifactId>spring-jdbcartifactId>
    <version>0.0.1-SNAPSHOTversion>
    <name>spring-jdbcname>
    <description>spring-jdbcdescription>

    <properties>
        <java.version>1.8java.version>
        <spring-cloud.version>Greenwich.RELEASEspring-cloud.version>
        <springfox.version>2.9.2springfox.version>
    properties>

    <dependencies>
        <dependency>
            <groupId>org.springframework.bootgroupId>
            <artifactId>spring-boot-starter-webartifactId>
        dependency>

        <dependency>
            <groupId>org.springframework.bootgroupId>
            <artifactId>spring-boot-devtoolsartifactId>
            <scope>runtimescope>
            <optional>trueoptional>
        dependency>
        <dependency>
            <groupId>org.springframework.bootgroupId>
            <artifactId>spring-boot-configuration-processorartifactId>
            <optional>trueoptional>
        dependency>
        <dependency>
            <groupId>org.projectlombokgroupId>
            <artifactId>lombokartifactId>
            <optional>trueoptional>
        dependency>
        <dependency>
            <groupId>org.springframework.bootgroupId>
            <artifactId>spring-boot-starter-testartifactId>
            <scope>testscope>
        dependency>
        <dependency>
            <groupId>io.springfoxgroupId>
            <artifactId>springfox-swagger2artifactId>
            <version>2.9.2version>
        dependency>
        <dependency>
            <groupId>io.springfoxgroupId>
            <artifactId>springfox-swagger-uiartifactId>
            <version>2.9.2version>
        dependency>
        <dependency>
            <groupId>mysqlgroupId>
            <artifactId>mysql-connector-javaartifactId>
        dependency>
                <dependency>
            <groupId>com.google.zxinggroupId>
            <artifactId>coreartifactId>
            <version>3.3.0version>
        dependency>
        <dependency>
            <groupId>com.google.zxinggroupId>
            <artifactId>javaseartifactId>
            <version>3.3.0version>
        dependency>
    dependencies>
    <dependencyManagement>
        <dependencies>
            <dependency>
                <groupId>org.springframework.cloudgroupId>
                <artifactId>spring-cloud-dependenciesartifactId>
                <version>${spring-cloud.version}version>
                <type>pomtype>
                <scope>importscope>
            dependency>
        dependencies>
    dependencyManagement>
    <build>
        <plugins>
            <plugin>
                <groupId>org.springframework.bootgroupId>
                <artifactId>spring-boot-maven-pluginartifactId>
            plugin>
        plugins>
    build>

project>

配置文件：

user:
  qrcode:
    expire-time: 100000
    file-path: C:/Users/Administrator/Desktop/jar
    file-type: PNG
    height: 400
    width: 400

controller代码：

package com.hcx.jdbc.qr;

import java.io.IOException;
import java.util.UUID;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RestController;

import com.google.zxing.WriterException;

import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;

@RestController
@Slf4j
@Api("二维码服务")
public class QRCodeController {

    @Autowired
    QRCodeService qrCodeService;
    
    @Autowired
    QRCodeConfig qrCodeConfig;
    
    @PostMapping("/generator-qrcode")
    @ApiOperation("生成二维码")
    public String generatorQRCode(){
        String baseUrl = "跳转页面";
        String key = UUID.randomUUID().toString();
        String path = null;
        try {
            path = qrCodeService.generateQRCodeImage(baseUrl,key);
            QRCode qrCode = new QRCode();
            long expireTime = qrCode.getCreateTime() + qrCodeConfig.getExpireTime();
            qrCode.setExpireTime(expireTime);
            QRCodeCache.getInstance().putQRCodeCache(key, qrCode);
            return path;
        } catch (WriterException e) {
            log.info("生成二维码失败：" + e.getMessage());
            return path;
        } catch (IOException e) {
            log.info("生成二维码失败：" + e.getMessage());
            return path;
        }    
    }
    
    @PostMapping("/generator-scan")
    @ApiOperation("扫码二维码")
    public String scanQRCode(String uuid){
        QRCode qrCode = QRCodeCache.getInstance().getQRCodeCache(uuid);
        if(null ==qrCode){
            return "二维码已过期，请刷新获取";
        }
        if(qrCode.isScan()){
            return "二维码扫码过，请刷新获取";
        }
        if(qrCode.getExpireTime() < System.currentTimeMillis()){
            return "二维码已过期，请刷新获取";
        }
        qrCode.setScan(true);
        return "登录页面";
    }
    
    @PostMapping("/generator-hand")
    @ApiOperation("处理二维码")
    public String handQRCode(String uuid,String userName){
        QRCode qrCode = QRCodeCache.getInstance().getQRCodeCache(uuid);
        if(null ==qrCode){
            return "二维码已过期，请刷新获取";
        }

        if(qrCode.getExpireTime() < System.currentTimeMillis()){
            return "二维码已过期，请刷新获取";
        }
        QRCodeCache.getInstance().removeQRCodeCache(uuid);
        return "登录成功返回逻辑处理";
    }
}

service代码：

package com.hcx.jdbc.qr;

import java.io.File;
import java.io.IOException;
import java.nio.file.FileSystems;
import java.nio.file.Path;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import com.google.zxing.BarcodeFormat;
import com.google.zxing.WriterException;
import com.google.zxing.client.j2se.MatrixToImageWriter;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.qrcode.QRCodeWriter;

@Service
public class QRCodeService {
    
    @Autowired
    QRCodeConfig qrCodeConfig;
    
    public String generateQRCodeImage(String basePath,String key) throws WriterException, IOException{
        String text = basePath + "/" + key;
        String path = qrCodeConfig.getFilePath() + File.separator + key +"." + qrCodeConfig.getFileType().toLowerCase();
        
        QRCodeWriter qrCodeWriter = new QRCodeWriter();
        BitMatrix bitMatrix = qrCodeWriter.encode(text, BarcodeFormat.QR_CODE, qrCodeConfig.getWidth(), qrCodeConfig.getHeight());
        Path filePath = FileSystems.getDefault().getPath(path);
        MatrixToImageWriter.writeToPath(bitMatrix, qrCodeConfig.getFileType(), filePath);
        return path;
    }
}

配置类：

package com.hcx.jdbc.qr;

import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Configuration;

import lombok.Data;

@Configuration
@Data
@ConfigurationProperties(prefix="user.qrcode")
public class QRCodeConfig {

    private int width;
    private int height;
    private long expireTime;
    private String filePath;
    private String fileType;
}

持久代码：

package com.hcx.jdbc.qr;

import java.util.HashMap;
import java.util.Map;

public class QRCodeCache {

    public static Map cacheMap = new HashMap();
    
    public static QRCodeCache instance = null;
    
    public static QRCodeCache getInstance() {
        if(instance == null) {
            instance = new QRCodeCache();
        }
        return instance;
    }
    
    private QRCodeCache(){
        
    }

    public void putQRCodeCache(String key,QRCode qrCode){
        cacheMap.put(key, qrCode);
    }
    
    public QRCode getQRCodeCache(String key){
        return cacheMap.get(key);
    }
    
    public void removeQRCodeCache(String key){
        cacheMap.remove(key);
    }
    
}

二维码实体：

@Component
@Data
public class QRCode {

    private Long createTime = System.currentTimeMillis();
    
    private Long expireTime;
        
    private boolean isScan = false;
}