61 lines
2.0 KiB
Go
61 lines
2.0 KiB
Go
// Code generated by goctl. DO NOT EDIT.
|
|
// goctl 1.7.3
|
|
// Source: aisvc.proto
|
|
|
|
package aiservice
|
|
|
|
import (
|
|
"context"
|
|
|
|
"schisandra-album-cloud-microservices/app/aisvc/rpc/pb"
|
|
|
|
"github.com/zeromicro/go-zero/zrpc"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
type (
|
|
CaffeClassificationRequest = pb.CaffeClassificationRequest
|
|
CaffeClassificationResponse = pb.CaffeClassificationResponse
|
|
FaceRecognitionRequest = pb.FaceRecognitionRequest
|
|
FaceRecognitionResponse = pb.FaceRecognitionResponse
|
|
TfClassificationRequest = pb.TfClassificationRequest
|
|
TfClassificationResponse = pb.TfClassificationResponse
|
|
|
|
AiService interface {
|
|
// FaceRecognition
|
|
FaceRecognition(ctx context.Context, in *FaceRecognitionRequest, opts ...grpc.CallOption) (*FaceRecognitionResponse, error)
|
|
// TfClassification
|
|
TfClassification(ctx context.Context, in *TfClassificationRequest, opts ...grpc.CallOption) (*TfClassificationResponse, error)
|
|
// CaffeClassification
|
|
CaffeClassification(ctx context.Context, in *CaffeClassificationRequest, opts ...grpc.CallOption) (*CaffeClassificationResponse, error)
|
|
}
|
|
|
|
defaultAiService struct {
|
|
cli zrpc.Client
|
|
}
|
|
)
|
|
|
|
func NewAiService(cli zrpc.Client) AiService {
|
|
return &defaultAiService{
|
|
cli: cli,
|
|
}
|
|
}
|
|
|
|
// FaceRecognition
|
|
func (m *defaultAiService) FaceRecognition(ctx context.Context, in *FaceRecognitionRequest, opts ...grpc.CallOption) (*FaceRecognitionResponse, error) {
|
|
client := pb.NewAiServiceClient(m.cli.Conn())
|
|
return client.FaceRecognition(ctx, in, opts...)
|
|
}
|
|
|
|
// TfClassification
|
|
func (m *defaultAiService) TfClassification(ctx context.Context, in *TfClassificationRequest, opts ...grpc.CallOption) (*TfClassificationResponse, error) {
|
|
client := pb.NewAiServiceClient(m.cli.Conn())
|
|
return client.TfClassification(ctx, in, opts...)
|
|
}
|
|
|
|
// CaffeClassification
|
|
func (m *defaultAiService) CaffeClassification(ctx context.Context, in *CaffeClassificationRequest, opts ...grpc.CallOption) (*CaffeClassificationResponse, error) {
|
|
client := pb.NewAiServiceClient(m.cli.Conn())
|
|
return client.CaffeClassification(ctx, in, opts...)
|
|
}
|