教你巧用mysql位運算解決多選值儲存的問題

2022-02-08 13:02:00

一．問題場景

工作中經常遇到多選值儲存問題，例如：使用者有多種認證方式，密碼認證、簡訊認證、掃碼認證等，一個使用者可能只開啟了其中某幾種認證方式。

二. 場景分析

比較容易理解的兩種實現方式，多欄位儲存、單個欄位拼接儲存。

1.多欄位儲存

每種認證方式用一個欄位儲存，0表示未開啟，1表示已開啟。

缺點：每增加一種認證方式都需要新增一個表欄位，擴充套件性差。

2.單欄位拼接

單欄位儲存，已開啟的認證方式用逗號(或其他分割符)拼接。例如：開始了密碼認證和簡訊認證，則儲存為：密碼認證,簡訊認證。

缺點：不利於查詢，需要使用模糊查詢，搞不好會影響效能。

三．巧用位運算

1.概述

參考Linux許可權控制思路，將每種認證方式對應到二進位制位中，例如：密碼認證–10000000，簡訊認證–01000000，掃碼認證–00100000，然後將其轉換成10進位制，密碼認證–1，簡訊認證–2，掃碼認證–4。Mysql儲存時使用單欄位(auth_method)int型別儲存，如果開啟了多種認證方式將多種認證方式對應的列舉數值相加後儲存，例如開啟了密碼認證和簡訊認證，則儲存為3(1+2)。

2.sql查詢

## 例1：判斷使用者是否開啟了密碼認證--1 (滿足條件時返回查詢結果，沒有滿足條件時返回為空)
Select * from user where auth_method & 1;

## 例2：判斷使用者是否開啟了密碼認證 + 簡訊認證 (1+2)
Select * from user where auth_method & 3;

## 例2：判斷使用者是否開啟了密碼認證 + 簡訊認證 + 掃碼認證 (1+2+4)
Select * from user where auth_method & 7;

3.Java解析與計算

import com.google.common.collect.Lists;
import lombok.Getter;
import org.springframework.util.CollectionUtils;

import java.util.Arrays;
import java.util.List;

@Getter
public enum AuthMethodEnum {

    PASSWORD(1, "密碼認證"),
    SMS(2, "簡訊認證"),
    QR_CODE(4, "掃碼認證");

    private Integer method;

    private String name;

    AuthMethodEnum(Integer method, String name) {
        this.method = method;
        this.name = name;
    }

    /**
     * 將mysql儲存值解析成多種認證方式
     * @param method
     * @return
     */
    public static List<Integer> parseAuthMethod(Integer method) {
        List<Integer> list = Lists.newArrayList();
        if (null == method) {
            return list;
        }
        AuthMethodEnum[] arr = AuthMethodEnum.values();
        // 需要先將method從大到小排序
        Arrays.sort(arr, (o1, o2) -> {
            if (o1.method > o2.method) {
                return -1;
            } else {
                return 0;
            }
        });
        for (AuthMethodEnum e : arr) {
            if (method >= e.method) {
                list.add(e.method);
                method = method - e.method;
            }
        }
        return list;
    }

    /**
     * 將任意種認證方式計算後得到儲存值
     * @param methods
     * @return
     */
    public static Integer calculateAuthMethod(List<Integer> methods) {
        if (CollectionUtils.isEmpty(methods)) {
            return 0;
        }
        return methods.stream().mapToInt(p -> p).sum();
    }

    public static void main(String[] args) {
        System.out.println(parseAuthMethod(8));
    }
}