handler.go

/*
 * Copyright 2024 CloudWeGo Authors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package main

import (
	"context"

	"gorm.io/gorm"

	"github.com/bytedance/gopkg/cloud/metainfo"
	"github.com/cloudwego/kitex-examples/bizdemo/kitex_swagger_gen/dao/mysql"
	user "github.com/cloudwego/kitex-examples/bizdemo/kitex_swagger_gen/kitex_gen/user"
	"github.com/cloudwego/kitex-examples/bizdemo/kitex_swagger_gen/model"
	"github.com/cloudwego/kitex-examples/bizdemo/kitex_swagger_gen/pack"
)

// UserServiceImpl implements the last service interface defined in the IDL.
type UserServiceImpl struct{}

// UpdateUser implements the UserServiceImpl interface.
func (s *UserServiceImpl) UpdateUser(ctx context.Context, req *user.UpdateUserRequest) (resp *user.UpdateUserResponse, err error) {
	u := &model.User{
		Model:     gorm.Model{ID: uint(req.UserId)},
		Name:      req.Name,
		Gender:    int64(req.Gender),
		Age:       req.Age,
		Introduce: req.Introduce,
	}

	t2, _ := metainfo.GetValue(ctx, "k")
	downstreamResp, _ := svr2Cli.QueryUser(ctx, &user.QueryUserRequest{})
	metainfo.SendBackwardValue(ctx, "反向元信息测试key", "反向元信息测试value")

	if err = mysql.UpdateUser(u); err != nil {
		resp.Msg = err.Error()
		resp.Code = user.Code_DBErr
		return
	}

	return &user.UpdateUserResponse{
		Code: user.Code_Success,
		Msg:  "当前rpc响应->" + req.String() + " 当前rpc临时元信息->" + t2 + " 下游rpc的响应->" + downstreamResp.String(),
	}, nil
}

// DeleteUser implements the UserServiceImpl interface.
func (s *UserServiceImpl) DeleteUser(ctx context.Context, req *user.DeleteUserRequest) (resp *user.DeleteUserResponse, err error) {
	resp = new(user.DeleteUserResponse)

	if err = mysql.DeleteUser(req.UserId); err != nil {
		resp.Msg = err.Error()
		resp.Code = user.Code_DBErr
		return
	}

	return &user.DeleteUserResponse{
		Code: user.Code_Success,
		Msg:  req.String(),
	}, nil
}

// QueryUser implements the UserServiceImpl interface.
func (s *UserServiceImpl) QueryUser(ctx context.Context, req *user.QueryUserRequest) (resp *user.QueryUserResponse, err error) {
	resp = new(user.QueryUserResponse)

	users, total, err := mysql.QueryUser(req.Keyword, req.Page, req.PageSize)
	if err != nil {
		resp.Msg = err.Error()
		resp.Code = user.Code_DBErr
	}

	resp.Total = total
	resp.Users = pack.Users(users)
	resp.Code = user.Code_Success
	return
}

// CreateUser implements the UserServiceImpl interface.
func (s *UserServiceImpl) CreateUser(ctx context.Context, req *user.CreateUserRequest) (resp *user.CreateUserResponse, err error) {
	resp = new(user.CreateUserResponse)

	if err = mysql.CreateUser([]*model.User{
		{
			Name:      req.Name,
			Gender:    int64(req.Gender),
			Age:       req.Age,
			Introduce: req.Introduce,
		},
	}); err != nil {
		resp.Msg = err.Error()
		resp.Code = user.Code_DBErr
	}

	resp.Code = user.Code_Success
	return
}