Отменить только утвержденные поля - PullRequest
Новая
       41

Отменить только утвержденные поля

1 голос
/ 03 мая 2020

Я пытаюсь дать пользователям разрешение на редактирование определенных полей структуры. Эти поля будут различаться в зависимости от действия, контекста и роли текущего пользователя.

В данный момент я делаю это настоятельно, но это довольно утомительно и не масштабируется.

Я думал, что список approved_fields может быть хорошим и более масштабируемым решением, но я не знаю, как с этим справиться go. Я думаю, что рефлексия - это путь к go, но я еще недостаточно знаю.

Любое вдохновение поможет. 

type Foo struct {
  Bar1 int
  Bar2 int
  Bar3 int
}

foo := Foo{}
approved_fields := []string{"Bar1", "Bar2"}
decode(json_data, &foo, approved_fields)

foo2 := Foo{}
approved_fields := []string{"Bar1", "Bar3"}
decode(json_data, &foo2, approved_fields)

1 Ответ

0 голосов
/ 03 мая 2020

Вот как бы я подумал о ее решении. Может быть что-то более эффективное, так как я разбираю весь элемент перед выбором полей. 

import (
    "encoding/json"
    "log"
    "reflect"

    "github.com/pkg/errors"
)

func fieldByName(s string, v interface{}) (reflect.Value, error) {
    el := reflect.ValueOf(v).Elem()
    fbn := el.FieldByName(s)
    if !fbn.IsValid() {
        return fbn, errors.Errorf("does not have field named %s", s)
    }
    return fbn, nil
}

func decode(data []byte, v interface{}, approvedFields []string) error {
    typeOf := reflect.TypeOf(v)
    if typeOf.Kind() == reflect.Ptr {
        typeOf = typeOf.Elem()
    }
    if typeOf.Kind() == reflect.Slice {
        return errors.New("does not support slices")
    }

    newItem := reflect.New(typeOf)
    newItemInterface := newItem.Interface()
    if err := json.Unmarshal(data, &newItemInterface); err != nil {
        return errors.Wrap(err, "json unmarshall")
    }
    for _, approvedField := range approvedFields {
        fbn, err := fieldByName(approvedField, v)
        if err != nil {
            return errors.Wrap(err, "field by name")
        }
        val, _ := fieldByName(approvedField, newItemInterface)
        fbn.Set(val)
    }
    return nil
}

тест: 

func TestBar1Bar2(t *testing.T) {
    var json_data []byte
    {
        f := &Foo{
            Bar1: 1,
            Bar2: 2,
            Bar3: 3,
        }
        json_data, _ = json.Marshal(f)
    }
    approved_fields := []string{"Bar1", "Bar2"}
    f := &Foo{}
    err := decode(json_data, f, approved_fields)
    if err != nil {
        t.Fatal(err)
    }
    assert.Equal(t, f.Bar1, 1)
    assert.Equal(t, f.Bar2, 2)
    assert.Equal(t, f.Bar3, 0)
}
...