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

时间:2017-12-27 14:43:29

标签: postgresql go grpc go-gorm proto3

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

是否有人能够在原型定义中成功使用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;
}

3 个答案:

答案 0 :(得分:1)

尝试使用protoc-gen-gorm。它将创建另一个文件.pb.gorm.go

答案 1 :(得分:1)

protoc-gen-gorm项目对我不起作用。似乎proto2和proto3混合在一起了,最终我无法使其工作。

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

如果这是我的原型profile/profile.proto

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

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

// ...
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生成文件后,我在go文件上调用哪个,而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" 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"`
}
// ...

答案 2 :(得分:0)

可能是使用这样的选项:https://github.com/favadi/protoc-go-inject-tag 自动生成标签(我自己仍在研究这个)