i2-bill-erp/transformclient/transform.go

// Code generated by goctl. DO NOT EDIT!
// Source: transform.proto

//go:generate mockgen -destination ./transform_mock.go -package transformclient -source $GOFILE

package transformclient

import (
	"context"

	"git.i2edu.net/i2/i2-bill-erp/transform"

	"git.i2edu.net/i2/go-zero/zrpc"
)

type (
	UserRequest  = transform.UserRequest
	UserResponse = transform.UserResponse

	Transform interface {
		GetUser(ctx context.Context, in *UserRequest) (*UserResponse, error)
		GetErpCityTree(ctx context.Context, in *transform.Empty) (*transform.TreeNodes, error)
	}

	defaultTransform struct {
		cli zrpc.Client
	}
)

func NewTransform(cli zrpc.Client) Transform {
	return &defaultTransform{
		cli: cli,
	}
}

func (m *defaultTransform) GetUser(ctx context.Context, in *UserRequest) (*UserResponse, error) {
	client := transform.NewTransformClient(m.cli.Conn())
	return client.GetUser(ctx, in)
}

func (m *defaultTransform) GetErpCityTree(ctx context.Context, in *transform.Empty) (*transform.TreeNodes, error) {
	client := transform.NewTransformClient(m.cli.Conn())
	return client.GetErpCityTree(ctx, in)
}