-
Notifications
You must be signed in to change notification settings - Fork 1.2k
/
error_model.go
146 lines (109 loc) · 3.18 KB
/
error_model.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
// Code generated by go-swagger; DO NOT EDIT.
package cli
// This file was generated by the swagger tool.
// Editing this file might prove futile when you re-run the swagger generate command
import (
"fmt"
"github.com/go-swagger/go-swagger/examples/cli/models"
"github.com/spf13/cobra"
)
// Schema cli for Error
// register flags to command
func registerModelErrorFlags(depth int, cmdPrefix string, cmd *cobra.Command) error {
if err := registerErrorCode(depth, cmdPrefix, cmd); err != nil {
return err
}
if err := registerErrorMessage(depth, cmdPrefix, cmd); err != nil {
return err
}
return nil
}
func registerErrorCode(depth int, cmdPrefix string, cmd *cobra.Command) error {
if depth > maxDepth {
return nil
}
codeDescription := ``
var codeFlagName string
if cmdPrefix == "" {
codeFlagName = "code"
} else {
codeFlagName = fmt.Sprintf("%v.code", cmdPrefix)
}
var codeFlagDefault int64
_ = cmd.PersistentFlags().Int64(codeFlagName, codeFlagDefault, codeDescription)
return nil
}
func registerErrorMessage(depth int, cmdPrefix string, cmd *cobra.Command) error {
if depth > maxDepth {
return nil
}
messageDescription := `Required. `
var messageFlagName string
if cmdPrefix == "" {
messageFlagName = "message"
} else {
messageFlagName = fmt.Sprintf("%v.message", cmdPrefix)
}
var messageFlagDefault string
_ = cmd.PersistentFlags().String(messageFlagName, messageFlagDefault, messageDescription)
return nil
}
// retrieve flags from commands, and set value in model. Return true if any flag is passed by user to fill model field.
func retrieveModelErrorFlags(depth int, m *models.Error, cmdPrefix string, cmd *cobra.Command) (error, bool) {
retAdded := false
err, codeAdded := retrieveErrorCodeFlags(depth, m, cmdPrefix, cmd)
if err != nil {
return err, false
}
retAdded = retAdded || codeAdded
err, messageAdded := retrieveErrorMessageFlags(depth, m, cmdPrefix, cmd)
if err != nil {
return err, false
}
retAdded = retAdded || messageAdded
return nil, retAdded
}
func retrieveErrorCodeFlags(depth int, m *models.Error, cmdPrefix string, cmd *cobra.Command) (error, bool) {
if depth > maxDepth {
return nil, false
}
retAdded := false
codeFlagName := fmt.Sprintf("%v.code", cmdPrefix)
if cmd.Flags().Changed(codeFlagName) {
var codeFlagName string
if cmdPrefix == "" {
codeFlagName = "code"
} else {
codeFlagName = fmt.Sprintf("%v.code", cmdPrefix)
}
codeFlagValue, err := cmd.Flags().GetInt64(codeFlagName)
if err != nil {
return err, false
}
m.Code = codeFlagValue
retAdded = true
}
return nil, retAdded
}
func retrieveErrorMessageFlags(depth int, m *models.Error, cmdPrefix string, cmd *cobra.Command) (error, bool) {
if depth > maxDepth {
return nil, false
}
retAdded := false
messageFlagName := fmt.Sprintf("%v.message", cmdPrefix)
if cmd.Flags().Changed(messageFlagName) {
var messageFlagName string
if cmdPrefix == "" {
messageFlagName = "message"
} else {
messageFlagName = fmt.Sprintf("%v.message", cmdPrefix)
}
messageFlagValue, err := cmd.Flags().GetString(messageFlagName)
if err != nil {
return err, false
}
m.Message = &messageFlagValue
retAdded = true
}
return nil, retAdded
}