go 实现用户特权判断的例子

需求：实现用户特权，可以满足下面1个或多个条件

• 红名
• 达人
• vip会员
• vip超级会员
• 蓝钻用户
• 红钻用户

1、实现方法一：可以根据数据库字段，每个特权用一个字段，进行判断
缺点：每增加一个特权，要进行表字段扩展，代码修改。

2、实现方法二：用一个字段表示多个特权，这个方式需要用到位操作概念

假设定义一个8位无符号int，它的二进制表示如下，每个位置用0、1表示

红钻  蓝钻 超级vip  vip   达人   红名
 0        0        0        0        0        0        0        0

程序实现逻辑：

一、设置其中一位为1
用1 左移 n 与目标数做 或操作

二、判断其中一位是否等于1
用1 左移 n 与目标数做 &操作

三、清除其中一位
用1 左移 n 与目标数做 异或操作

package main

import (
	"fmt"
)

/*
需求：实现用户用户特权，可以满足下面1个或多个条件
1. 红名
2. 达人
3. Vip会员
4. vip超级会员
5. 蓝钻用户
6. 红钻用户

实现方法一：可以根据数据库字段，每个特权用一个字段，进行判断
缺点：每增加一个特权，要进行表字段扩展，代码修改

实现方法二：用一个字段表示多个特权，这个方式需要用到位操作概念

假设定义一个8位无符号int，它的二进制表示如下，每个位置用0、1表示
             红钻   蓝钻  超级vip  vip   达人   红名
0      0      0      0      0      0      0      0

一、设置其中一位为1
用1 左移 n 与目标数做 或操作

二、判断其中一位是否等于1
用1 左移 n 与目标数做 &操作

三、清除其中一位
用1 左移 n 与目标数做 异或操作

代码实现如下：
 */

const (
	HongMing  = 1 << 0
	DaRen     = 1 << 1
	Vip       = 1 << 2
	SuperVip  = 1 << 3
	LanZhuan  = 1 << 4
	HongZhuan = 1 << 5
)

type User struct {
	name string
	flag uint8
}

func set_flag(user User, isSet bool, flag uint8) User {
	if isSet == true {
		user.flag = user.flag | flag
	} else {
		user.flag = user.flag ^ flag
	}
	return user
}

func is_set_flag(user User, flag uint8) bool {
	result := user.flag & flag
	return result == flag
}

func auth() {
	var user User
	user.name = "test01"
	user.flag = 0

	result := is_set_flag(user, HongMing)
	fmt.Printf("user is hongming:%t\n", result)

	user = set_flag(user, true, HongMing)
	result = is_set_flag(user, HongMing)
	fmt.Printf("user is hongming:%t\n", result)

	//user = set_flag(user, false, HongMing)
	//result = is_set_flag(user, HongMing)
	//fmt.Printf("user is hongming:%t\n", result)

	fmt.Printf("%v\n", user.flag)

	result = is_set_flag(user, DaRen)
	fmt.Printf("user is DaRen:%t\n", result)

	user = set_flag(user, true, DaRen)
	result = is_set_flag(user, DaRen)
	fmt.Printf("user is DaRen:%t\n", result)

	//user = set_flag(user, false, DaRen)
	//result = is_set_flag(user, DaRen)
	//fmt.Printf("user is DaRen:%t\n", result)

	fmt.Printf("%v\n", user.flag)

	result = is_set_flag(user, Vip)
	fmt.Printf("user is Vip:%t\n", result)

	user = set_flag(user, true, Vip)
	result = is_set_flag(user, Vip)
	fmt.Printf("user is Vip:%t\n", result)

	//user = set_flag(user, false, Vip)
	//result = is_set_flag(user, Vip)
	//fmt.Printf("user is Vip:%t\n", result)

	fmt.Printf("%v\n", user.flag)

	result = is_set_flag(user, SuperVip)
	fmt.Printf("user is SuperVip:%t\n", result)

	user = set_flag(user, true, SuperVip)
	result = is_set_flag(user, SuperVip)
	fmt.Printf("user is SuperVip:%t\n", result)

	//user = set_flag(user, false, SuperVip)
	//result = is_set_flag(user, SuperVip)
	//fmt.Printf("user is SuperVip:%t\n", result)

	fmt.Printf("%v\n", user.flag)

	result = is_set_flag(user, LanZhuan)
	fmt.Printf("user is LanZhuan:%t\n", result)

	user = set_flag(user, true, LanZhuan)
	result = is_set_flag(user, LanZhuan)
	fmt.Printf("user is LanZhuan:%t\n", result)

	//user = set_flag(user, false, LanZhuan)
	//result = is_set_flag(user, LanZhuan)
	//fmt.Printf("user is LanZhuan:%t\n", result)

	fmt.Printf("%v\n", user.flag)

	result = is_set_flag(user, HongZhuan)
	fmt.Printf("user is HongZhuan:%t\n", result)

	user = set_flag(user, true, HongZhuan)
	result = is_set_flag(user, HongZhuan)
	fmt.Printf("user is HongZhuan:%t\n", result)

	//user = set_flag(user, false, HongZhuan)
	//result = is_set_flag(user, HongZhuan)
	//fmt.Printf("user is HongZhuan:%t\n", result)

	fmt.Printf("%v\n", user.flag)

}

func main() {
	auth()
}