feat: ent select fields.
This commit is contained in:
25
entgo/query/select.go
Normal file
25
entgo/query/select.go
Normal file
@@ -0,0 +1,25 @@
|
||||
package entgo
|
||||
|
||||
import (
|
||||
"entgo.io/ent/dialect/sql"
|
||||
"github.com/tx7do/go-utils/stringcase"
|
||||
)
|
||||
|
||||
func BuildFieldSelect(s *sql.Selector, fields []string) {
|
||||
if len(fields) > 0 {
|
||||
for i, field := range fields {
|
||||
fields[i] = stringcase.ToSnakeCase(field)
|
||||
}
|
||||
s.Select(fields...)
|
||||
}
|
||||
}
|
||||
|
||||
func BuildFieldSelector(fields []string) (error, func(s *sql.Selector)) {
|
||||
if len(fields) > 0 {
|
||||
return nil, func(s *sql.Selector) {
|
||||
BuildFieldSelect(s, fields)
|
||||
}
|
||||
} else {
|
||||
return nil, nil
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user