TRKJ/EdgeAPI
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeAPI.git synced 2025-11-04 07:50:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化代码

Browse Source
This commit is contained in:
GoEdgeLab
2022-12-12 10:28:22 +08:00
parent ea6b1bcbfb
commit e745ad6dfd
11 changed files with 15 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
internal/db/models/client_browser_model_ext.go
View File

@@ -1 +0,0 @@
package models

1
internal/db/models/client_system_model_ext.go
View File

@@ -1 +0,0 @@
package models

5
internal/db/models/client_browser_dao.go → internal/db/models/clients/client_browser_dao.go
View File

@@ -1,6 +1,7 @@
package models
package clients
import (
"github.com/TeaOSLab/EdgeAPI/internal/db/models"
"github.com/TeaOSLab/EdgeAPI/internal/goman"
"github.com/TeaOSLab/EdgeAPI/internal/remotelogs"
"github.com/TeaOSLab/EdgeAPI/internal/utils/ttlcache"
@@ -130,7 +131,7 @@ func (this *ClientBrowserDAO) CreateBrowserIfNotExists(tx *dbs.Tx, browserName s
op.CreatedDay = timeutil.Format("Ymd")
op.State = ClientBrowserStateEnabled
browserId, err = this.SaveInt64(tx, op)
if err != nil && CheckSQLErrCode(err, 1062 /** duplicate entry **/) {
if err != nil && models.CheckSQLErrCode(err, 1062 /** duplicate entry **/) {
return nil
}

2
internal/db/models/client_browser_dao_test.go → internal/db/models/clients/client_browser_dao_test.go
View File

@@ -1,4 +1,4 @@
package models_test
package clients_test
import (
"github.com/TeaOSLab/EdgeAPI/internal/db/models"

2
internal/db/models/client_browser_model.go → internal/db/models/clients/client_browser_model.go
View File

@@ -1,4 +1,4 @@
package models
package clients
import "github.com/iwind/TeaGo/dbs"

1
internal/db/models/clients/client_browser_model_ext.go Normal file
View File

@@ -0,0 +1 @@
package clients

5
internal/db/models/client_system_dao.go → internal/db/models/clients/client_system_dao.go
View File

@@ -1,6 +1,7 @@
package models
package clients
import (
"github.com/TeaOSLab/EdgeAPI/internal/db/models"
"github.com/TeaOSLab/EdgeAPI/internal/goman"
"github.com/TeaOSLab/EdgeAPI/internal/remotelogs"
"github.com/TeaOSLab/EdgeAPI/internal/utils/ttlcache"
@@ -129,7 +130,7 @@ func (this *ClientSystemDAO) CreateSystemIfNotExists(tx *dbs.Tx, systemName stri
op.CreatedDay = timeutil.Format("Ymd")
op.State = ClientSystemStateEnabled
systemId, err = this.SaveInt64(tx, op)
if err != nil && CheckSQLErrCode(err, 1062 /** duplicate entry **/) {
if err != nil && models.CheckSQLErrCode(err, 1062 /** duplicate entry **/) {
return nil
}

2
internal/db/models/client_system_dao_test.go → internal/db/models/clients/client_system_dao_test.go
View File

@@ -1,4 +1,4 @@
package models_test
package clients_test
import (
"github.com/TeaOSLab/EdgeAPI/internal/db/models"

2
internal/db/models/client_system_model.go → internal/db/models/clients/client_system_model.go
View File

@@ -1,4 +1,4 @@
package models
package clients
import "github.com/iwind/TeaGo/dbs"

1
internal/db/models/clients/client_system_model_ext.go Normal file
View File

@@ -0,0 +1 @@
package clients

5
internal/rpc/services/service_server.go
View File

@@ -6,6 +6,7 @@ import (
"errors"
"fmt"
"github.com/TeaOSLab/EdgeAPI/internal/db/models"
"github.com/TeaOSLab/EdgeAPI/internal/db/models/clients"
"github.com/TeaOSLab/EdgeAPI/internal/db/models/dns"
"github.com/TeaOSLab/EdgeAPI/internal/db/models/regions"
"github.com/TeaOSLab/EdgeAPI/internal/utils"
@@ -1714,7 +1715,7 @@ func (this *ServerService) UploadServerHTTPRequestStat(ctx context.Context, req
return err
}
if systemId == 0 {
err = models.SharedClientSystemDAO.CreateSystemIfNotExists(tx, result.Name)
err = clients.SharedClientSystemDAO.CreateSystemIfNotExists(tx, result.Name)
if err != nil {
return err
}
@@ -1745,7 +1746,7 @@ func (this *ServerService) UploadServerHTTPRequestStat(ctx context.Context, req
return err
}
if browserId == 0 {
err = models.SharedClientBrowserDAO.CreateBrowserIfNotExists(tx, result.Name)
err = clients.SharedClientBrowserDAO.CreateBrowserIfNotExists(tx, result.Name)
if err != nil {
return err
}