wxgzh

user/user.go

@@ -14,11 +14,13 @@ package user
 
 import (
 	"github.com/jageros/hawox/attribute"
+	"sync"
 	"time"
 )
 
 type User struct {
 	attr *attribute.AttrMgr
+	mx   sync.RWMutex
 }
 
 func newUser(openId string) (*User, error) {
@@ -38,6 +40,8 @@ func (u *User) OpenID() string {
 }
 
 func (u *User) Codes(isFund bool) []string {
+	u.mx.RLock()
+	defer u.mx.RUnlock()
 	key := "stock"
 	if isFund {
 		key = "fund"
@@ -56,6 +60,8 @@ func (u *User) Codes(isFund bool) []string {
 
 // HasSubscribed 查询用户是否订阅此票
 func (u *User) HasSubscribed(isFund bool, code string) bool {
+	u.mx.RLock()
+	defer u.mx.RUnlock()
 	key := "stock"
 	if isFund {
 		key = "fund"
@@ -73,6 +79,8 @@ func (u *User) HasSubscribed(isFund bool, code string) bool {
 
 // Subscribe 订阅股票或基金
 func (u *User) Subscribe(isFund bool, codes ...string) {
+	u.mx.Lock()
+	defer u.mx.Unlock()
 	key := "stock"
 	if isFund {
 		key = "fund"
@@ -95,6 +103,8 @@ func (u *User) Subscribe(isFund bool, codes ...string) {
 
 // UnSubscribe 取消订阅股票或基金
 func (u *User) UnSubscribe(isFund bool, codes ...string) {
+	u.mx.Lock()
+	defer u.mx.Unlock()
 	key := "stock"
 	if isFund {
 		key = "fund"