diff --git a/Makefile b/Makefile index 4702996..ed1d893 100644 --- a/Makefile +++ b/Makefile @@ -1,17 +1,17 @@ all: # 构建 make tar windows: - CGO_ENABLED=1 GOOS=windows GOARCH=amd64 go build -o gormt.exe main.go + CGO_ENABLED=0 GOOS=windows GOARCH=amd64 go build -o gormt.exe main.go mac: CGO_ENABLED=1 GOOS=darwin GOARCH=amd64 go build -o gormt main.go linux: - CGO_ENABLED=1 GOOS=linux GOARCH=amd64 go build -o gormt main.go + CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -o gormt main.go tar: # 打包 - CGO_ENABLED=1 GOOS=windows GOARCH=amd64 go build -o gormt.exe main.go + CGO_ENABLED=0 GOOS=windows GOARCH=amd64 go build -o gormt.exe main.go tar czvf gormt_windows.zip gormt.exe config.yml CGO_ENABLED=1 GOOS=darwin GOARCH=amd64 go build -o gormt main.go tar czvf gormt_mac.zip gormt config.yml - CGO_ENABLED=1 GOOS=linux GOARCH=amd64 go build -o gormt main.go + CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -o gormt main.go tar czvf gormt_linux.zip gormt config.yml clear: test ! -d model/ || rm -rf model/* diff --git a/data/view/genfunc/def.go b/data/view/genfunc/def.go index 83658ec..58ecb7c 100644 --- a/data/view/genfunc/def.go +++ b/data/view/genfunc/def.go @@ -35,7 +35,19 @@ func (obj *_BaseMgr) SetTimeOut(timeout time.Duration) { // SetCtx set context func (obj *_BaseMgr) SetCtx(c context.Context) { - obj.ctx = c + if c != nil { + obj.ctx = c + } +} + +// Ctx get context +func (obj *_BaseMgr) GetCtx() context.Context { + return obj.ctx +} + +// Cancel cancel context +func (obj *_BaseMgr) Cancel(c context.Context) { + obj.cancel() } // GetDB get gorm.DB info @@ -112,14 +124,14 @@ func (obj *_{{$obj.StructName}}Mgr) GetTableName() string { // Get 获取 func (obj *_{{$obj.StructName}}Mgr) Get() (result {{$obj.StructName}}, err error) { - err = obj.DB.Table(obj.GetTableName()).Find(&result).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Find(&result).Error {{GenPreloadList $obj.PreloadList false}} return } // Gets 获取批量结果 func (obj *_{{$obj.StructName}}Mgr) Gets() (results []*{{$obj.StructName}}, err error) { - err = obj.DB.Table(obj.GetTableName()).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Find(&results).Error {{GenPreloadList $obj.PreloadList true}} return } @@ -141,7 +153,7 @@ func (obj *_{{$obj.StructName}}Mgr) GetByOption(opts ...Option) (result {{$obj.S o.apply(&options) } - err = obj.DB.Table(obj.GetTableName()).Where(options.query).Find(&result).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where(options.query).Find(&result).Error {{GenPreloadList $obj.PreloadList false}} return } @@ -155,7 +167,7 @@ func (obj *_{{$obj.StructName}}Mgr) GetByOptions(opts ...Option) (results []*{{$ o.apply(&options) } - err = obj.DB.Table(obj.GetTableName()).Where(options.query).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where(options.query).Find(&results).Error {{GenPreloadList $obj.PreloadList true}} return } @@ -164,20 +176,20 @@ func (obj *_{{$obj.StructName}}Mgr) GetByOptions(opts ...Option) (results []*{{$ {{range $oem := $obj.Em}} // GetFrom{{$oem.ColStructName}} 通过{{$oem.ColName}}获取内容 {{$oem.Notes}} {{if $oem.IsMulti}} func (obj *_{{$obj.StructName}}Mgr) GetFrom{{$oem.ColStructName}}({{CapLowercase $oem.ColStructName}} {{$oem.Type}}) (results []*{{$obj.StructName}}, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("{{$oem.ColName}} = ?", {{CapLowercase $oem.ColStructName}}).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("{{$oem.ColName}} = ?", {{CapLowercase $oem.ColStructName}}).Find(&results).Error {{GenPreloadList $obj.PreloadList true}} return } {{else}} func (obj *_{{$obj.StructName}}Mgr) GetFrom{{$oem.ColStructName}}({{CapLowercase $oem.ColStructName}} {{$oem.Type}}) (result {{$obj.StructName}}, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("{{$oem.ColName}} = ?", {{CapLowercase $oem.ColStructName}}).Find(&result).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("{{$oem.ColName}} = ?", {{CapLowercase $oem.ColStructName}}).Find(&result).Error {{GenPreloadList $obj.PreloadList false}} return } {{end}} // GetBatchFrom{{$oem.ColStructName}} 批量唯一主键查找 {{$oem.Notes}} func (obj *_{{$obj.StructName}}Mgr) GetBatchFrom{{$oem.ColStructName}}({{CapLowercase $oem.ColStructName}}s []{{$oem.Type}}) (results []*{{$obj.StructName}}, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("{{$oem.ColName}} IN (?)", {{CapLowercase $oem.ColStructName}}s).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("{{$oem.ColName}} IN (?)", {{CapLowercase $oem.ColStructName}}s).Find(&results).Error {{GenPreloadList $obj.PreloadList true}} return } @@ -186,7 +198,7 @@ func (obj *_{{$obj.StructName}}Mgr) GetBatchFrom{{$oem.ColStructName}}({{CapLowe {{range $ofm := $obj.Primay}} // {{GenFListIndex $ofm 1}} primay or index 获取唯一内容 func (obj *_{{$obj.StructName}}Mgr) {{GenFListIndex $ofm 1}}({{GenFListIndex $ofm 2}}) (result {{$obj.StructName}}, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("{{GenFListIndex $ofm 3}}", {{GenFListIndex $ofm 4}}).Find(&result).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("{{GenFListIndex $ofm 3}}", {{GenFListIndex $ofm 4}}).Find(&result).Error {{GenPreloadList $obj.PreloadList false}} return } @@ -195,7 +207,7 @@ func (obj *_{{$obj.StructName}}Mgr) GetBatchFrom{{$oem.ColStructName}}({{CapLowe {{range $ofm := $obj.Index}} // {{GenFListIndex $ofm 1}} 获取多个内容 func (obj *_{{$obj.StructName}}Mgr) {{GenFListIndex $ofm 1}}({{GenFListIndex $ofm 2}}) (results []*{{$obj.StructName}}, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("{{GenFListIndex $ofm 3}}", {{GenFListIndex $ofm 4}}).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("{{GenFListIndex $ofm 3}}", {{GenFListIndex $ofm 4}}).Find(&results).Error {{GenPreloadList $obj.PreloadList true}} return } diff --git a/data/view/genfunc/model/gen.base.go b/data/view/genfunc/model/gen.base.go index 4a6bf2d..241a88c 100644 --- a/data/view/genfunc/model/gen.base.go +++ b/data/view/genfunc/model/gen.base.go @@ -26,7 +26,19 @@ func (obj *_BaseMgr) SetTimeOut(timeout time.Duration) { // SetCtx set context func (obj *_BaseMgr) SetCtx(c context.Context) { - obj.ctx = c + if c != nil { + obj.ctx = c + } +} + +// Ctx get context +func (obj *_BaseMgr) GetCtx() context.Context { + return obj.ctx +} + +// Cancel cancel context +func (obj *_BaseMgr) Cancel(c context.Context) { + obj.cancel() } // GetDB get gorm.DB info diff --git a/data/view/genfunc/model/matrix.gen.account.go b/data/view/genfunc/model/matrix.gen.account.go index b4a23a3..00a02c2 100644 --- a/data/view/genfunc/model/matrix.gen.account.go +++ b/data/view/genfunc/model/matrix.gen.account.go @@ -27,7 +27,7 @@ func (obj *_AccountMgr) GetTableName() string { // Get 获取 func (obj *_AccountMgr) Get() (result Account, err error) { - err = obj.DB.Table(obj.GetTableName()).Find(&result).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Find(&result).Error if err == nil && obj.isRelated { if err = obj.New().Table("user").Where("user_id = ?", result.UserID).Find(&result.User).Error; err != nil { // if err != gorm.ErrRecordNotFound { // 非 没找到 @@ -41,7 +41,7 @@ func (obj *_AccountMgr) Get() (result Account, err error) { // Gets 获取批量结果 func (obj *_AccountMgr) Gets() (results []*Account, err error) { - err = obj.DB.Table(obj.GetTableName()).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Find(&results).Error if err == nil && obj.isRelated { for i := 0; i < len(results); i++ { if err = obj.New().Table("user").Where("user_id = ?", results[i].UserID).Find(&results[i].User).Error; err != nil { // @@ -90,7 +90,7 @@ func (obj *_AccountMgr) GetByOption(opts ...Option) (result Account, err error) o.apply(&options) } - err = obj.DB.Table(obj.GetTableName()).Where(options.query).Find(&result).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where(options.query).Find(&result).Error if err == nil && obj.isRelated { if err = obj.New().Table("user").Where("user_id = ?", result.UserID).Find(&result.User).Error; err != nil { // if err != gorm.ErrRecordNotFound { // 非 没找到 @@ -111,7 +111,7 @@ func (obj *_AccountMgr) GetByOptions(opts ...Option) (results []*Account, err er o.apply(&options) } - err = obj.DB.Table(obj.GetTableName()).Where(options.query).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where(options.query).Find(&results).Error if err == nil && obj.isRelated { for i := 0; i < len(results); i++ { if err = obj.New().Table("user").Where("user_id = ?", results[i].UserID).Find(&results[i].User).Error; err != nil { // @@ -128,7 +128,7 @@ func (obj *_AccountMgr) GetByOptions(opts ...Option) (results []*Account, err er // GetFromID 通过id获取内容 func (obj *_AccountMgr) GetFromID(id int) (result Account, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("id = ?", id).Find(&result).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("id = ?", id).Find(&result).Error if err == nil && obj.isRelated { if err = obj.New().Table("user").Where("user_id = ?", result.UserID).Find(&result.User).Error; err != nil { // if err != gorm.ErrRecordNotFound { // 非 没找到 @@ -142,7 +142,7 @@ func (obj *_AccountMgr) GetFromID(id int) (result Account, err error) { // GetBatchFromID 批量唯一主键查找 func (obj *_AccountMgr) GetBatchFromID(ids []int) (results []*Account, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("id IN (?)", ids).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("id IN (?)", ids).Find(&results).Error if err == nil && obj.isRelated { for i := 0; i < len(results); i++ { if err = obj.New().Table("user").Where("user_id = ?", results[i].UserID).Find(&results[i].User).Error; err != nil { // @@ -157,7 +157,7 @@ func (obj *_AccountMgr) GetBatchFromID(ids []int) (results []*Account, err error // GetFromAccountID 通过account_id获取内容 func (obj *_AccountMgr) GetFromAccountID(accountID int) (results []*Account, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("account_id = ?", accountID).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("account_id = ?", accountID).Find(&results).Error if err == nil && obj.isRelated { for i := 0; i < len(results); i++ { if err = obj.New().Table("user").Where("user_id = ?", results[i].UserID).Find(&results[i].User).Error; err != nil { // @@ -172,7 +172,7 @@ func (obj *_AccountMgr) GetFromAccountID(accountID int) (results []*Account, err // GetBatchFromAccountID 批量唯一主键查找 func (obj *_AccountMgr) GetBatchFromAccountID(accountIDs []int) (results []*Account, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("account_id IN (?)", accountIDs).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("account_id IN (?)", accountIDs).Find(&results).Error if err == nil && obj.isRelated { for i := 0; i < len(results); i++ { if err = obj.New().Table("user").Where("user_id = ?", results[i].UserID).Find(&results[i].User).Error; err != nil { // @@ -187,7 +187,7 @@ func (obj *_AccountMgr) GetBatchFromAccountID(accountIDs []int) (results []*Acco // GetFromUserID 通过user_id获取内容 func (obj *_AccountMgr) GetFromUserID(userID int) (results []*Account, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("user_id = ?", userID).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("user_id = ?", userID).Find(&results).Error if err == nil && obj.isRelated { for i := 0; i < len(results); i++ { if err = obj.New().Table("user").Where("user_id = ?", results[i].UserID).Find(&results[i].User).Error; err != nil { // @@ -202,7 +202,7 @@ func (obj *_AccountMgr) GetFromUserID(userID int) (results []*Account, err error // GetBatchFromUserID 批量唯一主键查找 func (obj *_AccountMgr) GetBatchFromUserID(userIDs []int) (results []*Account, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("user_id IN (?)", userIDs).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("user_id IN (?)", userIDs).Find(&results).Error if err == nil && obj.isRelated { for i := 0; i < len(results); i++ { if err = obj.New().Table("user").Where("user_id = ?", results[i].UserID).Find(&results[i].User).Error; err != nil { // @@ -217,7 +217,7 @@ func (obj *_AccountMgr) GetBatchFromUserID(userIDs []int) (results []*Account, e // GetFromType 通过type获取内容 func (obj *_AccountMgr) GetFromType(_type int) (results []*Account, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("type = ?", _type).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("type = ?", _type).Find(&results).Error if err == nil && obj.isRelated { for i := 0; i < len(results); i++ { if err = obj.New().Table("user").Where("user_id = ?", results[i].UserID).Find(&results[i].User).Error; err != nil { // @@ -232,7 +232,7 @@ func (obj *_AccountMgr) GetFromType(_type int) (results []*Account, err error) { // GetBatchFromType 批量唯一主键查找 func (obj *_AccountMgr) GetBatchFromType(_types []int) (results []*Account, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("type IN (?)", _types).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("type IN (?)", _types).Find(&results).Error if err == nil && obj.isRelated { for i := 0; i < len(results); i++ { if err = obj.New().Table("user").Where("user_id = ?", results[i].UserID).Find(&results[i].User).Error; err != nil { // @@ -247,7 +247,7 @@ func (obj *_AccountMgr) GetBatchFromType(_types []int) (results []*Account, err // GetFromName 通过name获取内容 func (obj *_AccountMgr) GetFromName(name string) (results []*Account, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("name = ?", name).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("name = ?", name).Find(&results).Error if err == nil && obj.isRelated { for i := 0; i < len(results); i++ { if err = obj.New().Table("user").Where("user_id = ?", results[i].UserID).Find(&results[i].User).Error; err != nil { // @@ -262,7 +262,7 @@ func (obj *_AccountMgr) GetFromName(name string) (results []*Account, err error) // GetBatchFromName 批量唯一主键查找 func (obj *_AccountMgr) GetBatchFromName(names []string) (results []*Account, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("name IN (?)", names).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("name IN (?)", names).Find(&results).Error if err == nil && obj.isRelated { for i := 0; i < len(results); i++ { if err = obj.New().Table("user").Where("user_id = ?", results[i].UserID).Find(&results[i].User).Error; err != nil { // @@ -279,7 +279,7 @@ func (obj *_AccountMgr) GetBatchFromName(names []string) (results []*Account, er // FetchByPrimaryKey primay or index 获取唯一内容 func (obj *_AccountMgr) FetchByPrimaryKey(id int) (result Account, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("id = ?", id).Find(&result).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("id = ?", id).Find(&result).Error if err == nil && obj.isRelated { if err = obj.New().Table("user").Where("user_id = ?", result.UserID).Find(&result.User).Error; err != nil { // if err != gorm.ErrRecordNotFound { // 非 没找到 @@ -293,7 +293,7 @@ func (obj *_AccountMgr) FetchByPrimaryKey(id int) (result Account, err error) { // FetchUniqueIndexByAccount primay or index 获取唯一内容 func (obj *_AccountMgr) FetchUniqueIndexByAccount(accountID int, userID int) (result Account, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("account_id = ? AND user_id = ?", accountID, userID).Find(&result).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("account_id = ? AND user_id = ?", accountID, userID).Find(&result).Error if err == nil && obj.isRelated { if err = obj.New().Table("user").Where("user_id = ?", result.UserID).Find(&result.User).Error; err != nil { // if err != gorm.ErrRecordNotFound { // 非 没找到 @@ -307,7 +307,7 @@ func (obj *_AccountMgr) FetchUniqueIndexByAccount(accountID int, userID int) (re // FetchIndexByTp 获取多个内容 func (obj *_AccountMgr) FetchIndexByTp(userID int, _type int) (results []*Account, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("user_id = ? AND type = ?", userID, _type).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("user_id = ? AND type = ?", userID, _type).Find(&results).Error if err == nil && obj.isRelated { for i := 0; i < len(results); i++ { if err = obj.New().Table("user").Where("user_id = ?", results[i].UserID).Find(&results[i].User).Error; err != nil { // diff --git a/data/view/genfunc/model/matrix.gen.user.go b/data/view/genfunc/model/matrix.gen.user.go index 3b177aa..9b04150 100644 --- a/data/view/genfunc/model/matrix.gen.user.go +++ b/data/view/genfunc/model/matrix.gen.user.go @@ -27,14 +27,14 @@ func (obj *_UserMgr) GetTableName() string { // Get 获取 func (obj *_UserMgr) Get() (result User, err error) { - err = obj.DB.Table(obj.GetTableName()).Find(&result).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Find(&result).Error return } // Gets 获取批量结果 func (obj *_UserMgr) Gets() (results []*User, err error) { - err = obj.DB.Table(obj.GetTableName()).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Find(&results).Error return } @@ -70,7 +70,7 @@ func (obj *_UserMgr) GetByOption(opts ...Option) (result User, err error) { o.apply(&options) } - err = obj.DB.Table(obj.GetTableName()).Where(options.query).Find(&result).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where(options.query).Find(&result).Error return } @@ -84,7 +84,7 @@ func (obj *_UserMgr) GetByOptions(opts ...Option) (results []*User, err error) { o.apply(&options) } - err = obj.DB.Table(obj.GetTableName()).Where(options.query).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where(options.query).Find(&results).Error return } @@ -93,56 +93,56 @@ func (obj *_UserMgr) GetByOptions(opts ...Option) (results []*User, err error) { // GetFromUserID 通过user_id获取内容 func (obj *_UserMgr) GetFromUserID(userID int) (result User, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("user_id = ?", userID).Find(&result).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("user_id = ?", userID).Find(&result).Error return } // GetBatchFromUserID 批量唯一主键查找 func (obj *_UserMgr) GetBatchFromUserID(userIDs []int) (results []*User, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("user_id IN (?)", userIDs).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("user_id IN (?)", userIDs).Find(&results).Error return } // GetFromName 通过name获取内容 func (obj *_UserMgr) GetFromName(name string) (results []*User, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("name = ?", name).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("name = ?", name).Find(&results).Error return } // GetBatchFromName 批量唯一主键查找 func (obj *_UserMgr) GetBatchFromName(names []string) (results []*User, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("name IN (?)", names).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("name IN (?)", names).Find(&results).Error return } // GetFromSex 通过sex获取内容 func (obj *_UserMgr) GetFromSex(sex int) (results []*User, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("sex = ?", sex).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("sex = ?", sex).Find(&results).Error return } // GetBatchFromSex 批量唯一主键查找 func (obj *_UserMgr) GetBatchFromSex(sexs []int) (results []*User, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("sex IN (?)", sexs).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("sex IN (?)", sexs).Find(&results).Error return } // GetFromJob 通过job获取内容 func (obj *_UserMgr) GetFromJob(job int) (results []*User, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("job = ?", job).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("job = ?", job).Find(&results).Error return } // GetBatchFromJob 批量唯一主键查找 func (obj *_UserMgr) GetBatchFromJob(jobs []int) (results []*User, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("job IN (?)", jobs).Find(&results).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("job IN (?)", jobs).Find(&results).Error return } @@ -151,7 +151,7 @@ func (obj *_UserMgr) GetBatchFromJob(jobs []int) (results []*User, err error) { // FetchByPrimaryKey primay or index 获取唯一内容 func (obj *_UserMgr) FetchByPrimaryKey(userID int) (result User, err error) { - err = obj.DB.Table(obj.GetTableName()).Where("user_id = ?", userID).Find(&result).Error + err = obj.WithContext(obj.ctx).Table(obj.GetTableName()).Where("user_id = ?", userID).Find(&result).Error return }