feat: 在结构体转换中添加字符串到时间Duration的钩子函数

Author: mingzaily

complex/hooks.go

@@ -25,6 +25,22 @@ func stringToTimeHookFunc() HookFunc {
 	}
 }
 
+// stringToDurationHookFunc returns a HookFunc that converts string to time.Duration.
+func stringToDurationHookFunc() HookFunc {
+	return func(from reflect.Type, to reflect.Type, data interface{}) (interface{}, error) {
+		if from.Kind() != reflect.String || to != reflect.TypeOf(time.Duration(0)) {
+			return data, nil
+		}
+
+		s, ok := data.(string)
+		if !ok {
+			return data, nil
+		}
+
+		return basic.ToDurationE(s)
+	}
+}
+
 // intToBoolHookFunc returns a HookFunc that converts integer to bool.
 func intToBoolHookFunc() HookFunc {
 	return func(from reflect.Type, to reflect.Type, data interface{}) (interface{}, error) {

complex/struct.go

@@ -33,7 +33,7 @@ func StructE(source, pointer interface{}, hooks ...HookFunc) error {
 	}
 
 	// Prepend default hooks
-	allHooks := []HookFunc{stringToTimeHookFunc(), intToBoolHookFunc()}
+	allHooks := []HookFunc{stringToTimeHookFunc(), stringToDurationHookFunc(), intToBoolHookFunc()}
 	allHooks = append(allHooks, hooks...)
 
 	// Get the reflect.Value of the pointer and the struct

test/struct_test.go

@@ -48,6 +48,11 @@ type UserWithBoolFloat struct {
 	Balance  float32
 }
 
+type TestDuration struct {
+	ID       int
+	Duration time.Duration `mconv:"duration"`
+}
+
 func TestStruct(t *testing.T) {
 	t.Run("SimpleConversion", func(t *testing.T) {
 		source := map[string]interface{}{"ID": 1, "Name": "Alice"}
@@ -172,6 +177,24 @@ func TestStruct(t *testing.T) {
 		}
 	})
 
+	t.Run("BuiltInStringToDurationHook", func(t *testing.T) {
+		source := map[string]interface{}{
+			"duration": "30s",
+		}
+		var target TestDuration
+		err := complex.StructE(source, &target)
+		if err != nil {
+			t.Fatalf("unexpected error: %v", err)
+		}
+
+		expected := TestDuration{
+			Duration: 30 * time.Second,
+		}
+		if !reflect.DeepEqual(target, expected) {
+			t.Errorf("expected %+v, got %+v", expected, target)
+		}
+	})
+
 	t.Run("CustomHook", func(t *testing.T) {
 		type HookUser struct {
 			ID     int