mirror of
https://github.com/ent/ent.git
synced 2026-05-24 09:31:56 +03:00
131 lines
3.4 KiB
Go
131 lines
3.4 KiB
Go
// Copyright (c) Facebook, Inc. and its affiliates. 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.
|
|
|
|
// Code generated by entc, DO NOT EDIT.
|
|
|
|
package ent
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/facebookincubator/ent/dialect/sql"
|
|
"github.com/facebookincubator/ent/entc/integration/template/ent/user"
|
|
)
|
|
|
|
// User is the model entity for the User schema.
|
|
type User struct {
|
|
config `json:"-"`
|
|
// ID of the ent.
|
|
ID int `json:"id,omitempty"`
|
|
// Name holds the value of the "name" field.
|
|
Name string `json:"name,omitempty"`
|
|
}
|
|
|
|
// FromRows scans the sql response data into User.
|
|
func (u *User) FromRows(rows *sql.Rows) error {
|
|
var scanu struct {
|
|
ID int
|
|
Name sql.NullString
|
|
}
|
|
// the order here should be the same as in the `user.Columns`.
|
|
if err := rows.Scan(
|
|
&scanu.ID,
|
|
&scanu.Name,
|
|
); err != nil {
|
|
return err
|
|
}
|
|
u.ID = scanu.ID
|
|
u.Name = scanu.Name.String
|
|
return nil
|
|
}
|
|
|
|
// scanValues returns the types for scanning values from sql.Rows.
|
|
func (*User) scanValues() []interface{} {
|
|
return []interface{}{
|
|
&sql.NullInt64{},
|
|
&sql.NullString{},
|
|
}
|
|
}
|
|
|
|
// assignValues assigns the values that were returned from sql.Rows (after scanning)
|
|
// to the User fields.
|
|
func (u *User) assignValues(values ...interface{}) error {
|
|
if m, n := len(values), len(user.Columns); m != n {
|
|
return fmt.Errorf("mismatch number of scan values: %d != %d", m, n)
|
|
}
|
|
value, ok := values[0].(*sql.NullInt64)
|
|
if !ok {
|
|
return fmt.Errorf("unexpected type %T for field id", value)
|
|
}
|
|
u.ID = int(value.Int64)
|
|
values = values[1:]
|
|
if value, ok := values[0].(*sql.NullString); !ok {
|
|
return fmt.Errorf("unexpected type %T for field name", values[0])
|
|
} else if value.Valid {
|
|
u.Name = value.String
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// QueryPets queries the pets edge of the User.
|
|
func (u *User) QueryPets() *PetQuery {
|
|
return (&UserClient{u.config}).QueryPets(u)
|
|
}
|
|
|
|
// QueryFriends queries the friends edge of the User.
|
|
func (u *User) QueryFriends() *UserQuery {
|
|
return (&UserClient{u.config}).QueryFriends(u)
|
|
}
|
|
|
|
// Update returns a builder for updating this User.
|
|
// Note that, you need to call User.Unwrap() before calling this method, if this User
|
|
// was returned from a transaction, and the transaction was committed or rolled back.
|
|
func (u *User) Update() *UserUpdateOne {
|
|
return (&UserClient{u.config}).UpdateOne(u)
|
|
}
|
|
|
|
// Unwrap unwraps the entity that was returned from a transaction after it was closed,
|
|
// so that all next queries will be executed through the driver which created the transaction.
|
|
func (u *User) Unwrap() *User {
|
|
tx, ok := u.config.driver.(*txDriver)
|
|
if !ok {
|
|
panic("ent: User is not a transactional entity")
|
|
}
|
|
u.config.driver = tx.drv
|
|
return u
|
|
}
|
|
|
|
// String implements the fmt.Stringer.
|
|
func (u *User) String() string {
|
|
var builder strings.Builder
|
|
builder.WriteString("User(")
|
|
builder.WriteString(fmt.Sprintf("id=%v", u.ID))
|
|
builder.WriteString(", name=")
|
|
builder.WriteString(u.Name)
|
|
builder.WriteByte(')')
|
|
return builder.String()
|
|
}
|
|
|
|
// Users is a parsable slice of User.
|
|
type Users []*User
|
|
|
|
// FromRows scans the sql response data into Users.
|
|
func (u *Users) FromRows(rows *sql.Rows) error {
|
|
for rows.Next() {
|
|
scanu := &User{}
|
|
if err := scanu.FromRows(rows); err != nil {
|
|
return err
|
|
}
|
|
*u = append(*u, scanu)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (u Users) config(cfg config) {
|
|
for _i := range u {
|
|
u[_i].config = cfg
|
|
}
|
|
}
|