snapshot of https://github.com/ent/ent.git (master, 2026-09-15; blobs > 131072B omitted)

entc/integration/customid/ent/schema/doc.go

@@ -0,0 +1,74 @@
+// Copyright 2019-present Facebook Inc. All rights reserved.
+// This source code is licensed under the Apache 2.0 license found
+// in the LICENSE file in the root directory of this source tree.
+
+package schema
+
+import (
+	"database/sql/driver"
+	"fmt"
+
+	"entgo.io/ent"
+	"entgo.io/ent/dialect"
+	"entgo.io/ent/schema/edge"
+	"entgo.io/ent/schema/field"
+	"github.com/google/uuid"
+
+	"ariga.io/atlas/sql/postgres"
+)
+
+// Doc holds the schema definition for the Doc entity.
+type Doc struct {
+	ent.Schema
+}
+
+// Fields of the Doc.
+func (Doc) Fields() []ent.Field {
+	return []ent.Field{
+		field.String("id").
+			GoType(DocID("")).
+			MaxLen(36).
+			NotEmpty().
+			Unique().
+			Immutable().
+			DefaultFunc(func() DocID {
+				return DocID(uuid.NewString())
+			}).
+			SchemaType(map[string]string{
+				dialect.Postgres: postgres.TypeUUID,
+			}),
+		field.String("text").
+			Optional(),
+	}
+}
+
+// Edges of the Doc.
+func (Doc) Edges() []ent.Edge {
+	return []ent.Edge{
+		edge.To("children", Doc.Type).
+			From("parent").
+			Unique(),
+		edge.To("related", Doc.Type),
+	}
+}
+
+type DocID string
+
+// Scan implements the Scanner interface.
+func (s *DocID) Scan(value any) (err error) {
+	switch v := value.(type) {
+	case nil:
+	case []byte:
+		*s = DocID(v)
+	case string:
+		*s = DocID(v)
+	default:
+		err = fmt.Errorf("unexpected type %T", v)
+	}
+	return
+}
+
+// Value implements the driver Valuer interface.
+func (s DocID) Value() (driver.Value, error) {
+	return string(s), nil
+}