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.0</modelVersion>
    <parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>2.1.12.RELEASE</version>
        <relativePath /> <!-- lookup parent from repository -->
    </parent>
    <groupId>com.hcx</groupId>
    <artifactId>spring-jdbc</artifactId>
    <version>0.0.1-SNAPSHOT</version>
    <name>spring-jdbc</name>
    <description>spring-jdbc</description>

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

    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>

        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-devtools</artifactId>
            <scope>runtime</scope>
            <optional>true</optional>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-configuration-processor</artifactId>
            <optional>true</optional>
        </dependency>
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <optional>true</optional>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
            <scope>test</scope>
        </dependency>
        <dependency>
            <groupId>io.springfox</groupId>
            <artifactId>springfox-swagger2</artifactId>
            <version>2.9.2</version>
        </dependency>
        <dependency>
            <groupId>io.springfox</groupId>
            <artifactId>springfox-swagger-ui</artifactId>
            <version>2.9.2</version>
        </dependency>
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
        </dependency>
                <dependency>
            <groupId>com.google.zxing</groupId>
            <artifactId>core</artifactId>
            <version>3.3.0</version>
        </dependency>
        <dependency>
            <groupId>com.google.zxing</groupId>
            <artifactId>javase</artifactId>
            <version>3.3.0</version>
        </dependency>
    </dependencies>
    <dependencyManagement>
        <dependencies>
            <dependency>
                <groupId>org.springframework.cloud</groupId>
                <artifactId>spring-cloud-dependencies</artifactId>
                <version>${spring-cloud.version}</version>
                <type>pom</type>
                <scope>import</scope>
            </dependency>
        </dependencies>
    </dependencyManagement>
    <build>
        <plugins>
            <plugin>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-maven-plugin</artifactId>
            </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<String, QRCode> cacheMap = new HashMap<String, QRCode>();
    
    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;
}