Matthias Andreas Benkard | 832a54e | 2019-01-29 09:27:38 +0100 | [diff] [blame] | 1 | package swagger |
| 2 | |
| 3 | // Copyright 2015 Ernest Micklei. All rights reserved. |
| 4 | // Use of this source code is governed by a license |
| 5 | // that can be found in the LICENSE file. |
| 6 | |
| 7 | import ( |
| 8 | "bytes" |
| 9 | "encoding/json" |
| 10 | ) |
| 11 | |
| 12 | // NamedModel associates a name with a Model (not using its Id) |
| 13 | type NamedModel struct { |
| 14 | Name string |
| 15 | Model Model |
| 16 | } |
| 17 | |
| 18 | // ModelList encapsulates a list of NamedModel (association) |
| 19 | type ModelList struct { |
| 20 | List []NamedModel |
| 21 | } |
| 22 | |
| 23 | // Put adds or replaces a Model by its name |
| 24 | func (l *ModelList) Put(name string, model Model) { |
| 25 | for i, each := range l.List { |
| 26 | if each.Name == name { |
| 27 | // replace |
| 28 | l.List[i] = NamedModel{name, model} |
| 29 | return |
| 30 | } |
| 31 | } |
| 32 | // add |
| 33 | l.List = append(l.List, NamedModel{name, model}) |
| 34 | } |
| 35 | |
| 36 | // At returns a Model by its name, ok is false if absent |
| 37 | func (l *ModelList) At(name string) (m Model, ok bool) { |
| 38 | for _, each := range l.List { |
| 39 | if each.Name == name { |
| 40 | return each.Model, true |
| 41 | } |
| 42 | } |
| 43 | return m, false |
| 44 | } |
| 45 | |
| 46 | // Do enumerates all the models, each with its assigned name |
| 47 | func (l *ModelList) Do(block func(name string, value Model)) { |
| 48 | for _, each := range l.List { |
| 49 | block(each.Name, each.Model) |
| 50 | } |
| 51 | } |
| 52 | |
| 53 | // MarshalJSON writes the ModelList as if it was a map[string]Model |
| 54 | func (l ModelList) MarshalJSON() ([]byte, error) { |
| 55 | var buf bytes.Buffer |
| 56 | encoder := json.NewEncoder(&buf) |
| 57 | buf.WriteString("{\n") |
| 58 | for i, each := range l.List { |
| 59 | buf.WriteString("\"") |
| 60 | buf.WriteString(each.Name) |
| 61 | buf.WriteString("\": ") |
| 62 | encoder.Encode(each.Model) |
| 63 | if i < len(l.List)-1 { |
| 64 | buf.WriteString(",\n") |
| 65 | } |
| 66 | } |
| 67 | buf.WriteString("}") |
| 68 | return buf.Bytes(), nil |
| 69 | } |
| 70 | |
| 71 | // UnmarshalJSON reads back a ModelList. This is an expensive operation. |
| 72 | func (l *ModelList) UnmarshalJSON(data []byte) error { |
| 73 | raw := map[string]interface{}{} |
| 74 | json.NewDecoder(bytes.NewReader(data)).Decode(&raw) |
| 75 | for k, v := range raw { |
| 76 | // produces JSON bytes for each value |
| 77 | data, err := json.Marshal(v) |
| 78 | if err != nil { |
| 79 | return err |
| 80 | } |
| 81 | var m Model |
| 82 | json.NewDecoder(bytes.NewReader(data)).Decode(&m) |
| 83 | l.Put(k, m) |
| 84 | } |
| 85 | return nil |
| 86 | } |