将仓储层/基础设施改为 Ent + 原生 SQL 执行路径,并移除 AutoMigrate 与 GORM 依赖。 重构内容包括: - 仓储层改用 Ent/SQL(含 usage_log/account 等复杂查询),统一错误映射 - 基础设施与 setup 初始化切换为 Ent + SQL migrations - 集成测试与 fixtures 迁移到 Ent 事务模型 - 清理遗留 GORM 模型/依赖,补充迁移与文档说明 - 增加根目录 Makefile 便于前后端编译 测试: - go test -tags unit ./... - go test -tags integration ./...
1537 lines
46 KiB
Go
1537 lines
46 KiB
Go
// Code generated by ent, DO NOT EDIT.
|
|
|
|
package ent
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"time"
|
|
|
|
"entgo.io/ent/dialect/sql"
|
|
"entgo.io/ent/dialect/sql/sqlgraph"
|
|
"entgo.io/ent/schema/field"
|
|
"github.com/Wei-Shaw/sub2api/ent/apikey"
|
|
"github.com/Wei-Shaw/sub2api/ent/group"
|
|
"github.com/Wei-Shaw/sub2api/ent/predicate"
|
|
"github.com/Wei-Shaw/sub2api/ent/redeemcode"
|
|
"github.com/Wei-Shaw/sub2api/ent/user"
|
|
"github.com/Wei-Shaw/sub2api/ent/usersubscription"
|
|
)
|
|
|
|
// UserUpdate is the builder for updating User entities.
|
|
type UserUpdate struct {
|
|
config
|
|
hooks []Hook
|
|
mutation *UserMutation
|
|
}
|
|
|
|
// Where appends a list predicates to the UserUpdate builder.
|
|
func (_u *UserUpdate) Where(ps ...predicate.User) *UserUpdate {
|
|
_u.mutation.Where(ps...)
|
|
return _u
|
|
}
|
|
|
|
// SetUpdatedAt sets the "updated_at" field.
|
|
func (_u *UserUpdate) SetUpdatedAt(v time.Time) *UserUpdate {
|
|
_u.mutation.SetUpdatedAt(v)
|
|
return _u
|
|
}
|
|
|
|
// SetDeletedAt sets the "deleted_at" field.
|
|
func (_u *UserUpdate) SetDeletedAt(v time.Time) *UserUpdate {
|
|
_u.mutation.SetDeletedAt(v)
|
|
return _u
|
|
}
|
|
|
|
// SetNillableDeletedAt sets the "deleted_at" field if the given value is not nil.
|
|
func (_u *UserUpdate) SetNillableDeletedAt(v *time.Time) *UserUpdate {
|
|
if v != nil {
|
|
_u.SetDeletedAt(*v)
|
|
}
|
|
return _u
|
|
}
|
|
|
|
// ClearDeletedAt clears the value of the "deleted_at" field.
|
|
func (_u *UserUpdate) ClearDeletedAt() *UserUpdate {
|
|
_u.mutation.ClearDeletedAt()
|
|
return _u
|
|
}
|
|
|
|
// SetEmail sets the "email" field.
|
|
func (_u *UserUpdate) SetEmail(v string) *UserUpdate {
|
|
_u.mutation.SetEmail(v)
|
|
return _u
|
|
}
|
|
|
|
// SetNillableEmail sets the "email" field if the given value is not nil.
|
|
func (_u *UserUpdate) SetNillableEmail(v *string) *UserUpdate {
|
|
if v != nil {
|
|
_u.SetEmail(*v)
|
|
}
|
|
return _u
|
|
}
|
|
|
|
// SetPasswordHash sets the "password_hash" field.
|
|
func (_u *UserUpdate) SetPasswordHash(v string) *UserUpdate {
|
|
_u.mutation.SetPasswordHash(v)
|
|
return _u
|
|
}
|
|
|
|
// SetNillablePasswordHash sets the "password_hash" field if the given value is not nil.
|
|
func (_u *UserUpdate) SetNillablePasswordHash(v *string) *UserUpdate {
|
|
if v != nil {
|
|
_u.SetPasswordHash(*v)
|
|
}
|
|
return _u
|
|
}
|
|
|
|
// SetRole sets the "role" field.
|
|
func (_u *UserUpdate) SetRole(v string) *UserUpdate {
|
|
_u.mutation.SetRole(v)
|
|
return _u
|
|
}
|
|
|
|
// SetNillableRole sets the "role" field if the given value is not nil.
|
|
func (_u *UserUpdate) SetNillableRole(v *string) *UserUpdate {
|
|
if v != nil {
|
|
_u.SetRole(*v)
|
|
}
|
|
return _u
|
|
}
|
|
|
|
// SetBalance sets the "balance" field.
|
|
func (_u *UserUpdate) SetBalance(v float64) *UserUpdate {
|
|
_u.mutation.ResetBalance()
|
|
_u.mutation.SetBalance(v)
|
|
return _u
|
|
}
|
|
|
|
// SetNillableBalance sets the "balance" field if the given value is not nil.
|
|
func (_u *UserUpdate) SetNillableBalance(v *float64) *UserUpdate {
|
|
if v != nil {
|
|
_u.SetBalance(*v)
|
|
}
|
|
return _u
|
|
}
|
|
|
|
// AddBalance adds value to the "balance" field.
|
|
func (_u *UserUpdate) AddBalance(v float64) *UserUpdate {
|
|
_u.mutation.AddBalance(v)
|
|
return _u
|
|
}
|
|
|
|
// SetConcurrency sets the "concurrency" field.
|
|
func (_u *UserUpdate) SetConcurrency(v int) *UserUpdate {
|
|
_u.mutation.ResetConcurrency()
|
|
_u.mutation.SetConcurrency(v)
|
|
return _u
|
|
}
|
|
|
|
// SetNillableConcurrency sets the "concurrency" field if the given value is not nil.
|
|
func (_u *UserUpdate) SetNillableConcurrency(v *int) *UserUpdate {
|
|
if v != nil {
|
|
_u.SetConcurrency(*v)
|
|
}
|
|
return _u
|
|
}
|
|
|
|
// AddConcurrency adds value to the "concurrency" field.
|
|
func (_u *UserUpdate) AddConcurrency(v int) *UserUpdate {
|
|
_u.mutation.AddConcurrency(v)
|
|
return _u
|
|
}
|
|
|
|
// SetStatus sets the "status" field.
|
|
func (_u *UserUpdate) SetStatus(v string) *UserUpdate {
|
|
_u.mutation.SetStatus(v)
|
|
return _u
|
|
}
|
|
|
|
// SetNillableStatus sets the "status" field if the given value is not nil.
|
|
func (_u *UserUpdate) SetNillableStatus(v *string) *UserUpdate {
|
|
if v != nil {
|
|
_u.SetStatus(*v)
|
|
}
|
|
return _u
|
|
}
|
|
|
|
// SetUsername sets the "username" field.
|
|
func (_u *UserUpdate) SetUsername(v string) *UserUpdate {
|
|
_u.mutation.SetUsername(v)
|
|
return _u
|
|
}
|
|
|
|
// SetNillableUsername sets the "username" field if the given value is not nil.
|
|
func (_u *UserUpdate) SetNillableUsername(v *string) *UserUpdate {
|
|
if v != nil {
|
|
_u.SetUsername(*v)
|
|
}
|
|
return _u
|
|
}
|
|
|
|
// SetWechat sets the "wechat" field.
|
|
func (_u *UserUpdate) SetWechat(v string) *UserUpdate {
|
|
_u.mutation.SetWechat(v)
|
|
return _u
|
|
}
|
|
|
|
// SetNillableWechat sets the "wechat" field if the given value is not nil.
|
|
func (_u *UserUpdate) SetNillableWechat(v *string) *UserUpdate {
|
|
if v != nil {
|
|
_u.SetWechat(*v)
|
|
}
|
|
return _u
|
|
}
|
|
|
|
// SetNotes sets the "notes" field.
|
|
func (_u *UserUpdate) SetNotes(v string) *UserUpdate {
|
|
_u.mutation.SetNotes(v)
|
|
return _u
|
|
}
|
|
|
|
// SetNillableNotes sets the "notes" field if the given value is not nil.
|
|
func (_u *UserUpdate) SetNillableNotes(v *string) *UserUpdate {
|
|
if v != nil {
|
|
_u.SetNotes(*v)
|
|
}
|
|
return _u
|
|
}
|
|
|
|
// AddAPIKeyIDs adds the "api_keys" edge to the ApiKey entity by IDs.
|
|
func (_u *UserUpdate) AddAPIKeyIDs(ids ...int64) *UserUpdate {
|
|
_u.mutation.AddAPIKeyIDs(ids...)
|
|
return _u
|
|
}
|
|
|
|
// AddAPIKeys adds the "api_keys" edges to the ApiKey entity.
|
|
func (_u *UserUpdate) AddAPIKeys(v ...*ApiKey) *UserUpdate {
|
|
ids := make([]int64, len(v))
|
|
for i := range v {
|
|
ids[i] = v[i].ID
|
|
}
|
|
return _u.AddAPIKeyIDs(ids...)
|
|
}
|
|
|
|
// AddRedeemCodeIDs adds the "redeem_codes" edge to the RedeemCode entity by IDs.
|
|
func (_u *UserUpdate) AddRedeemCodeIDs(ids ...int64) *UserUpdate {
|
|
_u.mutation.AddRedeemCodeIDs(ids...)
|
|
return _u
|
|
}
|
|
|
|
// AddRedeemCodes adds the "redeem_codes" edges to the RedeemCode entity.
|
|
func (_u *UserUpdate) AddRedeemCodes(v ...*RedeemCode) *UserUpdate {
|
|
ids := make([]int64, len(v))
|
|
for i := range v {
|
|
ids[i] = v[i].ID
|
|
}
|
|
return _u.AddRedeemCodeIDs(ids...)
|
|
}
|
|
|
|
// AddSubscriptionIDs adds the "subscriptions" edge to the UserSubscription entity by IDs.
|
|
func (_u *UserUpdate) AddSubscriptionIDs(ids ...int64) *UserUpdate {
|
|
_u.mutation.AddSubscriptionIDs(ids...)
|
|
return _u
|
|
}
|
|
|
|
// AddSubscriptions adds the "subscriptions" edges to the UserSubscription entity.
|
|
func (_u *UserUpdate) AddSubscriptions(v ...*UserSubscription) *UserUpdate {
|
|
ids := make([]int64, len(v))
|
|
for i := range v {
|
|
ids[i] = v[i].ID
|
|
}
|
|
return _u.AddSubscriptionIDs(ids...)
|
|
}
|
|
|
|
// AddAssignedSubscriptionIDs adds the "assigned_subscriptions" edge to the UserSubscription entity by IDs.
|
|
func (_u *UserUpdate) AddAssignedSubscriptionIDs(ids ...int64) *UserUpdate {
|
|
_u.mutation.AddAssignedSubscriptionIDs(ids...)
|
|
return _u
|
|
}
|
|
|
|
// AddAssignedSubscriptions adds the "assigned_subscriptions" edges to the UserSubscription entity.
|
|
func (_u *UserUpdate) AddAssignedSubscriptions(v ...*UserSubscription) *UserUpdate {
|
|
ids := make([]int64, len(v))
|
|
for i := range v {
|
|
ids[i] = v[i].ID
|
|
}
|
|
return _u.AddAssignedSubscriptionIDs(ids...)
|
|
}
|
|
|
|
// AddAllowedGroupIDs adds the "allowed_groups" edge to the Group entity by IDs.
|
|
func (_u *UserUpdate) AddAllowedGroupIDs(ids ...int64) *UserUpdate {
|
|
_u.mutation.AddAllowedGroupIDs(ids...)
|
|
return _u
|
|
}
|
|
|
|
// AddAllowedGroups adds the "allowed_groups" edges to the Group entity.
|
|
func (_u *UserUpdate) AddAllowedGroups(v ...*Group) *UserUpdate {
|
|
ids := make([]int64, len(v))
|
|
for i := range v {
|
|
ids[i] = v[i].ID
|
|
}
|
|
return _u.AddAllowedGroupIDs(ids...)
|
|
}
|
|
|
|
// Mutation returns the UserMutation object of the builder.
|
|
func (_u *UserUpdate) Mutation() *UserMutation {
|
|
return _u.mutation
|
|
}
|
|
|
|
// ClearAPIKeys clears all "api_keys" edges to the ApiKey entity.
|
|
func (_u *UserUpdate) ClearAPIKeys() *UserUpdate {
|
|
_u.mutation.ClearAPIKeys()
|
|
return _u
|
|
}
|
|
|
|
// RemoveAPIKeyIDs removes the "api_keys" edge to ApiKey entities by IDs.
|
|
func (_u *UserUpdate) RemoveAPIKeyIDs(ids ...int64) *UserUpdate {
|
|
_u.mutation.RemoveAPIKeyIDs(ids...)
|
|
return _u
|
|
}
|
|
|
|
// RemoveAPIKeys removes "api_keys" edges to ApiKey entities.
|
|
func (_u *UserUpdate) RemoveAPIKeys(v ...*ApiKey) *UserUpdate {
|
|
ids := make([]int64, len(v))
|
|
for i := range v {
|
|
ids[i] = v[i].ID
|
|
}
|
|
return _u.RemoveAPIKeyIDs(ids...)
|
|
}
|
|
|
|
// ClearRedeemCodes clears all "redeem_codes" edges to the RedeemCode entity.
|
|
func (_u *UserUpdate) ClearRedeemCodes() *UserUpdate {
|
|
_u.mutation.ClearRedeemCodes()
|
|
return _u
|
|
}
|
|
|
|
// RemoveRedeemCodeIDs removes the "redeem_codes" edge to RedeemCode entities by IDs.
|
|
func (_u *UserUpdate) RemoveRedeemCodeIDs(ids ...int64) *UserUpdate {
|
|
_u.mutation.RemoveRedeemCodeIDs(ids...)
|
|
return _u
|
|
}
|
|
|
|
// RemoveRedeemCodes removes "redeem_codes" edges to RedeemCode entities.
|
|
func (_u *UserUpdate) RemoveRedeemCodes(v ...*RedeemCode) *UserUpdate {
|
|
ids := make([]int64, len(v))
|
|
for i := range v {
|
|
ids[i] = v[i].ID
|
|
}
|
|
return _u.RemoveRedeemCodeIDs(ids...)
|
|
}
|
|
|
|
// ClearSubscriptions clears all "subscriptions" edges to the UserSubscription entity.
|
|
func (_u *UserUpdate) ClearSubscriptions() *UserUpdate {
|
|
_u.mutation.ClearSubscriptions()
|
|
return _u
|
|
}
|
|
|
|
// RemoveSubscriptionIDs removes the "subscriptions" edge to UserSubscription entities by IDs.
|
|
func (_u *UserUpdate) RemoveSubscriptionIDs(ids ...int64) *UserUpdate {
|
|
_u.mutation.RemoveSubscriptionIDs(ids...)
|
|
return _u
|
|
}
|
|
|
|
// RemoveSubscriptions removes "subscriptions" edges to UserSubscription entities.
|
|
func (_u *UserUpdate) RemoveSubscriptions(v ...*UserSubscription) *UserUpdate {
|
|
ids := make([]int64, len(v))
|
|
for i := range v {
|
|
ids[i] = v[i].ID
|
|
}
|
|
return _u.RemoveSubscriptionIDs(ids...)
|
|
}
|
|
|
|
// ClearAssignedSubscriptions clears all "assigned_subscriptions" edges to the UserSubscription entity.
|
|
func (_u *UserUpdate) ClearAssignedSubscriptions() *UserUpdate {
|
|
_u.mutation.ClearAssignedSubscriptions()
|
|
return _u
|
|
}
|
|
|
|
// RemoveAssignedSubscriptionIDs removes the "assigned_subscriptions" edge to UserSubscription entities by IDs.
|
|
func (_u *UserUpdate) RemoveAssignedSubscriptionIDs(ids ...int64) *UserUpdate {
|
|
_u.mutation.RemoveAssignedSubscriptionIDs(ids...)
|
|
return _u
|
|
}
|
|
|
|
// RemoveAssignedSubscriptions removes "assigned_subscriptions" edges to UserSubscription entities.
|
|
func (_u *UserUpdate) RemoveAssignedSubscriptions(v ...*UserSubscription) *UserUpdate {
|
|
ids := make([]int64, len(v))
|
|
for i := range v {
|
|
ids[i] = v[i].ID
|
|
}
|
|
return _u.RemoveAssignedSubscriptionIDs(ids...)
|
|
}
|
|
|
|
// ClearAllowedGroups clears all "allowed_groups" edges to the Group entity.
|
|
func (_u *UserUpdate) ClearAllowedGroups() *UserUpdate {
|
|
_u.mutation.ClearAllowedGroups()
|
|
return _u
|
|
}
|
|
|
|
// RemoveAllowedGroupIDs removes the "allowed_groups" edge to Group entities by IDs.
|
|
func (_u *UserUpdate) RemoveAllowedGroupIDs(ids ...int64) *UserUpdate {
|
|
_u.mutation.RemoveAllowedGroupIDs(ids...)
|
|
return _u
|
|
}
|
|
|
|
// RemoveAllowedGroups removes "allowed_groups" edges to Group entities.
|
|
func (_u *UserUpdate) RemoveAllowedGroups(v ...*Group) *UserUpdate {
|
|
ids := make([]int64, len(v))
|
|
for i := range v {
|
|
ids[i] = v[i].ID
|
|
}
|
|
return _u.RemoveAllowedGroupIDs(ids...)
|
|
}
|
|
|
|
// Save executes the query and returns the number of nodes affected by the update operation.
|
|
func (_u *UserUpdate) Save(ctx context.Context) (int, error) {
|
|
if err := _u.defaults(); err != nil {
|
|
return 0, err
|
|
}
|
|
return withHooks(ctx, _u.sqlSave, _u.mutation, _u.hooks)
|
|
}
|
|
|
|
// SaveX is like Save, but panics if an error occurs.
|
|
func (_u *UserUpdate) SaveX(ctx context.Context) int {
|
|
affected, err := _u.Save(ctx)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return affected
|
|
}
|
|
|
|
// Exec executes the query.
|
|
func (_u *UserUpdate) Exec(ctx context.Context) error {
|
|
_, err := _u.Save(ctx)
|
|
return err
|
|
}
|
|
|
|
// ExecX is like Exec, but panics if an error occurs.
|
|
func (_u *UserUpdate) ExecX(ctx context.Context) {
|
|
if err := _u.Exec(ctx); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
// defaults sets the default values of the builder before save.
|
|
func (_u *UserUpdate) defaults() error {
|
|
if _, ok := _u.mutation.UpdatedAt(); !ok {
|
|
if user.UpdateDefaultUpdatedAt == nil {
|
|
return fmt.Errorf("ent: uninitialized user.UpdateDefaultUpdatedAt (forgotten import ent/runtime?)")
|
|
}
|
|
v := user.UpdateDefaultUpdatedAt()
|
|
_u.mutation.SetUpdatedAt(v)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// check runs all checks and user-defined validators on the builder.
|
|
func (_u *UserUpdate) check() error {
|
|
if v, ok := _u.mutation.Email(); ok {
|
|
if err := user.EmailValidator(v); err != nil {
|
|
return &ValidationError{Name: "email", err: fmt.Errorf(`ent: validator failed for field "User.email": %w`, err)}
|
|
}
|
|
}
|
|
if v, ok := _u.mutation.PasswordHash(); ok {
|
|
if err := user.PasswordHashValidator(v); err != nil {
|
|
return &ValidationError{Name: "password_hash", err: fmt.Errorf(`ent: validator failed for field "User.password_hash": %w`, err)}
|
|
}
|
|
}
|
|
if v, ok := _u.mutation.Role(); ok {
|
|
if err := user.RoleValidator(v); err != nil {
|
|
return &ValidationError{Name: "role", err: fmt.Errorf(`ent: validator failed for field "User.role": %w`, err)}
|
|
}
|
|
}
|
|
if v, ok := _u.mutation.Status(); ok {
|
|
if err := user.StatusValidator(v); err != nil {
|
|
return &ValidationError{Name: "status", err: fmt.Errorf(`ent: validator failed for field "User.status": %w`, err)}
|
|
}
|
|
}
|
|
if v, ok := _u.mutation.Username(); ok {
|
|
if err := user.UsernameValidator(v); err != nil {
|
|
return &ValidationError{Name: "username", err: fmt.Errorf(`ent: validator failed for field "User.username": %w`, err)}
|
|
}
|
|
}
|
|
if v, ok := _u.mutation.Wechat(); ok {
|
|
if err := user.WechatValidator(v); err != nil {
|
|
return &ValidationError{Name: "wechat", err: fmt.Errorf(`ent: validator failed for field "User.wechat": %w`, err)}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (_u *UserUpdate) sqlSave(ctx context.Context) (_node int, err error) {
|
|
if err := _u.check(); err != nil {
|
|
return _node, err
|
|
}
|
|
_spec := sqlgraph.NewUpdateSpec(user.Table, user.Columns, sqlgraph.NewFieldSpec(user.FieldID, field.TypeInt64))
|
|
if ps := _u.mutation.predicates; len(ps) > 0 {
|
|
_spec.Predicate = func(selector *sql.Selector) {
|
|
for i := range ps {
|
|
ps[i](selector)
|
|
}
|
|
}
|
|
}
|
|
if value, ok := _u.mutation.UpdatedAt(); ok {
|
|
_spec.SetField(user.FieldUpdatedAt, field.TypeTime, value)
|
|
}
|
|
if value, ok := _u.mutation.DeletedAt(); ok {
|
|
_spec.SetField(user.FieldDeletedAt, field.TypeTime, value)
|
|
}
|
|
if _u.mutation.DeletedAtCleared() {
|
|
_spec.ClearField(user.FieldDeletedAt, field.TypeTime)
|
|
}
|
|
if value, ok := _u.mutation.Email(); ok {
|
|
_spec.SetField(user.FieldEmail, field.TypeString, value)
|
|
}
|
|
if value, ok := _u.mutation.PasswordHash(); ok {
|
|
_spec.SetField(user.FieldPasswordHash, field.TypeString, value)
|
|
}
|
|
if value, ok := _u.mutation.Role(); ok {
|
|
_spec.SetField(user.FieldRole, field.TypeString, value)
|
|
}
|
|
if value, ok := _u.mutation.Balance(); ok {
|
|
_spec.SetField(user.FieldBalance, field.TypeFloat64, value)
|
|
}
|
|
if value, ok := _u.mutation.AddedBalance(); ok {
|
|
_spec.AddField(user.FieldBalance, field.TypeFloat64, value)
|
|
}
|
|
if value, ok := _u.mutation.Concurrency(); ok {
|
|
_spec.SetField(user.FieldConcurrency, field.TypeInt, value)
|
|
}
|
|
if value, ok := _u.mutation.AddedConcurrency(); ok {
|
|
_spec.AddField(user.FieldConcurrency, field.TypeInt, value)
|
|
}
|
|
if value, ok := _u.mutation.Status(); ok {
|
|
_spec.SetField(user.FieldStatus, field.TypeString, value)
|
|
}
|
|
if value, ok := _u.mutation.Username(); ok {
|
|
_spec.SetField(user.FieldUsername, field.TypeString, value)
|
|
}
|
|
if value, ok := _u.mutation.Wechat(); ok {
|
|
_spec.SetField(user.FieldWechat, field.TypeString, value)
|
|
}
|
|
if value, ok := _u.mutation.Notes(); ok {
|
|
_spec.SetField(user.FieldNotes, field.TypeString, value)
|
|
}
|
|
if _u.mutation.APIKeysCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.APIKeysTable,
|
|
Columns: []string{user.APIKeysColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(apikey.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := _u.mutation.RemovedAPIKeysIDs(); len(nodes) > 0 && !_u.mutation.APIKeysCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.APIKeysTable,
|
|
Columns: []string{user.APIKeysColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(apikey.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := _u.mutation.APIKeysIDs(); len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.APIKeysTable,
|
|
Columns: []string{user.APIKeysColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(apikey.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
_spec.Edges.Add = append(_spec.Edges.Add, edge)
|
|
}
|
|
if _u.mutation.RedeemCodesCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.RedeemCodesTable,
|
|
Columns: []string{user.RedeemCodesColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(redeemcode.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := _u.mutation.RemovedRedeemCodesIDs(); len(nodes) > 0 && !_u.mutation.RedeemCodesCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.RedeemCodesTable,
|
|
Columns: []string{user.RedeemCodesColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(redeemcode.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := _u.mutation.RedeemCodesIDs(); len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.RedeemCodesTable,
|
|
Columns: []string{user.RedeemCodesColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(redeemcode.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
_spec.Edges.Add = append(_spec.Edges.Add, edge)
|
|
}
|
|
if _u.mutation.SubscriptionsCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.SubscriptionsTable,
|
|
Columns: []string{user.SubscriptionsColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(usersubscription.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := _u.mutation.RemovedSubscriptionsIDs(); len(nodes) > 0 && !_u.mutation.SubscriptionsCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.SubscriptionsTable,
|
|
Columns: []string{user.SubscriptionsColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(usersubscription.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := _u.mutation.SubscriptionsIDs(); len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.SubscriptionsTable,
|
|
Columns: []string{user.SubscriptionsColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(usersubscription.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
_spec.Edges.Add = append(_spec.Edges.Add, edge)
|
|
}
|
|
if _u.mutation.AssignedSubscriptionsCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.AssignedSubscriptionsTable,
|
|
Columns: []string{user.AssignedSubscriptionsColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(usersubscription.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := _u.mutation.RemovedAssignedSubscriptionsIDs(); len(nodes) > 0 && !_u.mutation.AssignedSubscriptionsCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.AssignedSubscriptionsTable,
|
|
Columns: []string{user.AssignedSubscriptionsColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(usersubscription.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := _u.mutation.AssignedSubscriptionsIDs(); len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.AssignedSubscriptionsTable,
|
|
Columns: []string{user.AssignedSubscriptionsColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(usersubscription.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
_spec.Edges.Add = append(_spec.Edges.Add, edge)
|
|
}
|
|
if _u.mutation.AllowedGroupsCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.M2M,
|
|
Inverse: false,
|
|
Table: user.AllowedGroupsTable,
|
|
Columns: user.AllowedGroupsPrimaryKey,
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(group.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
createE := &UserAllowedGroupCreate{config: _u.config, mutation: newUserAllowedGroupMutation(_u.config, OpCreate)}
|
|
createE.defaults()
|
|
_, specE := createE.createSpec()
|
|
edge.Target.Fields = specE.Fields
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := _u.mutation.RemovedAllowedGroupsIDs(); len(nodes) > 0 && !_u.mutation.AllowedGroupsCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.M2M,
|
|
Inverse: false,
|
|
Table: user.AllowedGroupsTable,
|
|
Columns: user.AllowedGroupsPrimaryKey,
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(group.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
createE := &UserAllowedGroupCreate{config: _u.config, mutation: newUserAllowedGroupMutation(_u.config, OpCreate)}
|
|
createE.defaults()
|
|
_, specE := createE.createSpec()
|
|
edge.Target.Fields = specE.Fields
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := _u.mutation.AllowedGroupsIDs(); len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.M2M,
|
|
Inverse: false,
|
|
Table: user.AllowedGroupsTable,
|
|
Columns: user.AllowedGroupsPrimaryKey,
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(group.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
createE := &UserAllowedGroupCreate{config: _u.config, mutation: newUserAllowedGroupMutation(_u.config, OpCreate)}
|
|
createE.defaults()
|
|
_, specE := createE.createSpec()
|
|
edge.Target.Fields = specE.Fields
|
|
_spec.Edges.Add = append(_spec.Edges.Add, edge)
|
|
}
|
|
if _node, err = sqlgraph.UpdateNodes(ctx, _u.driver, _spec); err != nil {
|
|
if _, ok := err.(*sqlgraph.NotFoundError); ok {
|
|
err = &NotFoundError{user.Label}
|
|
} else if sqlgraph.IsConstraintError(err) {
|
|
err = &ConstraintError{msg: err.Error(), wrap: err}
|
|
}
|
|
return 0, err
|
|
}
|
|
_u.mutation.done = true
|
|
return _node, nil
|
|
}
|
|
|
|
// UserUpdateOne is the builder for updating a single User entity.
|
|
type UserUpdateOne struct {
|
|
config
|
|
fields []string
|
|
hooks []Hook
|
|
mutation *UserMutation
|
|
}
|
|
|
|
// SetUpdatedAt sets the "updated_at" field.
|
|
func (_u *UserUpdateOne) SetUpdatedAt(v time.Time) *UserUpdateOne {
|
|
_u.mutation.SetUpdatedAt(v)
|
|
return _u
|
|
}
|
|
|
|
// SetDeletedAt sets the "deleted_at" field.
|
|
func (_u *UserUpdateOne) SetDeletedAt(v time.Time) *UserUpdateOne {
|
|
_u.mutation.SetDeletedAt(v)
|
|
return _u
|
|
}
|
|
|
|
// SetNillableDeletedAt sets the "deleted_at" field if the given value is not nil.
|
|
func (_u *UserUpdateOne) SetNillableDeletedAt(v *time.Time) *UserUpdateOne {
|
|
if v != nil {
|
|
_u.SetDeletedAt(*v)
|
|
}
|
|
return _u
|
|
}
|
|
|
|
// ClearDeletedAt clears the value of the "deleted_at" field.
|
|
func (_u *UserUpdateOne) ClearDeletedAt() *UserUpdateOne {
|
|
_u.mutation.ClearDeletedAt()
|
|
return _u
|
|
}
|
|
|
|
// SetEmail sets the "email" field.
|
|
func (_u *UserUpdateOne) SetEmail(v string) *UserUpdateOne {
|
|
_u.mutation.SetEmail(v)
|
|
return _u
|
|
}
|
|
|
|
// SetNillableEmail sets the "email" field if the given value is not nil.
|
|
func (_u *UserUpdateOne) SetNillableEmail(v *string) *UserUpdateOne {
|
|
if v != nil {
|
|
_u.SetEmail(*v)
|
|
}
|
|
return _u
|
|
}
|
|
|
|
// SetPasswordHash sets the "password_hash" field.
|
|
func (_u *UserUpdateOne) SetPasswordHash(v string) *UserUpdateOne {
|
|
_u.mutation.SetPasswordHash(v)
|
|
return _u
|
|
}
|
|
|
|
// SetNillablePasswordHash sets the "password_hash" field if the given value is not nil.
|
|
func (_u *UserUpdateOne) SetNillablePasswordHash(v *string) *UserUpdateOne {
|
|
if v != nil {
|
|
_u.SetPasswordHash(*v)
|
|
}
|
|
return _u
|
|
}
|
|
|
|
// SetRole sets the "role" field.
|
|
func (_u *UserUpdateOne) SetRole(v string) *UserUpdateOne {
|
|
_u.mutation.SetRole(v)
|
|
return _u
|
|
}
|
|
|
|
// SetNillableRole sets the "role" field if the given value is not nil.
|
|
func (_u *UserUpdateOne) SetNillableRole(v *string) *UserUpdateOne {
|
|
if v != nil {
|
|
_u.SetRole(*v)
|
|
}
|
|
return _u
|
|
}
|
|
|
|
// SetBalance sets the "balance" field.
|
|
func (_u *UserUpdateOne) SetBalance(v float64) *UserUpdateOne {
|
|
_u.mutation.ResetBalance()
|
|
_u.mutation.SetBalance(v)
|
|
return _u
|
|
}
|
|
|
|
// SetNillableBalance sets the "balance" field if the given value is not nil.
|
|
func (_u *UserUpdateOne) SetNillableBalance(v *float64) *UserUpdateOne {
|
|
if v != nil {
|
|
_u.SetBalance(*v)
|
|
}
|
|
return _u
|
|
}
|
|
|
|
// AddBalance adds value to the "balance" field.
|
|
func (_u *UserUpdateOne) AddBalance(v float64) *UserUpdateOne {
|
|
_u.mutation.AddBalance(v)
|
|
return _u
|
|
}
|
|
|
|
// SetConcurrency sets the "concurrency" field.
|
|
func (_u *UserUpdateOne) SetConcurrency(v int) *UserUpdateOne {
|
|
_u.mutation.ResetConcurrency()
|
|
_u.mutation.SetConcurrency(v)
|
|
return _u
|
|
}
|
|
|
|
// SetNillableConcurrency sets the "concurrency" field if the given value is not nil.
|
|
func (_u *UserUpdateOne) SetNillableConcurrency(v *int) *UserUpdateOne {
|
|
if v != nil {
|
|
_u.SetConcurrency(*v)
|
|
}
|
|
return _u
|
|
}
|
|
|
|
// AddConcurrency adds value to the "concurrency" field.
|
|
func (_u *UserUpdateOne) AddConcurrency(v int) *UserUpdateOne {
|
|
_u.mutation.AddConcurrency(v)
|
|
return _u
|
|
}
|
|
|
|
// SetStatus sets the "status" field.
|
|
func (_u *UserUpdateOne) SetStatus(v string) *UserUpdateOne {
|
|
_u.mutation.SetStatus(v)
|
|
return _u
|
|
}
|
|
|
|
// SetNillableStatus sets the "status" field if the given value is not nil.
|
|
func (_u *UserUpdateOne) SetNillableStatus(v *string) *UserUpdateOne {
|
|
if v != nil {
|
|
_u.SetStatus(*v)
|
|
}
|
|
return _u
|
|
}
|
|
|
|
// SetUsername sets the "username" field.
|
|
func (_u *UserUpdateOne) SetUsername(v string) *UserUpdateOne {
|
|
_u.mutation.SetUsername(v)
|
|
return _u
|
|
}
|
|
|
|
// SetNillableUsername sets the "username" field if the given value is not nil.
|
|
func (_u *UserUpdateOne) SetNillableUsername(v *string) *UserUpdateOne {
|
|
if v != nil {
|
|
_u.SetUsername(*v)
|
|
}
|
|
return _u
|
|
}
|
|
|
|
// SetWechat sets the "wechat" field.
|
|
func (_u *UserUpdateOne) SetWechat(v string) *UserUpdateOne {
|
|
_u.mutation.SetWechat(v)
|
|
return _u
|
|
}
|
|
|
|
// SetNillableWechat sets the "wechat" field if the given value is not nil.
|
|
func (_u *UserUpdateOne) SetNillableWechat(v *string) *UserUpdateOne {
|
|
if v != nil {
|
|
_u.SetWechat(*v)
|
|
}
|
|
return _u
|
|
}
|
|
|
|
// SetNotes sets the "notes" field.
|
|
func (_u *UserUpdateOne) SetNotes(v string) *UserUpdateOne {
|
|
_u.mutation.SetNotes(v)
|
|
return _u
|
|
}
|
|
|
|
// SetNillableNotes sets the "notes" field if the given value is not nil.
|
|
func (_u *UserUpdateOne) SetNillableNotes(v *string) *UserUpdateOne {
|
|
if v != nil {
|
|
_u.SetNotes(*v)
|
|
}
|
|
return _u
|
|
}
|
|
|
|
// AddAPIKeyIDs adds the "api_keys" edge to the ApiKey entity by IDs.
|
|
func (_u *UserUpdateOne) AddAPIKeyIDs(ids ...int64) *UserUpdateOne {
|
|
_u.mutation.AddAPIKeyIDs(ids...)
|
|
return _u
|
|
}
|
|
|
|
// AddAPIKeys adds the "api_keys" edges to the ApiKey entity.
|
|
func (_u *UserUpdateOne) AddAPIKeys(v ...*ApiKey) *UserUpdateOne {
|
|
ids := make([]int64, len(v))
|
|
for i := range v {
|
|
ids[i] = v[i].ID
|
|
}
|
|
return _u.AddAPIKeyIDs(ids...)
|
|
}
|
|
|
|
// AddRedeemCodeIDs adds the "redeem_codes" edge to the RedeemCode entity by IDs.
|
|
func (_u *UserUpdateOne) AddRedeemCodeIDs(ids ...int64) *UserUpdateOne {
|
|
_u.mutation.AddRedeemCodeIDs(ids...)
|
|
return _u
|
|
}
|
|
|
|
// AddRedeemCodes adds the "redeem_codes" edges to the RedeemCode entity.
|
|
func (_u *UserUpdateOne) AddRedeemCodes(v ...*RedeemCode) *UserUpdateOne {
|
|
ids := make([]int64, len(v))
|
|
for i := range v {
|
|
ids[i] = v[i].ID
|
|
}
|
|
return _u.AddRedeemCodeIDs(ids...)
|
|
}
|
|
|
|
// AddSubscriptionIDs adds the "subscriptions" edge to the UserSubscription entity by IDs.
|
|
func (_u *UserUpdateOne) AddSubscriptionIDs(ids ...int64) *UserUpdateOne {
|
|
_u.mutation.AddSubscriptionIDs(ids...)
|
|
return _u
|
|
}
|
|
|
|
// AddSubscriptions adds the "subscriptions" edges to the UserSubscription entity.
|
|
func (_u *UserUpdateOne) AddSubscriptions(v ...*UserSubscription) *UserUpdateOne {
|
|
ids := make([]int64, len(v))
|
|
for i := range v {
|
|
ids[i] = v[i].ID
|
|
}
|
|
return _u.AddSubscriptionIDs(ids...)
|
|
}
|
|
|
|
// AddAssignedSubscriptionIDs adds the "assigned_subscriptions" edge to the UserSubscription entity by IDs.
|
|
func (_u *UserUpdateOne) AddAssignedSubscriptionIDs(ids ...int64) *UserUpdateOne {
|
|
_u.mutation.AddAssignedSubscriptionIDs(ids...)
|
|
return _u
|
|
}
|
|
|
|
// AddAssignedSubscriptions adds the "assigned_subscriptions" edges to the UserSubscription entity.
|
|
func (_u *UserUpdateOne) AddAssignedSubscriptions(v ...*UserSubscription) *UserUpdateOne {
|
|
ids := make([]int64, len(v))
|
|
for i := range v {
|
|
ids[i] = v[i].ID
|
|
}
|
|
return _u.AddAssignedSubscriptionIDs(ids...)
|
|
}
|
|
|
|
// AddAllowedGroupIDs adds the "allowed_groups" edge to the Group entity by IDs.
|
|
func (_u *UserUpdateOne) AddAllowedGroupIDs(ids ...int64) *UserUpdateOne {
|
|
_u.mutation.AddAllowedGroupIDs(ids...)
|
|
return _u
|
|
}
|
|
|
|
// AddAllowedGroups adds the "allowed_groups" edges to the Group entity.
|
|
func (_u *UserUpdateOne) AddAllowedGroups(v ...*Group) *UserUpdateOne {
|
|
ids := make([]int64, len(v))
|
|
for i := range v {
|
|
ids[i] = v[i].ID
|
|
}
|
|
return _u.AddAllowedGroupIDs(ids...)
|
|
}
|
|
|
|
// Mutation returns the UserMutation object of the builder.
|
|
func (_u *UserUpdateOne) Mutation() *UserMutation {
|
|
return _u.mutation
|
|
}
|
|
|
|
// ClearAPIKeys clears all "api_keys" edges to the ApiKey entity.
|
|
func (_u *UserUpdateOne) ClearAPIKeys() *UserUpdateOne {
|
|
_u.mutation.ClearAPIKeys()
|
|
return _u
|
|
}
|
|
|
|
// RemoveAPIKeyIDs removes the "api_keys" edge to ApiKey entities by IDs.
|
|
func (_u *UserUpdateOne) RemoveAPIKeyIDs(ids ...int64) *UserUpdateOne {
|
|
_u.mutation.RemoveAPIKeyIDs(ids...)
|
|
return _u
|
|
}
|
|
|
|
// RemoveAPIKeys removes "api_keys" edges to ApiKey entities.
|
|
func (_u *UserUpdateOne) RemoveAPIKeys(v ...*ApiKey) *UserUpdateOne {
|
|
ids := make([]int64, len(v))
|
|
for i := range v {
|
|
ids[i] = v[i].ID
|
|
}
|
|
return _u.RemoveAPIKeyIDs(ids...)
|
|
}
|
|
|
|
// ClearRedeemCodes clears all "redeem_codes" edges to the RedeemCode entity.
|
|
func (_u *UserUpdateOne) ClearRedeemCodes() *UserUpdateOne {
|
|
_u.mutation.ClearRedeemCodes()
|
|
return _u
|
|
}
|
|
|
|
// RemoveRedeemCodeIDs removes the "redeem_codes" edge to RedeemCode entities by IDs.
|
|
func (_u *UserUpdateOne) RemoveRedeemCodeIDs(ids ...int64) *UserUpdateOne {
|
|
_u.mutation.RemoveRedeemCodeIDs(ids...)
|
|
return _u
|
|
}
|
|
|
|
// RemoveRedeemCodes removes "redeem_codes" edges to RedeemCode entities.
|
|
func (_u *UserUpdateOne) RemoveRedeemCodes(v ...*RedeemCode) *UserUpdateOne {
|
|
ids := make([]int64, len(v))
|
|
for i := range v {
|
|
ids[i] = v[i].ID
|
|
}
|
|
return _u.RemoveRedeemCodeIDs(ids...)
|
|
}
|
|
|
|
// ClearSubscriptions clears all "subscriptions" edges to the UserSubscription entity.
|
|
func (_u *UserUpdateOne) ClearSubscriptions() *UserUpdateOne {
|
|
_u.mutation.ClearSubscriptions()
|
|
return _u
|
|
}
|
|
|
|
// RemoveSubscriptionIDs removes the "subscriptions" edge to UserSubscription entities by IDs.
|
|
func (_u *UserUpdateOne) RemoveSubscriptionIDs(ids ...int64) *UserUpdateOne {
|
|
_u.mutation.RemoveSubscriptionIDs(ids...)
|
|
return _u
|
|
}
|
|
|
|
// RemoveSubscriptions removes "subscriptions" edges to UserSubscription entities.
|
|
func (_u *UserUpdateOne) RemoveSubscriptions(v ...*UserSubscription) *UserUpdateOne {
|
|
ids := make([]int64, len(v))
|
|
for i := range v {
|
|
ids[i] = v[i].ID
|
|
}
|
|
return _u.RemoveSubscriptionIDs(ids...)
|
|
}
|
|
|
|
// ClearAssignedSubscriptions clears all "assigned_subscriptions" edges to the UserSubscription entity.
|
|
func (_u *UserUpdateOne) ClearAssignedSubscriptions() *UserUpdateOne {
|
|
_u.mutation.ClearAssignedSubscriptions()
|
|
return _u
|
|
}
|
|
|
|
// RemoveAssignedSubscriptionIDs removes the "assigned_subscriptions" edge to UserSubscription entities by IDs.
|
|
func (_u *UserUpdateOne) RemoveAssignedSubscriptionIDs(ids ...int64) *UserUpdateOne {
|
|
_u.mutation.RemoveAssignedSubscriptionIDs(ids...)
|
|
return _u
|
|
}
|
|
|
|
// RemoveAssignedSubscriptions removes "assigned_subscriptions" edges to UserSubscription entities.
|
|
func (_u *UserUpdateOne) RemoveAssignedSubscriptions(v ...*UserSubscription) *UserUpdateOne {
|
|
ids := make([]int64, len(v))
|
|
for i := range v {
|
|
ids[i] = v[i].ID
|
|
}
|
|
return _u.RemoveAssignedSubscriptionIDs(ids...)
|
|
}
|
|
|
|
// ClearAllowedGroups clears all "allowed_groups" edges to the Group entity.
|
|
func (_u *UserUpdateOne) ClearAllowedGroups() *UserUpdateOne {
|
|
_u.mutation.ClearAllowedGroups()
|
|
return _u
|
|
}
|
|
|
|
// RemoveAllowedGroupIDs removes the "allowed_groups" edge to Group entities by IDs.
|
|
func (_u *UserUpdateOne) RemoveAllowedGroupIDs(ids ...int64) *UserUpdateOne {
|
|
_u.mutation.RemoveAllowedGroupIDs(ids...)
|
|
return _u
|
|
}
|
|
|
|
// RemoveAllowedGroups removes "allowed_groups" edges to Group entities.
|
|
func (_u *UserUpdateOne) RemoveAllowedGroups(v ...*Group) *UserUpdateOne {
|
|
ids := make([]int64, len(v))
|
|
for i := range v {
|
|
ids[i] = v[i].ID
|
|
}
|
|
return _u.RemoveAllowedGroupIDs(ids...)
|
|
}
|
|
|
|
// Where appends a list predicates to the UserUpdate builder.
|
|
func (_u *UserUpdateOne) Where(ps ...predicate.User) *UserUpdateOne {
|
|
_u.mutation.Where(ps...)
|
|
return _u
|
|
}
|
|
|
|
// Select allows selecting one or more fields (columns) of the returned entity.
|
|
// The default is selecting all fields defined in the entity schema.
|
|
func (_u *UserUpdateOne) Select(field string, fields ...string) *UserUpdateOne {
|
|
_u.fields = append([]string{field}, fields...)
|
|
return _u
|
|
}
|
|
|
|
// Save executes the query and returns the updated User entity.
|
|
func (_u *UserUpdateOne) Save(ctx context.Context) (*User, error) {
|
|
if err := _u.defaults(); err != nil {
|
|
return nil, err
|
|
}
|
|
return withHooks(ctx, _u.sqlSave, _u.mutation, _u.hooks)
|
|
}
|
|
|
|
// SaveX is like Save, but panics if an error occurs.
|
|
func (_u *UserUpdateOne) SaveX(ctx context.Context) *User {
|
|
node, err := _u.Save(ctx)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return node
|
|
}
|
|
|
|
// Exec executes the query on the entity.
|
|
func (_u *UserUpdateOne) Exec(ctx context.Context) error {
|
|
_, err := _u.Save(ctx)
|
|
return err
|
|
}
|
|
|
|
// ExecX is like Exec, but panics if an error occurs.
|
|
func (_u *UserUpdateOne) ExecX(ctx context.Context) {
|
|
if err := _u.Exec(ctx); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
// defaults sets the default values of the builder before save.
|
|
func (_u *UserUpdateOne) defaults() error {
|
|
if _, ok := _u.mutation.UpdatedAt(); !ok {
|
|
if user.UpdateDefaultUpdatedAt == nil {
|
|
return fmt.Errorf("ent: uninitialized user.UpdateDefaultUpdatedAt (forgotten import ent/runtime?)")
|
|
}
|
|
v := user.UpdateDefaultUpdatedAt()
|
|
_u.mutation.SetUpdatedAt(v)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// check runs all checks and user-defined validators on the builder.
|
|
func (_u *UserUpdateOne) check() error {
|
|
if v, ok := _u.mutation.Email(); ok {
|
|
if err := user.EmailValidator(v); err != nil {
|
|
return &ValidationError{Name: "email", err: fmt.Errorf(`ent: validator failed for field "User.email": %w`, err)}
|
|
}
|
|
}
|
|
if v, ok := _u.mutation.PasswordHash(); ok {
|
|
if err := user.PasswordHashValidator(v); err != nil {
|
|
return &ValidationError{Name: "password_hash", err: fmt.Errorf(`ent: validator failed for field "User.password_hash": %w`, err)}
|
|
}
|
|
}
|
|
if v, ok := _u.mutation.Role(); ok {
|
|
if err := user.RoleValidator(v); err != nil {
|
|
return &ValidationError{Name: "role", err: fmt.Errorf(`ent: validator failed for field "User.role": %w`, err)}
|
|
}
|
|
}
|
|
if v, ok := _u.mutation.Status(); ok {
|
|
if err := user.StatusValidator(v); err != nil {
|
|
return &ValidationError{Name: "status", err: fmt.Errorf(`ent: validator failed for field "User.status": %w`, err)}
|
|
}
|
|
}
|
|
if v, ok := _u.mutation.Username(); ok {
|
|
if err := user.UsernameValidator(v); err != nil {
|
|
return &ValidationError{Name: "username", err: fmt.Errorf(`ent: validator failed for field "User.username": %w`, err)}
|
|
}
|
|
}
|
|
if v, ok := _u.mutation.Wechat(); ok {
|
|
if err := user.WechatValidator(v); err != nil {
|
|
return &ValidationError{Name: "wechat", err: fmt.Errorf(`ent: validator failed for field "User.wechat": %w`, err)}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (_u *UserUpdateOne) sqlSave(ctx context.Context) (_node *User, err error) {
|
|
if err := _u.check(); err != nil {
|
|
return _node, err
|
|
}
|
|
_spec := sqlgraph.NewUpdateSpec(user.Table, user.Columns, sqlgraph.NewFieldSpec(user.FieldID, field.TypeInt64))
|
|
id, ok := _u.mutation.ID()
|
|
if !ok {
|
|
return nil, &ValidationError{Name: "id", err: errors.New(`ent: missing "User.id" for update`)}
|
|
}
|
|
_spec.Node.ID.Value = id
|
|
if fields := _u.fields; len(fields) > 0 {
|
|
_spec.Node.Columns = make([]string, 0, len(fields))
|
|
_spec.Node.Columns = append(_spec.Node.Columns, user.FieldID)
|
|
for _, f := range fields {
|
|
if !user.ValidColumn(f) {
|
|
return nil, &ValidationError{Name: f, err: fmt.Errorf("ent: invalid field %q for query", f)}
|
|
}
|
|
if f != user.FieldID {
|
|
_spec.Node.Columns = append(_spec.Node.Columns, f)
|
|
}
|
|
}
|
|
}
|
|
if ps := _u.mutation.predicates; len(ps) > 0 {
|
|
_spec.Predicate = func(selector *sql.Selector) {
|
|
for i := range ps {
|
|
ps[i](selector)
|
|
}
|
|
}
|
|
}
|
|
if value, ok := _u.mutation.UpdatedAt(); ok {
|
|
_spec.SetField(user.FieldUpdatedAt, field.TypeTime, value)
|
|
}
|
|
if value, ok := _u.mutation.DeletedAt(); ok {
|
|
_spec.SetField(user.FieldDeletedAt, field.TypeTime, value)
|
|
}
|
|
if _u.mutation.DeletedAtCleared() {
|
|
_spec.ClearField(user.FieldDeletedAt, field.TypeTime)
|
|
}
|
|
if value, ok := _u.mutation.Email(); ok {
|
|
_spec.SetField(user.FieldEmail, field.TypeString, value)
|
|
}
|
|
if value, ok := _u.mutation.PasswordHash(); ok {
|
|
_spec.SetField(user.FieldPasswordHash, field.TypeString, value)
|
|
}
|
|
if value, ok := _u.mutation.Role(); ok {
|
|
_spec.SetField(user.FieldRole, field.TypeString, value)
|
|
}
|
|
if value, ok := _u.mutation.Balance(); ok {
|
|
_spec.SetField(user.FieldBalance, field.TypeFloat64, value)
|
|
}
|
|
if value, ok := _u.mutation.AddedBalance(); ok {
|
|
_spec.AddField(user.FieldBalance, field.TypeFloat64, value)
|
|
}
|
|
if value, ok := _u.mutation.Concurrency(); ok {
|
|
_spec.SetField(user.FieldConcurrency, field.TypeInt, value)
|
|
}
|
|
if value, ok := _u.mutation.AddedConcurrency(); ok {
|
|
_spec.AddField(user.FieldConcurrency, field.TypeInt, value)
|
|
}
|
|
if value, ok := _u.mutation.Status(); ok {
|
|
_spec.SetField(user.FieldStatus, field.TypeString, value)
|
|
}
|
|
if value, ok := _u.mutation.Username(); ok {
|
|
_spec.SetField(user.FieldUsername, field.TypeString, value)
|
|
}
|
|
if value, ok := _u.mutation.Wechat(); ok {
|
|
_spec.SetField(user.FieldWechat, field.TypeString, value)
|
|
}
|
|
if value, ok := _u.mutation.Notes(); ok {
|
|
_spec.SetField(user.FieldNotes, field.TypeString, value)
|
|
}
|
|
if _u.mutation.APIKeysCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.APIKeysTable,
|
|
Columns: []string{user.APIKeysColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(apikey.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := _u.mutation.RemovedAPIKeysIDs(); len(nodes) > 0 && !_u.mutation.APIKeysCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.APIKeysTable,
|
|
Columns: []string{user.APIKeysColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(apikey.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := _u.mutation.APIKeysIDs(); len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.APIKeysTable,
|
|
Columns: []string{user.APIKeysColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(apikey.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
_spec.Edges.Add = append(_spec.Edges.Add, edge)
|
|
}
|
|
if _u.mutation.RedeemCodesCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.RedeemCodesTable,
|
|
Columns: []string{user.RedeemCodesColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(redeemcode.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := _u.mutation.RemovedRedeemCodesIDs(); len(nodes) > 0 && !_u.mutation.RedeemCodesCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.RedeemCodesTable,
|
|
Columns: []string{user.RedeemCodesColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(redeemcode.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := _u.mutation.RedeemCodesIDs(); len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.RedeemCodesTable,
|
|
Columns: []string{user.RedeemCodesColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(redeemcode.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
_spec.Edges.Add = append(_spec.Edges.Add, edge)
|
|
}
|
|
if _u.mutation.SubscriptionsCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.SubscriptionsTable,
|
|
Columns: []string{user.SubscriptionsColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(usersubscription.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := _u.mutation.RemovedSubscriptionsIDs(); len(nodes) > 0 && !_u.mutation.SubscriptionsCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.SubscriptionsTable,
|
|
Columns: []string{user.SubscriptionsColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(usersubscription.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := _u.mutation.SubscriptionsIDs(); len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.SubscriptionsTable,
|
|
Columns: []string{user.SubscriptionsColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(usersubscription.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
_spec.Edges.Add = append(_spec.Edges.Add, edge)
|
|
}
|
|
if _u.mutation.AssignedSubscriptionsCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.AssignedSubscriptionsTable,
|
|
Columns: []string{user.AssignedSubscriptionsColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(usersubscription.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := _u.mutation.RemovedAssignedSubscriptionsIDs(); len(nodes) > 0 && !_u.mutation.AssignedSubscriptionsCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.AssignedSubscriptionsTable,
|
|
Columns: []string{user.AssignedSubscriptionsColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(usersubscription.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := _u.mutation.AssignedSubscriptionsIDs(); len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.O2M,
|
|
Inverse: false,
|
|
Table: user.AssignedSubscriptionsTable,
|
|
Columns: []string{user.AssignedSubscriptionsColumn},
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(usersubscription.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
_spec.Edges.Add = append(_spec.Edges.Add, edge)
|
|
}
|
|
if _u.mutation.AllowedGroupsCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.M2M,
|
|
Inverse: false,
|
|
Table: user.AllowedGroupsTable,
|
|
Columns: user.AllowedGroupsPrimaryKey,
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(group.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
createE := &UserAllowedGroupCreate{config: _u.config, mutation: newUserAllowedGroupMutation(_u.config, OpCreate)}
|
|
createE.defaults()
|
|
_, specE := createE.createSpec()
|
|
edge.Target.Fields = specE.Fields
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := _u.mutation.RemovedAllowedGroupsIDs(); len(nodes) > 0 && !_u.mutation.AllowedGroupsCleared() {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.M2M,
|
|
Inverse: false,
|
|
Table: user.AllowedGroupsTable,
|
|
Columns: user.AllowedGroupsPrimaryKey,
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(group.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
createE := &UserAllowedGroupCreate{config: _u.config, mutation: newUserAllowedGroupMutation(_u.config, OpCreate)}
|
|
createE.defaults()
|
|
_, specE := createE.createSpec()
|
|
edge.Target.Fields = specE.Fields
|
|
_spec.Edges.Clear = append(_spec.Edges.Clear, edge)
|
|
}
|
|
if nodes := _u.mutation.AllowedGroupsIDs(); len(nodes) > 0 {
|
|
edge := &sqlgraph.EdgeSpec{
|
|
Rel: sqlgraph.M2M,
|
|
Inverse: false,
|
|
Table: user.AllowedGroupsTable,
|
|
Columns: user.AllowedGroupsPrimaryKey,
|
|
Bidi: false,
|
|
Target: &sqlgraph.EdgeTarget{
|
|
IDSpec: sqlgraph.NewFieldSpec(group.FieldID, field.TypeInt64),
|
|
},
|
|
}
|
|
for _, k := range nodes {
|
|
edge.Target.Nodes = append(edge.Target.Nodes, k)
|
|
}
|
|
createE := &UserAllowedGroupCreate{config: _u.config, mutation: newUserAllowedGroupMutation(_u.config, OpCreate)}
|
|
createE.defaults()
|
|
_, specE := createE.createSpec()
|
|
edge.Target.Fields = specE.Fields
|
|
_spec.Edges.Add = append(_spec.Edges.Add, edge)
|
|
}
|
|
_node = &User{config: _u.config}
|
|
_spec.Assign = _node.assignValues
|
|
_spec.ScanValues = _node.scanValues
|
|
if err = sqlgraph.UpdateNode(ctx, _u.driver, _spec); err != nil {
|
|
if _, ok := err.(*sqlgraph.NotFoundError); ok {
|
|
err = &NotFoundError{user.Label}
|
|
} else if sqlgraph.IsConstraintError(err) {
|
|
err = &ConstraintError{msg: err.Error(), wrap: err}
|
|
}
|
|
return nil, err
|
|
}
|
|
_u.mutation.done = true
|
|
return _node, nil
|
|
}
|