gRPC调用坎坷历程记录

syntax = "proto3";
package rpc.auth;
option go_package = "github.com/BigNoseCattyHome/aorb/backend/rpc/auth;auth";

import "google/protobuf/timestamp.proto";

// 定义消息，用于请求和响应结构
message LoginRequest {
  string username = 1; // 用户名/用户ID
  string password = 2; // 密码的md5摘要
  string device_id = 3; // 设备ID
  google.protobuf.Timestamp timestamp = 4; // 时间戳
  string nonce = 5; // 随机数
}

protoc --go_out=. person.proto	// 找到当前目录下的person.proto并生成go的代码，输出到当前目录(.)

PROTO_PATH=./idl
GOOGLE_PROTO_PATH=/usr/local/include
OUTPUT_DART_PATH=./frontend/lib/generated/
GO_OUT_PATH=./backend/rpc
PROTOC_GEN_DART=$(shell which protoc-gen-dart)

proto:
	@echo "Creating golang and dart grpc files..."
	@for file in $(PROTO_PATH)/*.proto; do \
		if [ -f "$$file" ]; then \
			prefix=$$(basename "$$file" .proto); \
			mkdir -p $(GO_OUT_PATH)/"$${prefix}"; \
			mkdir -p $(OUTPUT_DART_PATH); \
			echo "Created directory for $$prefix"; \
			protoc -I$(PROTO_PATH) -I$(GOOGLE_PROTO_PATH) \
				--go_out=$(GO_OUT_PATH)/$$prefix --go_opt=paths=source_relative \
				--go-grpc_out=$(GO_OUT_PATH)/$$prefix --go-grpc_opt=paths=source_relative \
				--dart_out=grpc:$(OUTPUT_DART_PATH) \
				--plugin=protoc-gen-dart=$(PROTOC_GEN_DART) \
				$$file; \
			echo "Generated gRPC code for $$prefix"; \
		fi; \
	done
	@protoc -I$(GOOGLE_PROTO_PATH) \
		--dart_out=grpc:$(OUTPUT_DART_PATH) \
		--plugin=protoc-gen-dart=$(PROTOC_GEN_DART) \
		google/protobuf/timestamp.proto
	@echo "Generated Dart code for Google's timestamp.proto"

// gRPC 服务名到 Consul 服务名的映射
var serviceNameMapping = map[string]string{
	"rpc.auth.AuthService":           config.AuthRpcServerName,
	"rpc.user.UserService":           config.UserRpcServerName,
	"rpc.comment.CommentService":     config.CommentRpcServerName,
	"rpc.vote.VoteService":           config.VoteRpcServerName,
	"rpc.poll.QuestionService":       config.PollRpcServerName,
	"rpc.recommend.RecommendService": config.RecommendRpcServerName,
}

syntax = "proto3";
package user;
option go_package = "github.com/BigNoseCattyHome/aorb/backend/rpc/user;user";


message CoinRecord {
  int64 consume = 1; // 消耗的金币数
  string question_id = 2; // 为其投币的问题ID
  string user_id = 3; // 使用者的ID
}

// TODO optional 字段在后续开发过程中应该逐步取消
message User {
  string avatar = 1; // 用户头像
  repeated string blacklist = 2; // 屏蔽好友
  optional double coins = 3; // 用户的金币数
  repeated CoinRecord coins_record = 4; // 用户金币流水记录
  repeated string followed = 5; // 关注者
  repeated string follower = 6; // 被关注者
  uint32 id = 7; // 用户ID
  optional string ipaddress = 8; // IP归属地
  string nickname = 9; // 用户昵称
  repeated uint32 questions_ask = 10; // 发起过的问题
  repeated uint32 questions_asw = 11; // 回答过的问题
  repeated uint32 questions_collect = 12; // 收藏的问题
  string username = 13; // 用户登录名
}

message UserRequest{
  uint32 user_id = 1;   // 用户id
  uint32 actor_id = 2;  // 发送请求的用户的id
}

message UserResponse{
  int32 status_code = 1;  // 状态码，0-成功，其他值-失败
  string status_msg = 2;  // 返回状态描述
  User user = 3;  // 用户信息
}

message UserExistRequest{
  uint32 user_id = 1; // 用户id
}

message UserExistResponse{
  int32 status_code = 1;  // 状态码，0-成功，其他值-失败
  string status_msg = 2;  // 返回状态描述
  bool existed = 3;  // 是否存在用户
}

service UserService{
  rpc GetUserInfo(UserRequest) returns (UserResponse);
  rpc GetUserExistInformation(UserExistRequest) returns (UserExistResponse);
}

// UserServiceClient is the client API for UserService 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 UserServiceClient interface {
	GetUserInfo(ctx context.Context, in *UserRequest, opts ...grpc.CallOption) (*UserResponse, error)
	GetUserExistInformation(ctx context.Context, in *UserExistRequest, opts ...grpc.CallOption) (*UserExistResponse, error)
}

type userServiceClient struct {
	cc grpc.ClientConnInterface
}

func NewUserServiceClient(cc grpc.ClientConnInterface) UserServiceClient {
	return &userServiceClient{cc}
}

func (c *userServiceClient) GetUserInfo(ctx context.Context, in *UserRequest, opts ...grpc.CallOption) (*UserResponse, error) {
	cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
	out := new(UserResponse)
	err := c.cc.Invoke(ctx, UserService_GetUserInfo_FullMethodName, in, out, cOpts...)
	if err != nil {
		return nil, err
	}
	return out, nil
}

func (c *userServiceClient) GetUserExistInformation(ctx context.Context, in *UserExistRequest, opts ...grpc.CallOption) (*UserExistResponse, error) {
	cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...)
	out := new(UserExistResponse)
	err := c.cc.Invoke(ctx, UserService_GetUserExistInformation_FullMethodName, in, out, cOpts...)
	if err != nil {
		return nil, err
	}
	return out, nil
}

// UserServiceServer is the server API for UserService service.
// All implementations must embed UnimplementedUserServiceServer
// for forward compatibility
type UserServiceServer interface {
	GetUserInfo(context.Context, *UserRequest) (*UserResponse, error)
	GetUserExistInformation(context.Context, *UserExistRequest) (*UserExistResponse, error)
	mustEmbedUnimplementedUserServiceServer()
}

// Register 注册
func (a AuthServiceImpl) Register(context context.Context, request *auth.RegisterRequest) (*auth.RegisterResponse, error) {
	log.Infof("Received Register request: %v", request)

	// 解析参数
	user := models.User{
		Username:  request.Username,
		Password:  request.Password,
		Nickname:  request.Nickname,
		Avatar:    request.Avatar,
		Ipaddress: request.Ipaddress,
	}

	// 调用服务
	err := services.RegisterUser(user)
	if err != nil {
		return nil, status.Errorf(codes.Unauthenticated, "register failed: %v", err)
	}

	// 返回响应
	registerResponse := &auth.RegisterResponse{
		Success: true,
		Message: "User registered successfully",
	}
	return registerResponse, nil
}