mmm-go / opp · 提交

转到一个项目

作者 yangfu
提交 552ef014c063707719bf901576142d9403bcfdd7 1 个父辈 3dea27d5

配置修改

内嵌 并排对比
正在显示 7 个修改的文件 包含 37 行增加 和 33 行删除
... ... @@ -90,6 +90,7 @@ func GetRequestHeader(ctx *context.Context) *protocol.RequestHeader {
h.Uid, _ = strconv.ParseInt(ctx.Input.Header("x-mmm-uid"), 10, 64)
}
h.CompanyId, _ = strconv.ParseInt(ctx.Input.Header("x-mmm-cid"), 10, 64)
h.UserId, _ = strconv.ParseInt(ctx.Input.Header("x-mmm-id"), 10, 64)
log.Debug(common.AssertJson(h))
return h
}
... ... @@ -103,7 +104,7 @@ func FilterComm(ctx *context.Context) {
//统计
MetricCounter(ctx)
if beego.BConfig.RunMode == "dev" && (ctx.Input.Header("x-mmm-uid")!="" || ctx.Input.Header("uid")!=""){
if beego.BConfig.RunMode == "dev" && (ctx.Input.Header("x-mmm-uid") != "" || ctx.Input.Header("uid") != "") {
return
}
//1.检查签名
... ... @@ -160,7 +161,7 @@ func CheckToken(ctx *context.Context) (result bool) {
msg *protocol.ResponseMessage
)
token := ctx.Input.Header("x-mmm-accesstoken")
if token==""{
if token == "" {
if strings.HasSuffix(ctx.Request.RequestURI, "login") ||
strings.HasSuffix(ctx.Request.RequestURI, "accessToken") ||
strings.HasSuffix(ctx.Request.RequestURI, "refreshToken") {
... ... @@ -183,6 +184,7 @@ func CheckToken(ctx *context.Context) (result bool) {
//设置附加数据
ctx.Request.Header.Add("x-mmm-uid", fmt.Sprintf("%v", rsp.UserInfo.UserId))
ctx.Request.Header.Add("x-mmm-cid", fmt.Sprintf("%v", rsp.UserInfo.CurrentCompanyId))
ctx.Request.Header.Add("x-mmm-id", fmt.Sprintf("%v", rsp.UserInfo.CurrentUserCompanyId))
}
}
return
... ...
... ... @@ -25,6 +25,7 @@ type UserAuth struct {
CreateAt time.Time `orm:"column(create_at);type(timestamp)" description:"创建时间"`
UpdateAt time.Time `orm:"column(update_at);type(timestamp)" description:"更新时间"`
CurrentCompanyId int64 `orm:"column(current_company_id)" description:"当前登录的公司id "`
CurrentUserCompanyId int64 `orm:"column(current_user_company_id)" description:"当前登录的公司 user_company.id "`
}
func (t *UserAuth) TableName() string {
... ... @@ -180,10 +181,10 @@ func GetUserAuthByRefreshToken(refreshToken string) (v *UserAuth, err error) {
return nil, err
}
func GetUserAuthByToken(refreshToken string) (v *UserAuth, err error) {
func GetUserAuthByToken(token string) (v *UserAuth, err error) {
o := orm.NewOrm()
sql := "select * from user_auth where access_token=? and access_token_exp >= Now()"
if err = o.Raw(sql, refreshToken).QueryRow(&v); err == nil {
if err = o.Raw(sql, token).QueryRow(&v); err == nil {
return v, nil
}
return nil, err
... ...
... ... @@ -8,7 +8,7 @@ import (
)
type UserCompany struct {
Id int `orm:"column(id);auto" description:"唯一标识"`
Id int64 `orm:"column(id)" description:"唯一标识"`
CompanyId int64 `orm:"column(company_id)" description:"表company.id 公司编号"`
UserId int64 `orm:"column(user_id)" description:"表user.id 用户编号"`
DepartmentId int `orm:"column(department_id)" description:"部门id"`
... ... @@ -37,7 +37,7 @@ func AddUserCompany(m *UserCompany) (id int64, err error) {
// GetUserCompanyById retrieves UserCompany by Id. Returns error if
// Id doesn't exist
func GetUserCompanyById(id int) (v *UserCompany, err error) {
func GetUserCompanyById(id int64) (v *UserCompany, err error) {
o := orm.NewOrm()
v = &UserCompany{Id: id}
if err = o.Read(v); err == nil {
... ... @@ -63,7 +63,7 @@ func UpdateUserCompanyById(m *UserCompany) (err error) {
// DeleteUserCompany deletes UserCompany by Id and returns error if
// the record to be deleted doesn't exist
func DeleteUserCompany(id int) (err error) {
func DeleteUserCompany(id int64) (err error) {
o := orm.NewOrm()
v := UserCompany{Id: id}
// ascertain id exists in the database
... ...
... ... @@ -7,8 +7,8 @@ const (
LoginTypeSmdcode = "signInCaptcha"
)
const(
DeviceType =1
const (
DeviceType = 1
)
const TokenExpire = 3600
... ... @@ -24,8 +24,9 @@ type RequestHeader struct {
DeviceType int
AppProject string
AccessToken string
Uid int64
Uid int64 //用户基本信息Id
CompanyId int64
UserId int64 //UserId 唯一标识,唯一关联所有用户信息（=user_company.id）
}
/*Login */
... ...
... ... @@ -15,18 +15,18 @@ import (
func GetUserBaseInfoAggregation(uid int64, companyId int64) (v *protocol.UserBaseInfoAggregation, err error) {
v = &protocol.UserBaseInfoAggregation{}
var wg sync.WaitGroup
if v.User, err = repository.User.GetUsersById(uid); err != nil {
if v.UserCompany, err = models.GetUserCompanyByUserId(uid, companyId); err != nil {
log.Error(err)
return
}
if v.UserCompany, err = repository.UserCompany.GetUserCompanyByUserId(uid, companyId); err != nil {
if v.User, err = models.GetUsersById(v.UserCompany.UserId); err != nil {
log.Error(err)
return
}
wg.Add(3)
go func() {
defer wg.Done()
if v.Company, err = repository.Company.GetCompanyById(int64(v.UserCompany.CompanyId)); err != nil {
if v.Company, err = models.GetCompanyById(int64(v.UserCompany.CompanyId)); err != nil {
log.Error(err)
return
}
... ... @@ -34,7 +34,7 @@ func GetUserBaseInfoAggregation(uid int64, companyId int64) (v *protocol.UserBas
go func() {
defer wg.Done()
if err = repository.UserDepartment.GetUserDepartment(uid, companyId, &v.Department); err != nil {
if err = models.GetUserDepartments(uid, companyId, &v.Department); err != nil {
log.Error(err)
return
}
... ... @@ -42,7 +42,7 @@ func GetUserBaseInfoAggregation(uid int64, companyId int64) (v *protocol.UserBas
go func() {
defer wg.Done()
if err = repository.UserPosition.GetUserPositions(uid, companyId, &v.Position); err != nil {
if err = models.GetUserPositions(uid, companyId, &v.Position); err != nil {
log.Error(err)
return
}
... ...
... ... @@ -41,7 +41,7 @@ func IComment(header *protocol.RequestHeader, request *protocol.ICommentRequest)
}
newComment := &models.Comment{
Id: idgen.Next(),
UserId: header.Uid,
UserId: header.UserId,
SourceType: int8(request.SourceType),
Content: request.Content,
CreateAt: time.Now(),
... ... @@ -74,7 +74,7 @@ func IComments(header *protocol.RequestHeader, request *protocol.ICommentsReques
baseUserInfo *protocol.BaseUserInfo
total int
)
if comments, total, err = repository.Comment.GetComments(header.Uid, protocol.SourceType_Chance, 0, request.LastId, request.PageSize); err != nil {
if comments, total, err = repository.Comment.GetComments(header.UserId, protocol.SourceType_Chance, 0, request.LastId, request.PageSize); err != nil {
log.Error(err)
return
}
... ... @@ -112,7 +112,7 @@ func Comments(header *protocol.RequestHeader, request *protocol.CommentsRequest)
total int
exists bool
)
if comments, total, err = repository.Comment.GetComments(header.Uid, request.SourceType, request.SourceId, request.LastId, request.PageSize); err != nil {
if comments, total, err = repository.Comment.GetComments(header.UserId, request.SourceType, request.SourceId, request.LastId, request.PageSize); err != nil {
log.Error(err)
return
}
... ... @@ -134,7 +134,7 @@ func Comments(header *protocol.RequestHeader, request *protocol.CommentsRequest)
ZanTotal: comment.ZanTotal,
CommentTotal: comment.CommentTotal,
}
if exists, _ = repository.ChanceFavorite.ExitsChanceFavorite(header.Uid, header.CompanyId, comment.Id, protocol.MarkFlag_Zan); exists {
if exists, _ = repository.ChanceFavorite.ExitsChanceFavorite(header.UserId, header.CompanyId, comment.Id, protocol.MarkFlag_Zan); exists {
item.IsZan = 1
}
rsp.Comments = append(rsp.Comments, item)
... ... @@ -167,7 +167,7 @@ func CommentDetailsSingle(header *protocol.RequestHeader, request *protocol.Comm
ZanTotal: comment.ZanTotal,
CommentTotal: comment.CommentTotal,
}
if exists, _ = repository.ChanceFavorite.ExitsChanceFavorite(header.Uid, header.CompanyId, comment.Id, protocol.MarkFlag_Zan); exists {
if exists, _ = repository.ChanceFavorite.ExitsChanceFavorite(header.UserId, header.CompanyId, comment.Id, protocol.MarkFlag_Zan); exists {
rsp.Comment.IsZan = 1
}
return
... ...
... ... @@ -183,15 +183,15 @@ func UserCompanys(header *protocol.RequestHeader, request *protocol.UserCompanys
companys []*models.Company
)
rsp = &protocol.UserCompanysResponse{}
if companys,err =models.GetCompanyByPermission(header.Uid);err!=nil{
if companys, err = models.GetCompanyByPermission(header.Uid); err != nil {
log.Error(err)
err = protocol.NewErrWithMessage(4201) //找不到这家公司
return
}
for i:=range companys{
rsp.Companys =append(rsp.Companys,protocol.Company{
Id:companys[i].Id,
Name:companys[i].Name,
for i := range companys {
rsp.Companys = append(rsp.Companys, protocol.Company{
Id: companys[i].Id,
Name: companys[i].Name,
})
}
return
... ... @@ -213,11 +213,11 @@ func SwitchCompany(header *protocol.RequestHeader, request *protocol.SwitchCompa
log.Error(err)
return
}
if auth.CurrentCompanyId == request.CompanyId{
log.Error(fmt.Sprintf("uid:%v 当前公司已经是:%v",header.Uid,request.CompanyId))
if auth.CurrentCompanyId == request.CompanyId {
log.Error(fmt.Sprintf("uid:%v 当前公司已经是:%v", header.Uid, request.CompanyId))
return
}
if err = utils.UpdateTableByMap(&models.UserAuth{Id: auth.Id}, map[string]interface{}{"CurrentCompanyId": company.CompanyId}); err != nil {
if err = utils.UpdateTableByMap(&models.UserAuth{Id: auth.Id}, map[string]interface{}{"CurrentCompanyId": company.CompanyId, "CurrentCompanyUserId": company.Id}); err != nil {
log.Error(err)
return
}
... ... @@ -228,14 +228,14 @@ func SwitchCompany(header *protocol.RequestHeader, request *protocol.SwitchCompa
func UserInfo(header *protocol.RequestHeader, request *protocol.UserInfoRequest) (rsp *protocol.UserInfoResponse, err error) {
var (
companyId = header.CompanyId
company *models.UserCompany
userCompany *models.UserCompany
baseInfo *protocol.BaseUserInfo
userAuth *models.UserAuth
user *models.User
)
if companyId == 0 {
if company, err = models.GetUserCompanyFirst(header.Uid); err != nil {
if userCompany, err = models.GetUserCompanyFirst(header.Uid); err != nil {
log.Error(err)
return
}
... ... @@ -243,17 +243,17 @@ func UserInfo(header *protocol.RequestHeader, request *protocol.UserInfoRequest)
log.Error(err)
return
}
if err = utils.UpdateTableByMap(&models.UserAuth{Id: userAuth.Id}, map[string]interface{}{"CurrentCompanyId": companyId}); err != nil {
if err = utils.UpdateTableByMap(&models.UserAuth{Id: userAuth.Id}, map[string]interface{}{"CurrentCompanyId": userCompany.CompanyId, "CurrentUserCompanyId": userCompany.Id}); err != nil {
log.Error(err)
return
}
companyId = int64(company.Id)
companyId = int64(userCompany.CompanyId)
}
if user, err = models.GetUsersById(header.Uid); err != nil {
log.Error(err)
return
}
if baseInfo, err = agg.GetUserBaseInfo(header.Uid, companyId); err != nil {
if baseInfo, err = agg.GetUserBaseInfo(header.UserId, companyId); err != nil {
log.Error(err)
return
}
... ...