Operator

包路径: github.com/caicloud/nirvana/definition

在每个 API Definition 中,都有一组 Parameters 和 Results。其中 Parameters 和业务函数的参数一一对应,而 Results 则和业务函数的返回值一一对应。一般情况下,如果我们需要对参数的合法性和结构进行验证和转换,那么就需要在业务函数中实现这些功能。为了让数据验证和结构转换等逻辑更加通用化和标准化,Nirvana 提供了 Operator 接口,可用于针对单个参数或返回值进行验证和修改:

  1. // Operator is used to operate an object and return an replacement object.
  2. //
  3. // For example:
  4. // A converter:
  5. // type ConverterForAnObject struct{}
  6. // func (c *ConverterForAnObject) Kind() {return "converter"}
  7. // func (c *ConverterForAnObject) In() reflect.Type {return definition.TypeOf(&ObjectV1{})}
  8. // func (c *ConverterForAnObject) Out() reflect.Type {return definition.TypeOf(&ObjectV2{})}
  9. // func (c *ConverterForAnObject) Operate(ctx context.Context, object interface{}) (interface{}, error) {
  10. // objV2, err := convertObjectV1ToObjectV2(object.(*ObjectV1))
  11. // return objV2, err
  12. // }
  13. //
  14. // A validator:
  15. // type ValidatorForAnObject struct{}
  16. // func (c *ValidatorForAnObject) Kind() {return "validator"}
  17. // func (c *ValidatorForAnObject) In() reflect.Type {return definition.TypeOf(&Object{})}
  18. // func (c *ValidatorForAnObject) Out() reflect.Type {return definition.TypeOf(&Object{})}
  19. // func (c *ValidatorForAnObject) Operate(ctx context.Context, object interface{}) (interface{}, error) {
  20. // if err := validate(object.(*Object)); err != nil {
  21. // return nil, err
  22. // }
  23. // return object, nil
  24. // }
  25. type Operator interface {
  26. // Kind indicates operator type.
  27. Kind() string
  28. // In returns the type of the only object parameter of operator.
  29. // The type must be a concrete struct or built-in type. It should
  30. // not be interface unless it's a file or stream reader.
  31. In() reflect.Type
  32. // Out returns the type of the only object result of operator.
  33. // The type must be a concrete struct or built-in type. It should
  34. // not be interface unless it's a file or stream reader.
  35. Out() reflect.Type
  36. // Operate operates an object and return one.
  37. Operate(ctx context.Context, field string, object interface{}) (interface{}, error)
  38. }

在没有 Operator 的情况下,Nirvana 从业务函数的参数中确定数据类型,并将请求数据转换为这个类型。但是如果设置了 Operator,那么 Nirvana 会从第一个 Operator 的 In() 方法获取类型, 并且会检查最后一个 Operator 的 Out() 类型是否和业务函数的参数类型一致。

在实际的使用过程中,并不需要实现这个复杂的接口。Nirvana 提供了两种类型的 Operator:Validator 和 Converter。

Validator

包路径: github.com/caicloud/nirvana/operators/validator

validator 包的实现基于 go-playground/validator,提供了用于生成 Operator 的方法:

  1. func Struct(instance interface{}) Validator
  2. func String(tag string) Validator
  3. func Int(tag string) Validator
  4. func Int64(tag string) Validator
  5. func Int32(tag string) Validator
  6. func Int16(tag string) Validator
  7. func Int8(tag string) Validator
  8. func Byte(tag string) Validator
  9. func Uint(tag string) Validator
  10. func Uint64(tag string) Validator
  11. func Uint32(tag string) Validator
  12. func Uint16(tag string) Validator
  13. func Uint8(tag string) Validator
  14. func Bool(tag string) Validator

对于结构体类型,在需要的字段上添加名为 validate 的 tag。

自定义验证器

有时候默认的验证器不能覆盖复杂的验证需求,因此 validator 包还提供了方法用于创建自定义验证器:

  1. // NewCustom calls f for validation, using description for doc gen.
  2. // User should only do custom validation in f.
  3. // Validations which can be done by other way should be done in another Operator.
  4. // Exp:
  5. // []definition.Operator{NewCustom(f,"custom validation description")}
  6. // f should be func(ctx context.Context, object AnyType) error
  7. func NewCustom(f interface{}, description string) Validator

验证器函数必须符合签名 func(ctx context.Context, object AnyType) error。其中 AnyType 是具体要验证的类型,不能使用接口。

Converter

包路径: github.com/caicloud/nirvana/operators/converter

除了对参数进行验证以外,在某些场景下还需要对参数进行类型转换。因此 converter 包提供了工具方法用于将转换函数包装成 Operator:

  1. // For creates converter for a converter func.
  2. //
  3. // A converter func should has signature:
  4. // func f(context.Context, string, AnyType) (AnyType, error)
  5. // The second parameter is a string that is used to generate error.
  6. // AnyType can be any type in go. But struct type and
  7. // built-in data type is recommended.
  8. func For(f interface{}) Converter

转换函数必须符合 func f(context.Context, string, AnyType) (AnyType, error)。其中参数的 AnyType 和返回值的 AnyType 可以不同。

在 Definition 中使用 Operator

这是一个在 List Messages 的 API 中添加 Operator 的示例:

  1. // Definition
  2. var listMessages = def.Definition{
  3. Method: def.List,
  4. Summary: "List Messages",
  5. Description: "Query a specified number of messages and returns an array",
  6. Function: message.ListMessages,
  7. Parameters: []def.Parameter{
  8. {
  9. Source: def.Query,
  10. Name: "count",
  11. Default: 10,
  12. Operators: []def.Operator{
  13. validator.Int("min=1"),
  14. converter.For(func(ctx context.Context, field string, value int) (uint, error) {
  15. return uint(value), nil
  16. }),
  17. },
  18. Description: "Number of messages",
  19. },
  20. },
  21. Results: def.DataErrorResults("A list of messages"),
  22. }
  23. // 业务函数
  24. // ListMessages returns all messages.
  25. func ListMessages(ctx context.Context, count uint) ([]Message, error) {
  26. messages := make([]Message, count)
  27. for i := 0; i < int(count); i++ {
  28. messages[i].ID = i
  29. messages[i].Title = fmt.Sprintf("%s %d", m.Title, i)
  30. messages[i].Content = fmt.Sprintf("%s %d", m.Content, i)
  31. }
  32. return messages, nil
  33. }

这个例子中,验证器要求 count 的最小值为 1,并且把 int 类型转换为了 uint 类型。业务函数的参数也响应的变成了 uint 类型。

注意:Operator 是链式调用的,也就是说上一个 Operator 的返回值会作为下一个 Operator 的参数。因此如果将验证器和转换器的顺序调换,则需要将验证器的类型修改为转换器返回的类型:

  1. Operators: []def.Operator{
  2. converter.For(func(ctx context.Context, field string, value int) (uint, error) {
  3. return uint(value), nil
  4. }),
  5. validator.Uint("min=1"),
  6. },

但是一般情况下,始终建议验证器放在前面,转换器放在后面。