Skip to content

Commit 20b4fe6

Browse files
sebrandon1klaskosk
authored andcommitted
nmstate: update create() and returns (openshift-kni#855)
1 parent 8e61ab3 commit 20b4fe6

File tree

2 files changed

+28
-20
lines changed

2 files changed

+28
-20
lines changed

pkg/nmstate/nmstate.go

+15-12
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@ func (builder *Builder) Get() (*nmstateV1.NMState, error) {
9999
return nil, err
100100
}
101101

102-
return nmstate, err
102+
return nmstate, nil
103103
}
104104

105105
// Create makes a NMState in the cluster and stores the created object in struct.
@@ -110,15 +110,18 @@ func (builder *Builder) Create() (*Builder, error) {
110110

111111
glog.V(100).Infof("Creating the NMState %s", builder.Definition.Name)
112112

113-
var err error
114-
if !builder.Exists() {
115-
err = builder.apiClient.Create(context.TODO(), builder.Definition)
116-
if err == nil {
117-
builder.Object = builder.Definition
118-
}
113+
if builder.Exists() {
114+
return builder, nil
119115
}
120116

121-
return builder, err
117+
err := builder.apiClient.Create(context.TODO(), builder.Definition)
118+
if err != nil {
119+
return builder, err
120+
}
121+
122+
builder.Object = builder.Definition
123+
124+
return builder, nil
122125
}
123126

124127
// Delete removes NMState object from a cluster.
@@ -197,7 +200,7 @@ func PullNMstate(apiClient *clients.Settings, name string) (*Builder, error) {
197200
return nil, err
198201
}
199202

200-
builder := Builder{
203+
builder := &Builder{
201204
apiClient: apiClient.Client,
202205
Definition: &nmstateV1.NMState{
203206
ObjectMeta: metav1.ObjectMeta{
@@ -218,7 +221,7 @@ func PullNMstate(apiClient *clients.Settings, name string) (*Builder, error) {
218221

219222
builder.Definition = builder.Object
220223

221-
return &builder, nil
224+
return builder, nil
222225
}
223226

224227
// validate will check that the builder and builder definition are properly initialized before
@@ -235,13 +238,13 @@ func (builder *Builder) validate() (bool, error) {
235238
if builder.Definition == nil {
236239
glog.V(100).Infof("The %s is undefined", resourceCRD)
237240

238-
builder.errorMsg = msg.UndefinedCrdObjectErrString(resourceCRD)
241+
return false, fmt.Errorf(msg.UndefinedCrdObjectErrString(resourceCRD))
239242
}
240243

241244
if builder.apiClient == nil {
242245
glog.V(100).Infof("The %s builder apiclient is nil", resourceCRD)
243246

244-
builder.errorMsg = fmt.Sprintf("%s builder cannot have nil apiClient", resourceCRD)
247+
return false, fmt.Errorf("%s builder cannot have nil apiClient", resourceCRD)
245248
}
246249

247250
if builder.errorMsg != "" {

pkg/nmstate/policy.go

+13-8
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,8 @@ func NewPolicyBuilder(apiClient *clients.Settings, name string, nodeSelector map
8686
glog.V(100).Infof("The nodeSelector of the NodeNetworkConfigurationPolicy is empty")
8787

8888
builder.errorMsg = "nodeNetworkConfigurationPolicy 'nodeSelector' cannot be empty map"
89+
90+
return builder
8991
}
9092

9193
return builder
@@ -112,7 +114,7 @@ func (builder *PolicyBuilder) Get() (*nmstateV1.NodeNetworkConfigurationPolicy,
112114
return nil, err
113115
}
114116

115-
return nmstatePolicy, err
117+
return nmstatePolicy, nil
116118
}
117119

118120
// Exists checks whether the given NodeNetworkConfigurationPolicy exists.
@@ -139,15 +141,18 @@ func (builder *PolicyBuilder) Create() (*PolicyBuilder, error) {
139141

140142
glog.V(100).Infof("Creating the NodeNetworkConfigurationPolicy %s", builder.Definition.Name)
141143

142-
var err error
143-
if !builder.Exists() {
144-
err = builder.apiClient.Create(context.TODO(), builder.Definition)
145-
if err == nil {
146-
builder.Object = builder.Definition
147-
}
144+
if builder.Exists() {
145+
return builder, nil
148146
}
149147

150-
return builder, err
148+
err := builder.apiClient.Create(context.TODO(), builder.Definition)
149+
if err != nil {
150+
return builder, err
151+
}
152+
153+
builder.Object = builder.Definition
154+
155+
return builder, nil
151156
}
152157

153158
// Delete removes NodeNetworkConfigurationPolicy object from a cluster.

0 commit comments

Comments
 (0)