Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Golang get string representation of specific struct field name

I really want a way to print the string representation of a field name in go. It has several use cases, but here is an example:

lets say I have a struct

type Test struct {
    Field      string `bson:"Field" json:"field"`
    OtherField int    `bson:"OtherField" json:"otherField"`
}

and, for example, I want to do a mongo find:

collection.Find(bson.M{"OtherField": someValue})

I don't like that I have to put the string "OtherField" in there. It seems brittle and easy to either misstype or have the struct change and then my query fails without me knowing it.

Is there any way to get the string "OtherField" without having to either declare a const or something like that? I know I can use reflection to a get a list of field names from a struct, but I'd really like to do something along the lines of

fieldName := nameOf(Test{}.OtherField) 
collection.Find(bson.M{fieldName: someValue})

is there any way to do this in Go?? C# 6 has the built in nameof, but digging through reflection I can't find any way to do this in Go.

like image 818
Fuzzerker Avatar asked May 04 '17 19:05

Fuzzerker


2 Answers

I don't really think there is. You may be able to load a set of types via reflection and generate a set of constants for the field names. So:

type Test struct {
    Field      string `bson:"Field" json:"field"`
    OtherField int    `bson:"OtherField" json:"otherField"`
}

Could generate something like:

var TestFields = struct{
      Field string
      OtherField string
}{"Field","OtherField"}

and you could use TestFields.Field as a constant.

Unfortunately, I don't know of any existing tool that does anything like that. Would be fairly simple to do, and wire up to go generate though.

EDIT:

How I'd generate it:

  1. Make a package that accepts an array of reflect.Type or interface{} and spits out a code file.
  2. Make a generate.go somewhere in my repo with main function:

    func main(){
       var text = mygenerator.Gen(Test{}, OtherStruct{}, ...)
       // write text to constants.go or something
    }
    
  3. Add //go:generate go run scripts/generate.go to my main app and run go generate
like image 181
captncraig Avatar answered Oct 11 '22 10:10

captncraig


Here is a function that will return a []string with the struct field names. I think it comes in the order they are defined.

WARNING: Reordering the fields in the struct definition will change the order in which they appear

https://play.golang.org/p/dNATzNn47S

package main

import (
    "fmt"
    "strings"
    "regexp"
    )

type Test struct {
    Field      string `bson:"Field" json:"field"`
    OtherField int    `bson:"OtherField" json:"otherField"`
}

func main() {
    fields, err := GetFieldNames(Test{})
    if err != nil {
        fmt.Println(err)
        return
    }
    fmt.Println(fields)
}


func GetFieldNames(i interface{}) ([]string, error) {
    // regular expression to find the unquoted json
    reg := regexp.MustCompile(`(\s*?{\s*?|\s*?,\s*?)(['"])?(?P<Field>[a-zA-Z0-9]+)(['"])?:`)

    // print struct in almost json form (fields unquoted)
    raw := fmt.Sprintf("%#v", i)

    // remove the struct name so string begins with "{"
    fjs := raw[strings.Index(raw,"{"):]

    // find and grab submatch 3
    matches := reg.FindAllStringSubmatch(fjs,-1)

    // collect
    fields := []string{}
    for _, v := range matches {
        if len(v) >= 3 && v[3] != "" {
            fields = append(fields, v[3])
        }

    }


return fields, nil

}

like image 34
RayfenWindspear Avatar answered Oct 11 '22 10:10

RayfenWindspear