postgresql - 将 gorm.Model 字段集成到 protobuf 定义中

Question

我正在尝试弄清楚如何将 gorm.Model 字段(deleted_at、create_at、id 等)集成到我的 proto3 定义中。但是，我不能为 proto3 设置日期时间类型。我试图寻找有关如何将 gorm 字段序列化为字符串的文档(因为 proto3 处理字符串)但我没有找到任何东西。

有没有人能够在他们的原型(prototype)定义中成功使用 gorm 模型字段？我正在使用 go-micro 的插件生成 *pb.go 文件。

这是我当前的消息定义，它不起作用。似乎空字符串被存储在 deleted_at 的数据库中，因为当查询 deleted_at 为 null 时，postgres 数据库不返回任何内容。

message DatabaseConfig {
    string address = 1;
    int32 port = 2;
    string databaseName = 3;
    string username = 4;
    string password = 5;
    string databaseType = 6;
    string quertStatement = 7;
    int32 id = 8;
    string createdAt = 9;
    string updatedAt = 10;
    string deletedAt = 11;
}

更新:我已经将我的 proto def 更新为以下内容，但 gorm 仍然没有正确使用 Id、CreatedAt、UpdatedAt 和 DeletedAt 字段

syntax = "proto3";

package go.micro.srv.importer;

import "google/protobuf/timestamp.proto";
import "github.com/gogo/protobuf/gogoproto/gogo.proto";

service ImporterService {
    rpc CreateDatabaseConfig(DatabaseConfig) returns (Response) {}
    rpc RetrieveDatabaseConfig(GetRequest) returns (Response) {}
    rpc UpdateDatabaseConfig(DatabaseConfig) returns (Response) {}
    rpc DeleteDatabaseConfig(DatabaseConfig) returns (Response) {}
}


message GetRequest {}


message DatabaseConfig {
    string address = 1;
    int32 port = 2;
    string databaseName = 3;
    string username = 4;
    string password = 5;
    string databaseType = 6;
    string quertStatement = 7;
    int32 id = 8;
    google.protobuf.Timestamp createdAt = 9 [(gogoproto.stdtime) = true];
    google.protobuf.Timestamp updatedAt = 10 [(gogoproto.stdtime) = true];
    google.protobuf.Timestamp deletedAt = 11 [(gogoproto.stdtime) = true];
}


message Response {
    bool created = 1;
    DatabaseConfig database_config = 2;

    repeated DatabaseConfig databaseConfigs = 3;
}

Answer 1

protoc-gen-gorm 项目对我不起作用。看起来 proto2 和 proto3 发生了一些混合，最终我无法让它工作。

我的解决方案是在从 protobuf 生成 go 文件后创建一个脚本来进行后处理。

如果这是我的原型(prototype)profile/profile.proto:

message Profile {
  uint64 id = 1;
  string name = 2;
  bool active = 3;
  // ...
}

使用标准 protoc 命令创建了 profile/profile.pb.go:

// ...
type Profile struct {
    state         protoimpl.MessageState
    sizeCache     protoimpl.SizeCache
    unknownFields protoimpl.UnknownFields

    Id     uint64 `protobuf:"varint,1,opt,name=id,proto3" json:"id,omitempty"`
    Name   string `protobuf:"bytes,2,opt,name=name,proto3" json:"name,omitempty"`
    Active bool   `protobuf:"varint,3,opt,name=active,proto3" json:"active,omitempty"`
}
// ...

我使用这个脚本gorm.sh:

#!/bin/bash

g () {
  sed "s/json:\"$1,omitempty\"/json:\"$1,omitempty\" gorm:\"type:$2\"/"
}

cat $1 \
| g "id" "primary_key" \
| g "name" "varchar(100)" \
> $1.tmp && mv $1{.tmp,}

在使用 ./gorm.sh profile/profile.pb.go 和 profile/profile.pb.go 的结果生成后，我在我的 go 文件上调用它> 是:

// ...
type Profile struct {
    state         protoimpl.MessageState
    sizeCache     protoimpl.SizeCache
    unknownFields protoimpl.UnknownFields

    Id     uint64 `protobuf:"varint,1,opt,name=id,proto3" json:"id,omitempty" gorm:"type:primary_key"`
    Name   string `protobuf:"bytes,2,opt,name=name,proto3" json:"name,omitempty" gorm:"type:varchar(100)"`
    Active bool   `protobuf:"varint,3,opt,name=active,proto3" json:"active,omitempty"`
}
// ...