Golang通用数据库单条记录到json

Question

我想从数据库中检索记录并将这些记录编组到 json。我有大约 30 个不同的 tables，所以我想要通用函数来处理所有这些 tables。我使用 xorm 进行数据库访问。

我已经设法创建了检索数据的 DRY 函数，这主要归功于此

这有效，可以将所有记录编组到 json:

type user struct {
   Id   int64  `json:"id"`
   Name string `json:"name"`
}
// type post
// etc.

type tableRecord struct {
    PrimaryKey string
    Data       interface{}
}

var ListOfTables = map[string]tableRecord{
    "users":{"id", &[]user{}},  // type user is struct for xorm with json annotation
    //"posts":{"post_id", &[]post{}},
    // etc.. 
}

for tableName, rec := range ListOfTables {
    err := xorm.Find(rec.Data)
    if err != nil {
        log.Print(err)
    }

    out, err := json.Marshal(rec.Data)
    if err != nil {
        log.Print(err)
    }
    log.Print(string(out)) // this yields json array
}

但是我很难将单个记录编组到 json。我四处寻找迭代 interface{} 的方法，它包含一个切片、found this 和类似的主题。尝试过：

switch reflect.TypeOf(reflect.ValueOf(rec.Data).Elem().Interface()).Kind() {
case reflect.Slice:
    s := reflect.ValueOf(reflect.ValueOf(rec.Data).Elem().Interface())
    for i := 0; i < s.Len(); i++ {
        entry := s.Index(i)
        log.Printf("%v\n", entry) // prints {1 John Doe}
        // log.Print(reflect.ValueOf(entry))
        data, err := json.MarshalIndent(entry, " ", "  ")
        if err != nil {
            log.Print(err)
        }
        log.Println(string(data)) // prints {} empty
    }
}

当然，如果我指定 rec.Data 是 *[]user 它可以工作，但是我必须为每个 table 重写这样的代码，这不是我想要的.

switch t := rec.Data.(type) {
case *[]user:
    for _, entry := range *t {
        // log.Printf("loop %v", entry)
        data, err := json.MarshalIndent(entry, " ", "  ")
        if err != nil {
            log.Print(err)
        }
        log.Println(string(data)) // yields needed json for single record
    }
}

或者也许有一种完全不同的更好的方法来解决这样的问题 - 数据库的任何记录到 json。

更新现在的问题是，Xorm 需要结构吗？我将不得不阅读 xorm 的可能性和限制。

slice := record.Slice()
log.Print(reflect.TypeOf(slice))

err = env.hxo.In(record.PrimaryKey(), insertIds).Find(slice) // or &slice
if err != nil {
    log.Print(err) // Table not found
}

// this works
var slice2 []*user
err = env.hxo.In(record.PrimaryKey(), insertIds).Find(&slice2)
if err != nil {
    log.Print(err) // 
}

Answer 1

因此，正如我在评论中提到的，如果您希望能够从 tableRecord.Data 字段中获取单个元素，最简单的方法就是将字段类型更改为实际类型：

type tableRecord struct {
    PrimaryKey string
    Data       []interface{} // slice of whatever
}

这样，你就可以写出非常通用的东西了：

for tbl, record := range records {
    fmt.Printf("First record from table %s\n", tbl)
    b, _ := json.MarshalIndent(record[0], " ", "  ")
    fmt.Println(string(b))
    fmt.Prinln("other records...")
    b, _ = json.MarshalIndend(record[1:], " ", "  ")
    fmt.Println(string(b))
}

不过，如果我是你，我会考虑在我的数据库类型中实现一个接口。大致如下：

type DBType interface {
    PrimaryKey() string
    TableName() string // xorm can use this to get the table name
    Slice() []DBType // can return []user or whatever
}

所以你真的不需要 tableRecord 类型了，可以像这样使用 var:

listOfTables := []DBType{user{}, ...}
for _, tbl := range listOfTables {
    data := tbl.Slice()
    // find data here
    fmt.Printf("First record from table %s\n", tbl.TableName())
    b, _ := json.MarshalIndent(data[0], " ", "  ")
    fmt.Println(string(b))
    fmt.Prinln("other records...")
    b, _ = json.MarshalIndend(data[1:], " ", "  ")
    fmt.Println(string(b))
}

所以我的 answer/comments 中缺少的 TL;DR:

从类型 []user{}（或 []DBTable）到 []interface{} 的转换不起作用，因为您不能将所有元素都转换为一个切片在一个表达式中。您必须创建第二个 []interface{} 类型的切片并像这样复制值：

切片 := userVar.Slice() 数据：=制作（[]接口{}，len（切片））对于我：=范围切片{ data[i] = slice[i] // 将类型复制到 interface{} 切片 } return tableRecord{userVar.PrimaryKey(), data}

我创建了一个小的工作示例来说明如何使用上述接口。

DEMO

为避免过多混乱，您可以立即将 Slice 函数更改为 return a []interface{}：

func(v T) Slice() []interface{
    return []interface{
        &T{},
    }
}

您实施 Slice 的错误在于您有这样的事情：

func (u *user) Slice() []DBTable {
    u = &user{} // you're re-assigning the receiver, losing all state!
    return []DBTable{u}
}

接收器是指针类型，因此您所做的任何重新分配都会影响调用 func 的变量。那不是一个好主意。只使用值接收器，或者，如果你想确保接口只在指针变量上实现（一个常见的技巧，例如 gRPC 使用的）是像这样实现函数：

func(*user) Slice() []DBTable{
    return []DBTable{&user{}}
}

这个技巧的一个很好的例子可以在使用协议缓冲区时生成的 pb.go 文件中找到。消息类型将具有如下功能：

func(*MsgType) ProtoMessage() {}

Golang通用数据库单条记录到json

Golang generic database single record to json

go

go-xorm

DEMO