Files
ent/entc/integration/ent/group.go
Ariel Mashraki 56656dfcb6 ent/entc: configure storage driver in codegen
Summary: Pull Request resolved: https://github.com/facebookexternal/fbc/pull/1229

Reviewed By: alexsn

Differential Revision: D16539934

fbshipit-source-id: b3a8bf1f1be6f65ad3f649cd921ea20fc24182bf
2019-07-30 02:49:22 -07:00

201 lines
5.1 KiB
Go

// Code generated (@generated) by entc, DO NOT EDIT.
package ent
import (
"bytes"
"fmt"
"strconv"
"time"
"fbc/ent/dialect/gremlin"
"fbc/ent/dialect/sql"
)
// Group is the model entity for the Group schema.
type Group struct {
config
// ID of the ent.
ID string `json:"id,omitempty"`
// Active holds the value of the "active" field.
Active bool `json:"active,omitempty"`
// Expire holds the value of the "expire" field.
Expire time.Time `json:"expire,omitempty"`
// Type holds the value of the "type" field.
Type *string `json:"type,omitempty"`
// MaxUsers holds the value of the "max_users" field.
MaxUsers int `json:"max_users,omitempty"`
// Name holds the value of the "name" field.
Name string `json:"name,omitempty"`
}
// FromRows scans the sql response data into Group.
func (gr *Group) FromRows(rows *sql.Rows) error {
var vgr struct {
ID int
Active bool
Expire time.Time
Type sql.NullString
MaxUsers sql.NullInt64
Name string
}
// the order here should be the same as in the `group.Columns`.
if err := rows.Scan(
&vgr.ID,
&vgr.Active,
&vgr.Expire,
&vgr.Type,
&vgr.MaxUsers,
&vgr.Name,
); err != nil {
return err
}
gr.ID = strconv.Itoa(vgr.ID)
gr.Active = vgr.Active
gr.Expire = vgr.Expire
if vgr.Type.Valid {
gr.Type = new(string)
*gr.Type = vgr.Type.String
}
gr.MaxUsers = int(vgr.MaxUsers.Int64)
gr.Name = vgr.Name
return nil
}
// FromResponse scans the gremlin response data into Group.
func (gr *Group) FromResponse(res *gremlin.Response) error {
vmap, err := res.ReadValueMap()
if err != nil {
return err
}
var vgr struct {
ID string `json:"id,omitempty"`
Active bool `json:"active,omitempty"`
Expire int64 `json:"expire,omitempty"`
Type *string `json:"type,omitempty"`
MaxUsers int `json:"max_users,omitempty"`
Name string `json:"name,omitempty"`
}
if err := vmap.Decode(&vgr); err != nil {
return err
}
gr.ID = vgr.ID
gr.Active = vgr.Active
gr.Expire = time.Unix(vgr.Expire, 0)
gr.Type = vgr.Type
gr.MaxUsers = vgr.MaxUsers
gr.Name = vgr.Name
return nil
}
// QueryFiles queries the files edge of the Group.
func (gr *Group) QueryFiles() *FileQuery {
return (&GroupClient{gr.config}).QueryFiles(gr)
}
// QueryBlocked queries the blocked edge of the Group.
func (gr *Group) QueryBlocked() *UserQuery {
return (&GroupClient{gr.config}).QueryBlocked(gr)
}
// QueryUsers queries the users edge of the Group.
func (gr *Group) QueryUsers() *UserQuery {
return (&GroupClient{gr.config}).QueryUsers(gr)
}
// QueryInfo queries the info edge of the Group.
func (gr *Group) QueryInfo() *GroupInfoQuery {
return (&GroupClient{gr.config}).QueryInfo(gr)
}
// Update returns a builder for updating this Group.
// Note that, you need to call Group.Unwrap() before calling this method, if this Group
// was returned from a transaction, and the transaction was committed or rolled back.
func (gr *Group) Update() *GroupUpdateOne {
return (&GroupClient{gr.config}).UpdateOne(gr)
}
// 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 (gr *Group) Unwrap() *Group {
tx, ok := gr.config.driver.(*txDriver)
if !ok {
panic("ent: Group is not a transactional entity")
}
gr.config.driver = tx.drv
return gr
}
// String implements the fmt.Stringer.
func (gr *Group) String() string {
buf := bytes.NewBuffer(nil)
buf.WriteString("Group(")
buf.WriteString(fmt.Sprintf("id=%v", gr.ID))
buf.WriteString(fmt.Sprintf(", active=%v", gr.Active))
buf.WriteString(fmt.Sprintf(", expire=%v", gr.Expire))
if v := gr.Type; v != nil {
buf.WriteString(fmt.Sprintf(", type=%v", *v))
}
buf.WriteString(fmt.Sprintf(", max_users=%v", gr.MaxUsers))
buf.WriteString(fmt.Sprintf(", name=%v", gr.Name))
buf.WriteString(")")
return buf.String()
}
// id returns the int representation of the ID field.
func (gr *Group) id() int {
id, _ := strconv.Atoi(gr.ID)
return id
}
// Groups is a parsable slice of Group.
type Groups []*Group
// FromRows scans the sql response data into Groups.
func (gr *Groups) FromRows(rows *sql.Rows) error {
for rows.Next() {
vgr := &Group{}
if err := vgr.FromRows(rows); err != nil {
return err
}
*gr = append(*gr, vgr)
}
return nil
}
// FromResponse scans the gremlin response data into Groups.
func (gr *Groups) FromResponse(res *gremlin.Response) error {
vmap, err := res.ReadValueMap()
if err != nil {
return err
}
var vgr []struct {
ID string `json:"id,omitempty"`
Active bool `json:"active,omitempty"`
Expire int64 `json:"expire,omitempty"`
Type *string `json:"type,omitempty"`
MaxUsers int `json:"max_users,omitempty"`
Name string `json:"name,omitempty"`
}
if err := vmap.Decode(&vgr); err != nil {
return err
}
for _, v := range vgr {
*gr = append(*gr, &Group{
ID: v.ID,
Active: v.Active,
Expire: time.Unix(v.Expire, 0),
Type: v.Type,
MaxUsers: v.MaxUsers,
Name: v.Name,
})
}
return nil
}
func (gr Groups) config(cfg config) {
for i := range gr {
gr[i].config = cfg
}
}