Go基础之--位操作中你所不知道的用法

 1 package main
 2 
 3 import (
 4     "fmt"
 5 )
 6 
 7 type users struct {
 8     name   string
 9     flag   uint8
10     vip    bool
11     svip   bool
12     blue   bool
13     red    bool
14     yellow bool
15 }
16 
17 func setVip(user users) users {
18     user.vip = true
19     return user
20 }
21 
22 func isVip(user users) {
23     if user.vip {
24         fmt.Println("user is vip")
25     } else {
26         fmt.Println("user is not vip")
27     }
28 }
29 
30 func binaryTest() {
31     var user users
32     user.name = "test01"
33     user.vip = true
34     isVip(user)
35     user.vip = false
36     isVip(user)
37 }
38 
39 func main() {
40     binaryTest()
41 
42 }

 1 package main
 2 
 3 import (
 4     "fmt"
 5 )
 6 
 7 type users struct {
 8     name string
 9     flag uint8
10 }
11 
12 // 这里通过位移的方式默认00000 从左边一次为vip,svip,blue,red,yellow
13 const (
14     vip   = 1
15     svip  = (1 << 1)
16     blue  = (1 << 2)
17     red   = (1 << 3)
18     yello = (1 << 4)
19 )
20 
21 // setFlag 用于设置用户开通了哪些特权
22 func setFlag(user users, isSet bool, typeFlag uint8) users {
23     if isSet == true {
24         user.flag = user.flag | typeFlag
25     } else {
26         user.flag = user.flag ^ typeFlag
27     }
28     return user
29 }
30 
31 //isFlag 用于判断用户是否开通某项特权
32 func isFlag(user users, typeFlag uint8) bool {
33     result := user.flag & typeFlag
34     return result == typeFlag
35 }
36 
37 func binaryTest() {
38     var user users
39     user.name = "coder"
40     user.flag = 0
41 
42     //判断用户是否是vip
43     result := isFlag(user, vip)
44     fmt.Printf("user is Vip:%t\n", result)
45 
46     //给用户开通vip,并看用户是否开通vip
47     user = setFlag(user, true, vip)
48     result = isFlag(user, vip)
49     fmt.Printf("user is Vip:%t\n", result)
50 
51     //取消用户的vip，并查看用户是否还是vip
52     user = setFlag(user, false, vip)
53     result = isFlag(user, vip)
54     fmt.Printf("user is Vip:%t\n", result)
55 }
56 
57 func main() {
58     binaryTest()
59 }