golang/ent
1
0
Fork 0
mirror of https://github.com/ent/ent.git synced 2026-05-24 09:31:56 +03:00
Code Issues Packages Projects Releases Wiki Activity

entc/cmd/entc: support more numeric types for ids

Browse Source 
Reviewed By: alexsn

Differential Revision: D17527952

fbshipit-source-id: 1465b054531cee8ca1dd445b6654063f32f4c4e3
This commit is contained in:
Ariel Mashraki
2019-09-23 09:46:48 -07:00
committed by Facebook Github Bot
parent 3f7553af20
commit 0f71ca6ad8
25 changed files with 2984 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
entc/gen/template/dialect/sql/create.tmpl
View File

@@ -40,7 +40,7 @@ func ({{ $receiver }} *{{ $builder }}) sqlSave(ctx context.Context) (*{{ $.Name
if err != nil {
return nil, rollback(tx, err)
}
{{ $.Receiver }}.ID = {{ if $.ID.IsString }}strconv.FormatInt(id, 10){{ else }}int(id){{ end }}
{{ $.Receiver }}.ID = {{ if $.ID.IsString }}strconv.FormatInt(id, 10){{ else }}{{ $.ID.Type }}(id){{ end }}
{{- range $_, $e := $.Edges }}
if len({{ $receiver }}.{{ $e.StructField }}) > 0 {
{{- if and $e.Unique $e.SelfRef }}{{/* O2O with self reference */}}

4
entc/gen/template/dialect/sql/update.tmpl
View File

@@ -35,7 +35,7 @@ func ({{ $receiver }} *{{ $builder }}) sqlSave(ctx context.Context) ({{ $ret }}
if err := {{ $.Receiver }}.FromRows(rows); err != nil {
return {{ $zero }}, fmt.Errorf("{{ $pkg }}: failed scanning row into {{ $.Name }}: %v", err)
}
id = {{ $.Receiver }}.{{- if $.ID.IsString }}id(){{ else }}ID{{ end }}
id = {{ if $.ID.IsString }}{{ $.Receiver }}.id(){{ else if not $.ID.IsInt }}int({{ $.Receiver }}.ID){{ else }}{{ $.Receiver }}.ID{{ end }}
{{- else }}
if err := rows.Scan(&id); err != nil {
return {{ $zero }}, fmt.Errorf("{{ $pkg }}: failed reading id: %v", err)
@@ -311,5 +311,7 @@ func ({{ $receiver }} *{{ $builder }}) sqlSave(ctx context.Context) ({{ $ret }}
err = rollback(tx, serr)
return {{/* return is not knwon at this point. */}}
}
{{- else if not $.Type.ID.IsInt }}
eid := int(eid)
{{- end }}
{{ end }}