✨ add face recognition

2025-01-22 10:36:28 +08:00
parent eab806fb9b
commit c6af9a0461
47 changed files with 3621 additions and 454 deletions
--- a/app/aisvc/rpc/pb/aisvc.pb.go
+++ b/app/aisvc/rpc/pb/aisvc.pb.go
@@ -0,0 +1,192 @@
+// Code generated by protoc-gen-go. DO NOT EDIT.
+// versions:
+// 	protoc-gen-go v1.35.1
+// 	protoc        v3.19.4
+// source: aisvc.proto
+
+package pb
+
+import (
+	protoreflect "google.golang.org/protobuf/reflect/protoreflect"
+	protoimpl "google.golang.org/protobuf/runtime/protoimpl"
+	reflect "reflect"
+	sync "sync"
+)
+
+const (
+	// Verify that this generated code is sufficiently up-to-date.
+	_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion)
+	// Verify that runtime/protoimpl is sufficiently up-to-date.
+	_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
+)
+
+type FaceRecognitionRequest struct {
+	state         protoimpl.MessageState
+	sizeCache     protoimpl.SizeCache
+	unknownFields protoimpl.UnknownFields
+
+	Face   []byte `protobuf:"bytes,1,opt,name=face,proto3" json:"face,omitempty"`
+	UserId string `protobuf:"bytes,2,opt,name=user_id,json=userId,proto3" json:"user_id,omitempty"`
+}
+
+func (x *FaceRecognitionRequest) Reset() {
+	*x = FaceRecognitionRequest{}
+	mi := &file_aisvc_proto_msgTypes[0]
+	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+	ms.StoreMessageInfo(mi)
+}
+
+func (x *FaceRecognitionRequest) String() string {
+	return protoimpl.X.MessageStringOf(x)
+}
+
+func (*FaceRecognitionRequest) ProtoMessage() {}
+
+func (x *FaceRecognitionRequest) ProtoReflect() protoreflect.Message {
+	mi := &file_aisvc_proto_msgTypes[0]
+	if x != nil {
+		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+		if ms.LoadMessageInfo() == nil {
+			ms.StoreMessageInfo(mi)
+		}
+		return ms
+	}
+	return mi.MessageOf(x)
+}
+
+// Deprecated: Use FaceRecognitionRequest.ProtoReflect.Descriptor instead.
+func (*FaceRecognitionRequest) Descriptor() ([]byte, []int) {
+	return file_aisvc_proto_rawDescGZIP(), []int{0}
+}
+
+func (x *FaceRecognitionRequest) GetFace() []byte {
+	if x != nil {
+		return x.Face
+	}
+	return nil
+}
+
+func (x *FaceRecognitionRequest) GetUserId() string {
+	if x != nil {
+		return x.UserId
+	}
+	return ""
+}
+
+type FaceRecognitionResponse struct {
+	state         protoimpl.MessageState
+	sizeCache     protoimpl.SizeCache
+	unknownFields protoimpl.UnknownFields
+
+	FaceId int64 `protobuf:"varint,1,opt,name=face_id,json=faceId,proto3" json:"face_id,omitempty"`
+}
+
+func (x *FaceRecognitionResponse) Reset() {
+	*x = FaceRecognitionResponse{}
+	mi := &file_aisvc_proto_msgTypes[1]
+	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+	ms.StoreMessageInfo(mi)
+}
+
+func (x *FaceRecognitionResponse) String() string {
+	return protoimpl.X.MessageStringOf(x)
+}
+
+func (*FaceRecognitionResponse) ProtoMessage() {}
+
+func (x *FaceRecognitionResponse) ProtoReflect() protoreflect.Message {
+	mi := &file_aisvc_proto_msgTypes[1]
+	if x != nil {
+		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+		if ms.LoadMessageInfo() == nil {
+			ms.StoreMessageInfo(mi)
+		}
+		return ms
+	}
+	return mi.MessageOf(x)
+}
+
+// Deprecated: Use FaceRecognitionResponse.ProtoReflect.Descriptor instead.
+func (*FaceRecognitionResponse) Descriptor() ([]byte, []int) {
+	return file_aisvc_proto_rawDescGZIP(), []int{1}
+}
+
+func (x *FaceRecognitionResponse) GetFaceId() int64 {
+	if x != nil {
+		return x.FaceId
+	}
+	return 0
+}
+
+var File_aisvc_proto protoreflect.FileDescriptor
+
+var file_aisvc_proto_rawDesc = []byte{
+	0x0a, 0x0b, 0x61, 0x69, 0x73, 0x76, 0x63, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x02, 0x61,
+	0x69, 0x22, 0x45, 0x0a, 0x16, 0x46, 0x61, 0x63, 0x65, 0x52, 0x65, 0x63, 0x6f, 0x67, 0x6e, 0x69,
+	0x74, 0x69, 0x6f, 0x6e, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x12, 0x12, 0x0a, 0x04, 0x66,
+	0x61, 0x63, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x04, 0x66, 0x61, 0x63, 0x65, 0x12,
+	0x17, 0x0a, 0x07, 0x75, 0x73, 0x65, 0x72, 0x5f, 0x69, 0x64, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09,
+	0x52, 0x06, 0x75, 0x73, 0x65, 0x72, 0x49, 0x64, 0x22, 0x32, 0x0a, 0x17, 0x46, 0x61, 0x63, 0x65,
+	0x52, 0x65, 0x63, 0x6f, 0x67, 0x6e, 0x69, 0x74, 0x69, 0x6f, 0x6e, 0x52, 0x65, 0x73, 0x70, 0x6f,
+	0x6e, 0x73, 0x65, 0x12, 0x17, 0x0a, 0x07, 0x66, 0x61, 0x63, 0x65, 0x5f, 0x69, 0x64, 0x18, 0x01,
+	0x20, 0x01, 0x28, 0x03, 0x52, 0x06, 0x66, 0x61, 0x63, 0x65, 0x49, 0x64, 0x32, 0x57, 0x0a, 0x09,
+	0x41, 0x69, 0x53, 0x65, 0x72, 0x76, 0x69, 0x63, 0x65, 0x12, 0x4a, 0x0a, 0x0f, 0x46, 0x61, 0x63,
+	0x65, 0x52, 0x65, 0x63, 0x6f, 0x67, 0x6e, 0x69, 0x74, 0x69, 0x6f, 0x6e, 0x12, 0x1a, 0x2e, 0x61,
+	0x69, 0x2e, 0x46, 0x61, 0x63, 0x65, 0x52, 0x65, 0x63, 0x6f, 0x67, 0x6e, 0x69, 0x74, 0x69, 0x6f,
+	0x6e, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x1b, 0x2e, 0x61, 0x69, 0x2e, 0x46, 0x61,
+	0x63, 0x65, 0x52, 0x65, 0x63, 0x6f, 0x67, 0x6e, 0x69, 0x74, 0x69, 0x6f, 0x6e, 0x52, 0x65, 0x73,
+	0x70, 0x6f, 0x6e, 0x73, 0x65, 0x42, 0x06, 0x5a, 0x04, 0x2e, 0x2f, 0x70, 0x62, 0x62, 0x06, 0x70,
+	0x72, 0x6f, 0x74, 0x6f, 0x33,
+}
+
+var (
+	file_aisvc_proto_rawDescOnce sync.Once
+	file_aisvc_proto_rawDescData = file_aisvc_proto_rawDesc
+)
+
+func file_aisvc_proto_rawDescGZIP() []byte {
+	file_aisvc_proto_rawDescOnce.Do(func() {
+		file_aisvc_proto_rawDescData = protoimpl.X.CompressGZIP(file_aisvc_proto_rawDescData)
+	})
+	return file_aisvc_proto_rawDescData
+}
+
+var file_aisvc_proto_msgTypes = make([]protoimpl.MessageInfo, 2)
+var file_aisvc_proto_goTypes = []any{
+	(*FaceRecognitionRequest)(nil),  // 0: ai.FaceRecognitionRequest
+	(*FaceRecognitionResponse)(nil), // 1: ai.FaceRecognitionResponse
+}
+var file_aisvc_proto_depIdxs = []int32{
+	0, // 0: ai.AiService.FaceRecognition:input_type -> ai.FaceRecognitionRequest
+	1, // 1: ai.AiService.FaceRecognition:output_type -> ai.FaceRecognitionResponse
+	1, // [1:2] is the sub-list for method output_type
+	0, // [0:1] is the sub-list for method input_type
+	0, // [0:0] is the sub-list for extension type_name
+	0, // [0:0] is the sub-list for extension extendee
+	0, // [0:0] is the sub-list for field type_name
+}
+
+func init() { file_aisvc_proto_init() }
+func file_aisvc_proto_init() {
+	if File_aisvc_proto != nil {
+		return
+	}
+	type x struct{}
+	out := protoimpl.TypeBuilder{
+		File: protoimpl.DescBuilder{
+			GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
+			RawDescriptor: file_aisvc_proto_rawDesc,
+			NumEnums:      0,
+			NumMessages:   2,
+			NumExtensions: 0,
+			NumServices:   1,
+		},
+		GoTypes:           file_aisvc_proto_goTypes,
+		DependencyIndexes: file_aisvc_proto_depIdxs,
+		MessageInfos:      file_aisvc_proto_msgTypes,
+	}.Build()
+	File_aisvc_proto = out.File
+	file_aisvc_proto_rawDesc = nil
+	file_aisvc_proto_goTypes = nil
+	file_aisvc_proto_depIdxs = nil
+}
--- a/app/aisvc/rpc/pb/aisvc_grpc.pb.go
+++ b/app/aisvc/rpc/pb/aisvc_grpc.pb.go
@@ -0,0 +1,123 @@
+// Code generated by protoc-gen-go-grpc. DO NOT EDIT.
+// versions:
+// - protoc-gen-go-grpc v1.5.1
+// - protoc             v3.19.4
+// source: aisvc.proto
+
+package pb
+
+import (
+	context "context"
+	grpc "google.golang.org/grpc"
+	codes "google.golang.org/grpc/codes"
+	status "google.golang.org/grpc/status"
+)
+
+// This is a compile-time assertion to ensure that this generated file
+// is compatible with the grpc package it is being compiled against.
+// Requires gRPC-Go v1.64.0 or later.
+const _ = grpc.SupportPackageIsVersion9
+
+const (
+	AiService_FaceRecognition_FullMethodName = "/ai.AiService/FaceRecognition"
+)
+
+// AiServiceClient is the client API for AiService service.
+//
+// For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream.
+type AiServiceClient interface {
+	// FaceRecognition
+	FaceRecognition(ctx context.Context, in *FaceRecognitionRequest, opts ...grpc.CallOption) (*FaceRecognitionResponse, error)
+}
+
+type aiServiceClient struct {
+	cc grpc.ClientConnInterface
+}
+
+func NewAiServiceClient(cc grpc.ClientConnInterface) AiServiceClient {
+	return &aiServiceClient{cc}
+}
+
+func (c *aiServiceClient) FaceRecognition(ctx context.Context, in *FaceRecognitionRequest, opts ...grpc.CallOption) (*FaceRecognitionResponse, error) {
+	cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
+	out := new(FaceRecognitionResponse)
+	err := c.cc.Invoke(ctx, AiService_FaceRecognition_FullMethodName, in, out, cOpts...)
+	if err != nil {
+		return nil, err
+	}
+	return out, nil
+}
+
+// AiServiceServer is the server API for AiService service.
+// All implementations must embed UnimplementedAiServiceServer
+// for forward compatibility.
+type AiServiceServer interface {
+	// FaceRecognition
+	FaceRecognition(context.Context, *FaceRecognitionRequest) (*FaceRecognitionResponse, error)
+	mustEmbedUnimplementedAiServiceServer()
+}
+
+// UnimplementedAiServiceServer must be embedded to have
+// forward compatible implementations.
+//
+// NOTE: this should be embedded by value instead of pointer to avoid a nil
+// pointer dereference when methods are called.
+type UnimplementedAiServiceServer struct{}
+
+func (UnimplementedAiServiceServer) FaceRecognition(context.Context, *FaceRecognitionRequest) (*FaceRecognitionResponse, error) {
+	return nil, status.Errorf(codes.Unimplemented, "method FaceRecognition not implemented")
+}
+func (UnimplementedAiServiceServer) mustEmbedUnimplementedAiServiceServer() {}
+func (UnimplementedAiServiceServer) testEmbeddedByValue()                   {}
+
+// UnsafeAiServiceServer may be embedded to opt out of forward compatibility for this service.
+// Use of this interface is not recommended, as added methods to AiServiceServer will
+// result in compilation errors.
+type UnsafeAiServiceServer interface {
+	mustEmbedUnimplementedAiServiceServer()
+}
+
+func RegisterAiServiceServer(s grpc.ServiceRegistrar, srv AiServiceServer) {
+	// If the following call pancis, it indicates UnimplementedAiServiceServer was
+	// embedded by pointer and is nil.  This will cause panics if an
+	// unimplemented method is ever invoked, so we test this at initialization
+	// time to prevent it from happening at runtime later due to I/O.
+	if t, ok := srv.(interface{ testEmbeddedByValue() }); ok {
+		t.testEmbeddedByValue()
+	}
+	s.RegisterService(&AiService_ServiceDesc, srv)
+}
+
+func _AiService_FaceRecognition_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
+	in := new(FaceRecognitionRequest)
+	if err := dec(in); err != nil {
+		return nil, err
+	}
+	if interceptor == nil {
+		return srv.(AiServiceServer).FaceRecognition(ctx, in)
+	}
+	info := &grpc.UnaryServerInfo{
+		Server:     srv,
+		FullMethod: AiService_FaceRecognition_FullMethodName,
+	}
+	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
+		return srv.(AiServiceServer).FaceRecognition(ctx, req.(*FaceRecognitionRequest))
+	}
+	return interceptor(ctx, in, info, handler)
+}
+
+// AiService_ServiceDesc is the grpc.ServiceDesc for AiService service.
+// It's only intended for direct use with grpc.RegisterService,
+// and not to be introspected or modified (even as a copy)
+var AiService_ServiceDesc = grpc.ServiceDesc{
+	ServiceName: "ai.AiService",
+	HandlerType: (*AiServiceServer)(nil),
+	Methods: []grpc.MethodDesc{
+		{
+			MethodName: "FaceRecognition",
+			Handler:    _AiService_FaceRecognition_Handler,
+		},
+	},
+	Streams:  []grpc.StreamDesc{},
+	Metadata: "aisvc.proto",
+}