totole/services
1
0
Fork 0
mirror of https://github.com/kevin-DL/services.git synced 2026-01-21 23:15:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

use tableName func for Count

Browse Source
This commit is contained in:
Asim Aslam
2021-10-29 09:33:34 +01:00
parent c9d933b0ba
commit 289ae7531c
1 changed files with 4 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
db/handler/db.go
View File

@@ -321,17 +321,13 @@ func (e *Db) Truncate(ctx context.Context, req *db.TruncateRequest, rsp *db.Trun
} }
func (e *Db) Count(ctx context.Context, req *db.CountRequest, rsp *db.CountResponse) error { func (e *Db) Count(ctx context.Context, req *db.CountRequest, rsp *db.CountResponse) error {
tenantId, ok := tenant.FromContext(ctx)
if !ok {
tenantId = "micro"
}
if req.Table == "" { if req.Table == "" {
req.Table = "default" req.Table = "default"
} }
tenantId = strings.Replace(strings.Replace(tenantId, "/", "_", -1), "-", "_", -1)
tableName := tenantId + "_" + req.Table tableName, err := e.tableName(ctx, req.Table)
if !re.Match([]byte(tableName)) { if err != nil {
return errors.BadRequest("db.count", fmt.Sprintf("table name %v is invalid", req.Table)) return err
} }
db, err := e.GetDBConn(ctx) db, err := e.GetDBConn(ctx)