Skip to content

Commit c6c8ccf

Browse files
committed
Move exported Build and Update functions to the top
and helper functions to the bottom
1 parent 008183c commit c6c8ccf

File tree

1 file changed

+35
-35
lines changed

1 file changed

+35
-35
lines changed

internal/resource/admin_secret.go

+35-35
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ package resource
1212
import (
1313
"bytes"
1414
"fmt"
15+
1516
"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"
1617

1718
rabbitmqv1beta1 "github.com/rabbitmq/cluster-operator/api/v1beta1"
@@ -38,33 +39,34 @@ func (builder *RabbitmqResourceBuilder) AdminSecret() *AdminSecretBuilder {
3839
}
3940
}
4041

41-
func generateDefaultUserConf(username, password string) ([]byte, error) {
42-
43-
ini.PrettySection = false // Remove trailing new line because default_user.conf has only a default section.
44-
cfg, err := ini.Load([]byte{})
42+
func (builder *AdminSecretBuilder) Build() (runtime.Object, error) {
43+
username, err := randomEncodedString(24)
4544
if err != nil {
4645
return nil, err
4746
}
48-
defaultSection := cfg.Section("")
49-
50-
if _, err := defaultSection.NewKey("default_user", username); err != nil {
51-
return nil, err
52-
}
5347

54-
if _, err := defaultSection.NewKey("default_pass", password); err != nil {
48+
password, err := randomEncodedString(24)
49+
if err != nil {
5550
return nil, err
5651
}
5752

58-
var userConfBuffer bytes.Buffer
59-
if _, err := cfg.WriteTo(&userConfBuffer); err != nil {
53+
defaultUserConf, err := generateDefaultUserConf(username, password)
54+
if err != nil {
6055
return nil, err
6156
}
6257

63-
return userConfBuffer.Bytes(), nil
64-
}
65-
66-
func (builder *AdminSecretBuilder) UpdateRequiresStsRestart() bool {
67-
return false
58+
return &corev1.Secret{
59+
ObjectMeta: metav1.ObjectMeta{
60+
Name: builder.Instance.ChildResourceName(AdminSecretName),
61+
Namespace: builder.Instance.Namespace,
62+
},
63+
Type: corev1.SecretTypeOpaque,
64+
Data: map[string][]byte{
65+
"username": []byte(username),
66+
"password": []byte(password),
67+
"default_user.conf": defaultUserConf,
68+
},
69+
}, nil
6870
}
6971

7072
func (builder *AdminSecretBuilder) Update(object runtime.Object) error {
@@ -79,32 +81,30 @@ func (builder *AdminSecretBuilder) Update(object runtime.Object) error {
7981
return nil
8082
}
8183

82-
func (builder *AdminSecretBuilder) Build() (runtime.Object, error) {
83-
username, err := randomEncodedString(24)
84+
func (builder *AdminSecretBuilder) UpdateRequiresStsRestart() bool {
85+
return false
86+
}
87+
88+
func generateDefaultUserConf(username, password string) ([]byte, error) {
89+
ini.PrettySection = false // Remove trailing new line because default_user.conf has only a default section.
90+
cfg, err := ini.Load([]byte{})
8491
if err != nil {
8592
return nil, err
8693
}
94+
defaultSection := cfg.Section("")
8795

88-
password, err := randomEncodedString(24)
89-
if err != nil {
96+
if _, err := defaultSection.NewKey("default_user", username); err != nil {
9097
return nil, err
9198
}
9299

93-
defaultUserConf, err := generateDefaultUserConf(username, password)
94-
if err != nil {
100+
if _, err := defaultSection.NewKey("default_pass", password); err != nil {
95101
return nil, err
96102
}
97103

98-
return &corev1.Secret{
99-
ObjectMeta: metav1.ObjectMeta{
100-
Name: builder.Instance.ChildResourceName(AdminSecretName),
101-
Namespace: builder.Instance.Namespace,
102-
},
103-
Type: corev1.SecretTypeOpaque,
104-
Data: map[string][]byte{
105-
"username": []byte(username),
106-
"password": []byte(password),
107-
"default_user.conf": defaultUserConf,
108-
},
109-
}, nil
104+
var userConfBuffer bytes.Buffer
105+
if _, err := cfg.WriteTo(&userConfBuffer); err != nil {
106+
return nil, err
107+
}
108+
109+
return userConfBuffer.Bytes(), nil
110110
}

0 commit comments

Comments
 (0)