删除无用代码，加入RPC调用

pb/shotrdp.pb.go

@@ -0,0 +1,177 @@
+// Code generated by protoc-gen-go. DO NOT EDIT.
+// versions:
+// 	protoc-gen-go v1.36.5
+// 	protoc        v6.30.0
+// source: shotrdp.proto
+
+package pb
+
+import (
+	protoreflect "google.golang.org/protobuf/reflect/protoreflect"
+	protoimpl "google.golang.org/protobuf/runtime/protoimpl"
+	reflect "reflect"
+	sync "sync"
+	unsafe "unsafe"
+)
+
+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 Request struct {
+	state         protoimpl.MessageState `protogen:"open.v1"`
+	Host          string                 `protobuf:"bytes,1,opt,name=host,proto3" json:"host,omitempty"`
+	unknownFields protoimpl.UnknownFields
+	sizeCache     protoimpl.SizeCache
+}
+
+func (x *Request) Reset() {
+	*x = Request{}
+	mi := &file_shotrdp_proto_msgTypes[0]
+	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+	ms.StoreMessageInfo(mi)
+}
+
+func (x *Request) String() string {
+	return protoimpl.X.MessageStringOf(x)
+}
+
+func (*Request) ProtoMessage() {}
+
+func (x *Request) ProtoReflect() protoreflect.Message {
+	mi := &file_shotrdp_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 Request.ProtoReflect.Descriptor instead.
+func (*Request) Descriptor() ([]byte, []int) {
+	return file_shotrdp_proto_rawDescGZIP(), []int{0}
+}
+
+func (x *Request) GetHost() string {
+	if x != nil {
+		return x.Host
+	}
+	return ""
+}
+
+type Response struct {
+	state         protoimpl.MessageState `protogen:"open.v1"`
+	ImageBytes    []byte                 `protobuf:"bytes,1,opt,name=image_bytes,json=imageBytes,proto3" json:"image_bytes,omitempty"`
+	unknownFields protoimpl.UnknownFields
+	sizeCache     protoimpl.SizeCache
+}
+
+func (x *Response) Reset() {
+	*x = Response{}
+	mi := &file_shotrdp_proto_msgTypes[1]
+	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
+	ms.StoreMessageInfo(mi)
+}
+
+func (x *Response) String() string {
+	return protoimpl.X.MessageStringOf(x)
+}
+
+func (*Response) ProtoMessage() {}
+
+func (x *Response) ProtoReflect() protoreflect.Message {
+	mi := &file_shotrdp_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 Response.ProtoReflect.Descriptor instead.
+func (*Response) Descriptor() ([]byte, []int) {
+	return file_shotrdp_proto_rawDescGZIP(), []int{1}
+}
+
+func (x *Response) GetImageBytes() []byte {
+	if x != nil {
+		return x.ImageBytes
+	}
+	return nil
+}
+
+var File_shotrdp_proto protoreflect.FileDescriptor
+
+var file_shotrdp_proto_rawDesc = string([]byte{
+	0x0a, 0x0d, 0x73, 0x68, 0x6f, 0x74, 0x72, 0x64, 0x70, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12,
+	0x02, 0x70, 0x62, 0x22, 0x1d, 0x0a, 0x07, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x12, 0x12,
+	0x0a, 0x04, 0x68, 0x6f, 0x73, 0x74, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x04, 0x68, 0x6f,
+	0x73, 0x74, 0x22, 0x2b, 0x0a, 0x08, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x12, 0x1f,
+	0x0a, 0x0b, 0x69, 0x6d, 0x61, 0x67, 0x65, 0x5f, 0x62, 0x79, 0x74, 0x65, 0x73, 0x18, 0x01, 0x20,
+	0x01, 0x28, 0x0c, 0x52, 0x0a, 0x69, 0x6d, 0x61, 0x67, 0x65, 0x42, 0x79, 0x74, 0x65, 0x73, 0x32,
+	0x33, 0x0a, 0x09, 0x47, 0x65, 0x74, 0x53, 0x63, 0x72, 0x65, 0x65, 0x6e, 0x12, 0x26, 0x0a, 0x09,
+	0x47, 0x65, 0x74, 0x53, 0x63, 0x72, 0x65, 0x65, 0x6e, 0x12, 0x0b, 0x2e, 0x70, 0x62, 0x2e, 0x52,
+	0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x0c, 0x2e, 0x70, 0x62, 0x2e, 0x52, 0x65, 0x73, 0x70,
+	0x6f, 0x6e, 0x73, 0x65, 0x42, 0x0c, 0x5a, 0x0a, 0x73, 0x68, 0x6f, 0x74, 0x72, 0x64, 0x70, 0x2f,
+	0x70, 0x62, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33,
+})
+
+var (
+	file_shotrdp_proto_rawDescOnce sync.Once
+	file_shotrdp_proto_rawDescData []byte
+)
+
+func file_shotrdp_proto_rawDescGZIP() []byte {
+	file_shotrdp_proto_rawDescOnce.Do(func() {
+		file_shotrdp_proto_rawDescData = protoimpl.X.CompressGZIP(unsafe.Slice(unsafe.StringData(file_shotrdp_proto_rawDesc), len(file_shotrdp_proto_rawDesc)))
+	})
+	return file_shotrdp_proto_rawDescData
+}
+
+var file_shotrdp_proto_msgTypes = make([]protoimpl.MessageInfo, 2)
+var file_shotrdp_proto_goTypes = []any{
+	(*Request)(nil),  // 0: pb.Request
+	(*Response)(nil), // 1: pb.Response
+}
+var file_shotrdp_proto_depIdxs = []int32{
+	0, // 0: pb.GetScreen.GetScreen:input_type -> pb.Request
+	1, // 1: pb.GetScreen.GetScreen:output_type -> pb.Response
+	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_shotrdp_proto_init() }
+func file_shotrdp_proto_init() {
+	if File_shotrdp_proto != nil {
+		return
+	}
+	type x struct{}
+	out := protoimpl.TypeBuilder{
+		File: protoimpl.DescBuilder{
+			GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
+			RawDescriptor: unsafe.Slice(unsafe.StringData(file_shotrdp_proto_rawDesc), len(file_shotrdp_proto_rawDesc)),
+			NumEnums:      0,
+			NumMessages:   2,
+			NumExtensions: 0,
+			NumServices:   1,
+		},
+		GoTypes:           file_shotrdp_proto_goTypes,
+		DependencyIndexes: file_shotrdp_proto_depIdxs,
+		MessageInfos:      file_shotrdp_proto_msgTypes,
+	}.Build()
+	File_shotrdp_proto = out.File
+	file_shotrdp_proto_goTypes = nil
+	file_shotrdp_proto_depIdxs = nil
+}

pb/shotrdp.proto

@@ -0,0 +1,16 @@
+syntax = "proto3";
+option go_package = "shotrdp/pb";
+
+package pb;
+
+message Request {
+  string host = 1;
+}
+
+message Response {
+  bytes image_bytes = 1;
+}
+
+service GetScreen {
+  rpc GetScreen (Request) returns (Response);
+}

pb/shotrdp_grpc.pb.go

@@ -0,0 +1,121 @@
+// Code generated by protoc-gen-go-grpc. DO NOT EDIT.
+// versions:
+// - protoc-gen-go-grpc v1.5.1
+// - protoc             v6.30.0
+// source: shotrdp.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 (
+	GetScreen_GetScreen_FullMethodName = "/pb.GetScreen/GetScreen"
+)
+
+// GetScreenClient is the client API for GetScreen 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 GetScreenClient interface {
+	GetScreen(ctx context.Context, in *Request, opts ...grpc.CallOption) (*Response, error)
+}
+
+type getScreenClient struct {
+	cc grpc.ClientConnInterface
+}
+
+func NewGetScreenClient(cc grpc.ClientConnInterface) GetScreenClient {
+	return &getScreenClient{cc}
+}
+
+func (c *getScreenClient) GetScreen(ctx context.Context, in *Request, opts ...grpc.CallOption) (*Response, error) {
+	cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
+	out := new(Response)
+	err := c.cc.Invoke(ctx, GetScreen_GetScreen_FullMethodName, in, out, cOpts...)
+	if err != nil {
+		return nil, err
+	}
+	return out, nil
+}
+
+// GetScreenServer is the server API for GetScreen service.
+// All implementations must embed UnimplementedGetScreenServer
+// for forward compatibility.
+type GetScreenServer interface {
+	GetScreen(context.Context, *Request) (*Response, error)
+	mustEmbedUnimplementedGetScreenServer()
+}
+
+// UnimplementedGetScreenServer 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 UnimplementedGetScreenServer struct{}
+
+func (UnimplementedGetScreenServer) GetScreen(context.Context, *Request) (*Response, error) {
+	return nil, status.Errorf(codes.Unimplemented, "method GetScreen not implemented")
+}
+func (UnimplementedGetScreenServer) mustEmbedUnimplementedGetScreenServer() {}
+func (UnimplementedGetScreenServer) testEmbeddedByValue()                   {}
+
+// UnsafeGetScreenServer may be embedded to opt out of forward compatibility for this service.
+// Use of this interface is not recommended, as added methods to GetScreenServer will
+// result in compilation errors.
+type UnsafeGetScreenServer interface {
+	mustEmbedUnimplementedGetScreenServer()
+}
+
+func RegisterGetScreenServer(s grpc.ServiceRegistrar, srv GetScreenServer) {
+	// If the following call pancis, it indicates UnimplementedGetScreenServer 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(&GetScreen_ServiceDesc, srv)
+}
+
+func _GetScreen_GetScreen_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
+	in := new(Request)
+	if err := dec(in); err != nil {
+		return nil, err
+	}
+	if interceptor == nil {
+		return srv.(GetScreenServer).GetScreen(ctx, in)
+	}
+	info := &grpc.UnaryServerInfo{
+		Server:     srv,
+		FullMethod: GetScreen_GetScreen_FullMethodName,
+	}
+	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
+		return srv.(GetScreenServer).GetScreen(ctx, req.(*Request))
+	}
+	return interceptor(ctx, in, info, handler)
+}
+
+// GetScreen_ServiceDesc is the grpc.ServiceDesc for GetScreen service.
+// It's only intended for direct use with grpc.RegisterService,
+// and not to be introspected or modified (even as a copy)
+var GetScreen_ServiceDesc = grpc.ServiceDesc{
+	ServiceName: "pb.GetScreen",
+	HandlerType: (*GetScreenServer)(nil),
+	Methods: []grpc.MethodDesc{
+		{
+			MethodName: "GetScreen",
+			Handler:    _GetScreen_GetScreen_Handler,
+		},
+	},
+	Streams:  []grpc.StreamDesc{},
+	Metadata: "shotrdp.proto",
+}