[用户]实现对平台用户的增删改查

internal/rpc/services/service_user.go

@@ -0,0 +1,139 @@
+package services
+
+import (
+	"context"
+	"github.com/TeaOSLab/EdgeAPI/internal/db/models"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+)
+
+// 用户相关服务
+type UserService struct {
+	BaseService
+}
+
+// 创建用户
+func (this *UserService) CreateUser(ctx context.Context, req *pb.CreateUserRequest) (*pb.CreateUserResponse, error) {
+	_, err := this.ValidateAdmin(ctx, 0)
+	if err != nil {
+		return nil, err
+	}
+
+	userId, err := models.SharedUserDAO.CreateUser(req.Username, req.Password, req.Fullname, req.Mobile, req.Tel, req.Email, req.Remark, req.Source)
+	if err != nil {
+		return nil, err
+	}
+	return &pb.CreateUserResponse{UserId: userId}, nil
+}
+
+// 修改用户
+func (this *UserService) UpdateUser(ctx context.Context, req *pb.UpdateUserRequest) (*pb.RPCSuccess, error) {
+	_, err := this.ValidateAdmin(ctx, 0)
+	if err != nil {
+		return nil, err
+	}
+
+	err = models.SharedUserDAO.UpdateUser(req.UserId, req.Username, req.Password, req.Fullname, req.Mobile, req.Tel, req.Email, req.Remark, req.IsOn)
+	if err != nil {
+		return nil, err
+	}
+	return this.Success()
+}
+
+// 删除用户
+func (this *UserService) DeleteUser(ctx context.Context, req *pb.DeleteUserRequest) (*pb.RPCSuccess, error) {
+	_, err := this.ValidateAdmin(ctx, 0)
+	if err != nil {
+		return nil, err
+	}
+
+	_, err = models.SharedUserDAO.DisableUser(req.UserId)
+	if err != nil {
+		return nil, err
+	}
+	return this.Success()
+}
+
+// 计算用户数量
+func (this *UserService) CountAllEnabledUsers(ctx context.Context, req *pb.CountAllEnabledUsersRequest) (*pb.RPCCountResponse, error) {
+	_, err := this.ValidateAdmin(ctx, 0)
+	if err != nil {
+		return nil, err
+	}
+
+	count, err := models.SharedUserDAO.CountAllEnabledUsers(req.Keyword)
+	if err != nil {
+		return nil, err
+	}
+	return this.SuccessCount(count)
+}
+
+// 列出单页用户
+func (this *UserService) ListEnabledUsers(ctx context.Context, req *pb.ListEnabledUsersRequest) (*pb.ListEnabledUsersResponse, error) {
+	_, err := this.ValidateAdmin(ctx, 0)
+	if err != nil {
+		return nil, err
+	}
+
+	users, err := models.SharedUserDAO.ListEnabledUsers(req.Keyword)
+	if err != nil {
+		return nil, err
+	}
+
+	result := []*pb.User{}
+	for _, user := range users {
+		result = append(result, &pb.User{
+			Id:        int64(user.Id),
+			Username:  user.Username,
+			Fullname:  user.Fullname,
+			Mobile:    user.Mobile,
+			Tel:       user.Tel,
+			Email:     user.Email,
+			Remark:    user.Remark,
+			IsOn:      user.IsOn == 1,
+			CreatedAt: int64(user.CreatedAt),
+		})
+	}
+
+	return &pb.ListEnabledUsersResponse{Users: result}, nil
+}
+
+// 查询单个用户信息
+func (this *UserService) FindEnabledUser(ctx context.Context, req *pb.FindEnabledUserRequest) (*pb.FindEnabledUserResponse, error) {
+	_, err := this.ValidateAdmin(ctx, 0)
+	if err != nil {
+		return nil, err
+	}
+
+	user, err := models.SharedUserDAO.FindEnabledUser(req.UserId)
+	if err != nil {
+		return nil, err
+	}
+	if user == nil {
+		return &pb.FindEnabledUserResponse{User: nil}, nil
+	}
+	return &pb.FindEnabledUserResponse{User: &pb.User{
+		Id:        int64(user.Id),
+		Username:  user.Username,
+		Fullname:  user.Fullname,
+		Mobile:    user.Mobile,
+		Tel:       user.Tel,
+		Email:     user.Email,
+		Remark:    user.Remark,
+		IsOn:      user.IsOn == 1,
+		CreatedAt: int64(user.CreatedAt),
+	}}, nil
+}
+
+// 检查用户名是否存在
+func (this *UserService) CheckUsername(ctx context.Context, req *pb.CheckUsernameRequest) (*pb.CheckUsernameResponse, error) {
+	_, err := this.ValidateAdmin(ctx, 0)
+	if err != nil {
+		return nil, err
+	}
+
+	b, err := models.SharedUserDAO.ExistUser(req.UserId, req.Username)
+	if err != nil {
+		return nil, err
+	}
+	return &pb.CheckUsernameResponse{Exists: b}, nil
+}