handler.go 1.33 KB
Newer Older
kzkzzzz's avatar
kzkzzzz committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
package server

import (
	"context"
	"github.com/spf13/cast"
	"gomicro-base/api/demo"
	"gomicro-base/service/demo/internal/model"
	"gomicro-base/service/demo/internal/service"
	"google.golang.org/protobuf/types/known/timestamppb"
)

// 检查接口实现
var _ demo.DemoHandler = &handler{}

// handler类似控制器
type handler struct {
	svc *service.DemoService
}

func (h *handler) GetDemo(ctx context.Context, req *demo.DemoReq, resp *demo.DemoResp) error {
	getDemo, err := h.svc.GetDemo(ctx, req)
	if err != nil {
		return err
	}
kzkzzzz's avatar
kzkzzzz committed
25
	*resp = *h.convertUserModel(getDemo)
kzkzzzz's avatar
kzkzzzz committed
26 27 28 29 30 31 32
	return nil
}

func (h *handler) ListDemo(ctx context.Context, req *demo.ListDemoReq, resp *demo.ListDemoResp) error {
	if req.Limit <= 0 {
		req.Limit = 15
	}
kzkzzzz's avatar
kzkzzzz committed
33

kzkzzzz's avatar
kzkzzzz committed
34 35 36 37 38 39 40
	listDemo, err := h.svc.ListDemo(ctx, req)
	if err != nil {
		return err
	}

	list := make([]*demo.DemoResp, 0)
	for _, v := range listDemo {
kzkzzzz's avatar
kzkzzzz committed
41
		c := *h.convertUserModel(v)
kzkzzzz's avatar
kzkzzzz committed
42 43 44 45 46 47 48
		list = append(list, &c)
	}
	resp.List = list
	resp.Limit = req.Limit
	return nil
}

kzkzzzz's avatar
kzkzzzz committed
49 50
func (h *handler) convertUserModel(users *model.User) (resp *demo.DemoResp) {
	r := &demo.DemoResp{
kzkzzzz's avatar
kzkzzzz committed
51 52 53 54 55 56 57 58
		Id:        cast.ToString(users.Id),
		Username:  users.Username,
		Email:     users.Email,
		Avatar:    users.Avatar,
		Status:    int32(users.Status),
		CreatedAt: timestamppb.New(users.CreatedAt),
		UpdatedAt: timestamppb.New(users.UpdatedAt),
	}
kzkzzzz's avatar
kzkzzzz committed
59
	return r
kzkzzzz's avatar
kzkzzzz committed
60
}