-
Notifications
You must be signed in to change notification settings - Fork 62
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Pull request #53: Get fixture fields with reflection
Merge in AW/go-starter from aj/fixture-fields-with-reflection to master * commit '7ef261e577f3c7b60125525e4494795a9c04ea06': Update changelog fix CVE-2022-41723 vulnerability Remove not needed test snapshot Fix linter Use GetFieldsImplementing with Upsertable Remove fixture dependency in struct tests Use generics fix trivy CVE-2022-41723 vulnerability get fixture insertable fields with reflection
- Loading branch information
Showing
7 changed files
with
329 additions
and
34 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,83 @@ | ||
| package util | ||
|
|
||
| import ( | ||
| "errors" | ||
| "reflect" | ||
| ) | ||
|
|
||
| // GetFieldsImplementing returns all fields of a struct implementing a certain interface. | ||
| // Returned fields are pointers to a type or interface objects. | ||
| // | ||
| // Parameter structPtr must be a pointer to a struct. | ||
| // Parameter interfaceObject must be given as a pointer to an interface, | ||
| // for example (*Insertable)(nil), where Insertable is an interface name. | ||
| func GetFieldsImplementing[T any](structPtr interface{}, interfaceObject *T) ([]T, error) { | ||
|
|
||
| // Verify if structPtr is a pointer to a struct | ||
| inputParamStructType := reflect.TypeOf(structPtr) | ||
| if inputParamStructType == nil || | ||
| inputParamStructType.Kind() != reflect.Ptr || | ||
| inputParamStructType.Elem().Kind() != reflect.Struct { | ||
| return nil, errors.New("invalid input structPtr param: should be a pointer to a struct") | ||
| } | ||
|
|
||
| inputParamIfcType := reflect.TypeOf(interfaceObject) | ||
| // Verify if interfaceObject is a pointer to an interface | ||
| if inputParamIfcType == nil || | ||
| inputParamIfcType.Kind() != reflect.Ptr || | ||
| inputParamIfcType.Elem().Kind() != reflect.Interface { | ||
|
|
||
| return nil, errors.New("invalid input interfaceObject param: should be a pointer to an interface") | ||
| } | ||
|
|
||
| // We need the type, not the pointer to it. | ||
| // By using Elem() we can get the value pointed by the pointer. | ||
| interfaceType := inputParamIfcType.Elem() | ||
| structType := inputParamStructType.Elem() | ||
|
|
||
| structValue := reflect.ValueOf(structPtr).Elem() | ||
|
|
||
| retFields := make([]T, 0) | ||
|
|
||
| // Getting the VisibleFields returns all public fields in the struct | ||
| for i, field := range reflect.VisibleFields(structType) { | ||
|
|
||
| // Check if the field can be exported. | ||
| // Interface() can be called only on exportable fields. | ||
| if !field.IsExported() { | ||
| continue | ||
| } | ||
|
|
||
| fieldValue := structValue.Field(i) | ||
|
|
||
| // Depending on the field type, different checks apply. | ||
| switch field.Type.Kind() { | ||
|
|
||
| case reflect.Pointer: | ||
|
|
||
| // Let's check if it implements the interface. | ||
| if field.Type.Implements(interfaceType) { | ||
| // Great, we can add it to the return slice | ||
| retFields = append(retFields, fieldValue.Interface().(T)) | ||
| } | ||
|
|
||
| case reflect.Interface: | ||
| // If it's an interface, make sure it's not nil. | ||
| if fieldValue.IsNil() { | ||
| continue | ||
| } | ||
|
|
||
| // Now we can check if it's the same interface. | ||
| if field.Type.Implements(interfaceType) { | ||
| // Great, we can add it to the return slice | ||
| retFields = append(retFields, fieldValue.Interface().(T)) | ||
| } | ||
|
|
||
| default: | ||
| // We can skip any other cases. | ||
| continue | ||
| } | ||
| } | ||
|
|
||
| return retFields, nil | ||
| } |
Oops, something went wrong.