mirror of
https://github.com/ent/ent.git
synced 2026-05-22 09:31:45 +03:00
50 lines
1.1 KiB
Go
50 lines
1.1 KiB
Go
// 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 edge
|
|
|
|
import "entgo.io/ent/schema"
|
|
|
|
// Annotation is a builtin schema annotation for
|
|
// configuring the edges' behavior in codegen.
|
|
type Annotation struct {
|
|
// The StructTag option allows overriding the struct-tag
|
|
// of the `Edges` field in the generated entity. For example:
|
|
//
|
|
// edge.Annotation{
|
|
// StructTag: `json:"pet_edges"`
|
|
// }
|
|
//
|
|
StructTag string
|
|
}
|
|
|
|
// Name describes the annotation name.
|
|
func (Annotation) Name() string {
|
|
return "Edges"
|
|
}
|
|
|
|
// Merge implements the schema.Merger interface.
|
|
func (a Annotation) Merge(other schema.Annotation) schema.Annotation {
|
|
var ant Annotation
|
|
switch other := other.(type) {
|
|
case Annotation:
|
|
ant = other
|
|
case *Annotation:
|
|
if other != nil {
|
|
ant = *other
|
|
}
|
|
default:
|
|
return a
|
|
}
|
|
if tag := ant.StructTag; tag != "" {
|
|
a.StructTag = tag
|
|
}
|
|
return a
|
|
}
|
|
|
|
var (
|
|
_ schema.Annotation = (*Annotation)(nil)
|
|
_ schema.Merger = (*Annotation)(nil)
|
|
)
|