jiazhizhong/polaris
1
0
Fork 0
mirror of https://github.com/simon-ding/polaris.git synced 2026-06-09 11:39:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: update stun proxy logic

Browse Source
This commit is contained in:
Simon Ding
2025-05-08 16:13:30 +08:00
parent bb2c567da7
commit 992fa7ddd0
12 changed files with 310 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
ent/downloadclients_update.go
View File

@@ -146,6 +146,26 @@ func (dcu *DownloadClientsUpdate) AddPriority1(i int) *DownloadClientsUpdate {
return dcu
}
// SetUseNatTraversal sets the "use_nat_traversal" field.
func (dcu *DownloadClientsUpdate) SetUseNatTraversal(b bool) *DownloadClientsUpdate {
dcu.mutation.SetUseNatTraversal(b)
return dcu
}
// SetNillableUseNatTraversal sets the "use_nat_traversal" field if the given value is not nil.
func (dcu *DownloadClientsUpdate) SetNillableUseNatTraversal(b *bool) *DownloadClientsUpdate {
if b != nil {
dcu.SetUseNatTraversal(*b)
}
return dcu
}
// ClearUseNatTraversal clears the value of the "use_nat_traversal" field.
func (dcu *DownloadClientsUpdate) ClearUseNatTraversal() *DownloadClientsUpdate {
dcu.mutation.ClearUseNatTraversal()
return dcu
}
// SetRemoveCompletedDownloads sets the "remove_completed_downloads" field.
func (dcu *DownloadClientsUpdate) SetRemoveCompletedDownloads(b bool) *DownloadClientsUpdate {
dcu.mutation.SetRemoveCompletedDownloads(b)
@@ -274,6 +294,12 @@ func (dcu *DownloadClientsUpdate) sqlSave(ctx context.Context) (n int, err error
if value, ok := dcu.mutation.AddedPriority1(); ok {
_spec.AddField(downloadclients.FieldPriority1, field.TypeInt, value)
}
if value, ok := dcu.mutation.UseNatTraversal(); ok {
_spec.SetField(downloadclients.FieldUseNatTraversal, field.TypeBool, value)
}
if dcu.mutation.UseNatTraversalCleared() {
_spec.ClearField(downloadclients.FieldUseNatTraversal, field.TypeBool)
}
if value, ok := dcu.mutation.RemoveCompletedDownloads(); ok {
_spec.SetField(downloadclients.FieldRemoveCompletedDownloads, field.TypeBool, value)
}
@@ -425,6 +451,26 @@ func (dcuo *DownloadClientsUpdateOne) AddPriority1(i int) *DownloadClientsUpdate
return dcuo
}
// SetUseNatTraversal sets the "use_nat_traversal" field.
func (dcuo *DownloadClientsUpdateOne) SetUseNatTraversal(b bool) *DownloadClientsUpdateOne {
dcuo.mutation.SetUseNatTraversal(b)
return dcuo
}
// SetNillableUseNatTraversal sets the "use_nat_traversal" field if the given value is not nil.
func (dcuo *DownloadClientsUpdateOne) SetNillableUseNatTraversal(b *bool) *DownloadClientsUpdateOne {
if b != nil {
dcuo.SetUseNatTraversal(*b)
}
return dcuo
}
// ClearUseNatTraversal clears the value of the "use_nat_traversal" field.
func (dcuo *DownloadClientsUpdateOne) ClearUseNatTraversal() *DownloadClientsUpdateOne {
dcuo.mutation.ClearUseNatTraversal()
return dcuo
}
// SetRemoveCompletedDownloads sets the "remove_completed_downloads" field.
func (dcuo *DownloadClientsUpdateOne) SetRemoveCompletedDownloads(b bool) *DownloadClientsUpdateOne {
dcuo.mutation.SetRemoveCompletedDownloads(b)
@@ -583,6 +629,12 @@ func (dcuo *DownloadClientsUpdateOne) sqlSave(ctx context.Context) (_node *Downl
if value, ok := dcuo.mutation.AddedPriority1(); ok {
_spec.AddField(downloadclients.FieldPriority1, field.TypeInt, value)
}
if value, ok := dcuo.mutation.UseNatTraversal(); ok {
_spec.SetField(downloadclients.FieldUseNatTraversal, field.TypeBool, value)
}
if dcuo.mutation.UseNatTraversalCleared() {
_spec.ClearField(downloadclients.FieldUseNatTraversal, field.TypeBool)
}
if value, ok := dcuo.mutation.RemoveCompletedDownloads(); ok {
_spec.SetField(downloadclients.FieldRemoveCompletedDownloads, field.TypeBool, value)
}