Browse Source

allow removing default WebRTC ICE server with environment variables (#2064)

pull/2065/head
Alessandro Ros 3 years ago committed by GitHub
parent
commit
1398f42e08
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 6
      internal/conf/env/env.go
  2. 6
      internal/conf/env/env_test.go

6
internal/conf/env/env.go vendored

@ -150,7 +150,7 @@ func loadEnvInternal(env map[string]string, prefix string, rv reflect.Value) err
if rt.Elem() == reflect.TypeOf("") { if rt.Elem() == reflect.TypeOf("") {
if ev, ok := env[prefix]; ok { if ev, ok := env[prefix]; ok {
if ev == "" { if ev == "" {
rv.Set(reflect.ValueOf([]string{})) rv.Set(reflect.MakeSlice(rv.Type(), 0, 0))
} else { } else {
rv.Set(reflect.ValueOf(strings.Split(ev, ","))) rv.Set(reflect.ValueOf(strings.Split(ev, ",")))
} }
@ -159,6 +159,9 @@ func loadEnvInternal(env map[string]string, prefix string, rv reflect.Value) err
} }
if rt.Elem().Kind() == reflect.Struct { if rt.Elem().Kind() == reflect.Struct {
if ev, ok := env[prefix]; ok && ev == "" { // special case: empty list
rv.Set(reflect.MakeSlice(rv.Type(), 0, 0))
} else {
for i := 0; ; i++ { for i := 0; ; i++ {
itemPrefix := prefix + "_" + strconv.FormatInt(int64(i), 10) itemPrefix := prefix + "_" + strconv.FormatInt(int64(i), 10)
if !envHasAtLeastAKeyWithPrefix(env, itemPrefix) { if !envHasAtLeastAKeyWithPrefix(env, itemPrefix) {
@ -173,6 +176,7 @@ func loadEnvInternal(env map[string]string, prefix string, rv reflect.Value) err
rv.Set(reflect.Append(rv, elem.Elem())) rv.Set(reflect.Append(rv, elem.Elem()))
} }
}
return nil return nil
} }
} }

6
internal/conf/env/env_test.go vendored

@ -56,6 +56,7 @@ type testStruct struct {
MySlice []string MySlice []string
MySliceEmpty []string MySliceEmpty []string
MySliceSubStruct []mySubStruct MySliceSubStruct []mySubStruct
MySliceSubStructEmpty []mySubStruct
} }
func TestLoad(t *testing.T) { func TestLoad(t *testing.T) {
@ -104,6 +105,9 @@ func TestLoad(t *testing.T) {
os.Setenv("MYPREFIX_MYSLICESUBSTRUCT_1_PASSWORD", "pass2") os.Setenv("MYPREFIX_MYSLICESUBSTRUCT_1_PASSWORD", "pass2")
defer os.Unsetenv("MYPREFIX_MYSLICESUBSTRUCT_1_PASSWORD") defer os.Unsetenv("MYPREFIX_MYSLICESUBSTRUCT_1_PASSWORD")
os.Setenv("MYPREFIX_MYSLICESUBSTRUCTEMPTY", "")
defer os.Unsetenv("MYPREFIX_MYSLICESUBSTRUCTEMPTY")
var s testStruct var s testStruct
err := Load("MYPREFIX", &s) err := Load("MYPREFIX", &s)
require.NoError(t, err) require.NoError(t, err)
@ -136,4 +140,6 @@ func TestLoad(t *testing.T) {
Password: "pass2", Password: "pass2",
}, },
}, s.MySliceSubStruct) }, s.MySliceSubStruct)
require.Equal(t, []mySubStruct{}, s.MySliceSubStructEmpty)
} }

Loading…
Cancel
Save