46 lines
1.0 KiB
Go
46 lines
1.0 KiB
Go
package user
|
|
|
|
import (
|
|
"go_dreamfactory/comm"
|
|
"go_dreamfactory/pb"
|
|
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
func (this *apiComp) ModifybgpCheck(session comm.IUserSession, req *pb.UserModifybgpReq) (code pb.ErrorCode) {
|
|
if req.BgpId == 0 {
|
|
code = pb.ErrorCode_ReqParameterError
|
|
}
|
|
return
|
|
}
|
|
|
|
func (this *apiComp) Modifybgp(session comm.IUserSession, req *pb.UserModifybgpReq) (code pb.ErrorCode, data proto.Message) {
|
|
if code = this.ModifybgpCheck(session, req); code != pb.ErrorCode_Success {
|
|
return
|
|
}
|
|
|
|
user := this.module.GetUser(session.GetUserId())
|
|
|
|
if user == nil {
|
|
code = pb.ErrorCode_UserSessionNobeing
|
|
return
|
|
}
|
|
|
|
update := map[string]interface{}{
|
|
"bgp": req.BgpId,
|
|
}
|
|
|
|
if err := this.module.modelUser.Change(session.GetUserId(), update); err != nil {
|
|
code = pb.ErrorCode_DBError
|
|
return
|
|
}
|
|
|
|
if err := session.SendMsg(string(this.module.GetType()), UserSubTypeModifyName, &pb.UserModifybgpResp{
|
|
Uid: session.GetUserId(),
|
|
BgpId: req.BgpId,
|
|
}); err != nil {
|
|
code = pb.ErrorCode_SystemError
|
|
}
|
|
return
|
|
}
|