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 (
	TreeNodes     = transform.TreeNodes
	TreeNode      = transform.TreeNode
	Tag           = transform.Tag
	UserRequest   = transform.UserRequest
	UserResponse  = transform.UserResponse
	TokenRequest  = transform.TokenRequest
	TokenResponse = transform.TokenResponse
	Empty         = transform.Empty

	Transform interface {
		GetUser(ctx context.Context, in *UserRequest) (*UserResponse, error)
		ParseToken(ctx context.Context, in *TokenRequest) (*TokenResponse, error)
		GetErpCityTree(ctx context.Context, in *Empty) (*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) ParseToken(ctx context.Context, in *TokenRequest) (*TokenResponse, error) {
	client := transform.NewTransformClient(m.cli.Conn())
	return client.ParseToken(ctx, in)
}

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