Skip to content

Commit

Permalink
handle bool pointer override
Browse files Browse the repository at this point in the history
Signed-off-by: Nicolas De Loof <nicolas.deloof@gmail.com>
  • Loading branch information
ndeloof committed Nov 30, 2022
1 parent 999ce24 commit 9670eb3
Show file tree
Hide file tree
Showing 2 changed files with 52 additions and 1 deletion.
14 changes: 13 additions & 1 deletion loader/merge.go
Expand Up @@ -43,6 +43,15 @@ var serviceSpecials = &specials{
}

func (s *specials) Transformer(t reflect.Type) func(dst, src reflect.Value) error {
// TODO this is a workaround waiting for imdario/mergo#131
if t.Kind() == reflect.Pointer && t.Elem().Kind() == reflect.Bool {
return func(dst, src reflect.Value) error {
if dst.CanSet() && !src.IsNil() {
dst.Set(src)
}
return nil
}
}
if fn, ok := s.m[t]; ok {
return fn
}
Expand Down Expand Up @@ -113,7 +122,10 @@ func mergeServices(base, override []types.ServiceConfig) ([]types.ServiceConfig,
}

func _merge(baseService *types.ServiceConfig, overrideService *types.ServiceConfig) (*types.ServiceConfig, error) {
if err := mergo.Merge(baseService, overrideService, mergo.WithAppendSlice, mergo.WithOverride, mergo.WithTransformers(serviceSpecials)); err != nil {
if err := mergo.Merge(baseService, overrideService,
mergo.WithAppendSlice,
mergo.WithOverride,
mergo.WithTransformers(serviceSpecials)); err != nil {
return nil, err
}
if overrideService.Command != nil {
Expand Down
39 changes: 39 additions & 0 deletions loader/merge_test.go
Expand Up @@ -1110,6 +1110,45 @@ func TestMergeUlimitsConfig(t *testing.T) {
)
}

func TestInitOverride(t *testing.T) {
var (
bt = true
bf = false
)
cases := []struct {
base *bool
override *bool
expect bool
}{
{
base: &bt,
override: &bf,
expect: false,
},
{
base: nil,
override: &bt,
expect: true,
},
{
base: &bt,
override: nil,
expect: true,
},
}
for _, test := range cases {
base := types.ServiceConfig{
Init: test.base,
}
override := types.ServiceConfig{
Init: test.override,
}
config, err := _merge(&base, &override)
assert.NilError(t, err)
assert.Check(t, *config.Init == test.expect)
}
}

func TestMergeServiceNetworkConfig(t *testing.T) {
specials := &specials{
m: map[reflect.Type]func(dst, src reflect.Value) error{
Expand Down

0 comments on commit 9670eb3

Please sign in to comment.