From 4e95ccbf21d156688c8c3ba038d3c4a78a71d27b Mon Sep 17 00:00:00 2001
From: xxjwxc <xie1xiao1jun@qq.com>
Date: Wed, 22 Jun 2022 23:15:30 +0800
Subject: [PATCH] Update def.go

---
 data/view/genfunc/def.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/data/view/genfunc/def.go b/data/view/genfunc/def.go
index 6b100b9..52486ee 100755
--- a/data/view/genfunc/def.go
+++ b/data/view/genfunc/def.go
@@ -19,6 +19,7 @@ var {{.StructName}}Columns = struct { {{range $em := .Em}}
 package {{.PackageName}}
 import (
 	"context"
+	"errors"
 	"fmt"
 	"time"
 
@@ -27,6 +28,11 @@ import (
 
 var globalIsRelated bool = true  // 全局预加载
 
+// IsNotFound ErrRecordNotFound
+func IsNotFound(err error) bool {
+	return errors.Is(err, gorm.ErrRecordNotFound)
+}
+
 // prepare for other
 type _BaseMgr struct {
 	*gorm.DB
@@ -89,6 +95,11 @@ func (obj *_BaseMgr) NewDB() *gorm.DB {
 	return obj.DB.Session(&gorm.Session{NewDB: true, Context: obj.ctx})
 }
 
+// IsNotFound ErrRecordNotFound
+func (obj *_BaseMgr) IsNotFound(err error) bool {
+	return errors.Is(err, gorm.ErrRecordNotFound)
+}
+
 type options struct {
 	query map[string]interface{}
 }