entgql: fixed pulid example

Signed-off-by: Giau. Tran Minh <hello@giautm.dev>
ent · Feb 15, 2023 · 6fa0371 · 6fa0371
1 parent e104430
commit 6fa0371
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/entgql/internal/todopulid/ent/schema/pulid/pulid.go b/entgql/internal/todopulid/ent/schema/pulid/pulid.go
@@ -62,11 +62,14 @@ func (u *ID) Scan(src interface{}) error {
 	if src == nil {
 		return fmt.Errorf("pulid: expected a value")
 	}
-	s, ok := src.(string)
-	if !ok {
-		return fmt.Errorf("pulid: expected a string")
+	switch src := src.(type) {
+	case string:
+		*u = ID(src)
+	case ID:
+		*u = src
+	default:
+		return fmt.Errorf("pulid: unexpected type, %T", src)
 	}
-	*u = ID(s)
 	return nil
 }