From f6d59d5b4abc720c23f2305458ddaf75504c69be Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=8E=8B=E5=BF=A0=E5=BC=BA?= Date: Wed, 15 Nov 2023 17:32:53 +0800 Subject: [PATCH] no message --- session_get.go | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/session_get.go b/session_get.go index 0d590330..0f653dba 100644 --- a/session_get.go +++ b/session_get.go @@ -34,6 +34,21 @@ func (session *Session) Get(beans ...interface{}) (bool, error) { return session.get(beans...) } +// MustGet like Get, but will return error if record not exist +func (session *Session) MustGet(beans ...interface{}) error { + if session.isAutoClose { + defer session.Close() + } + exist, err := session.get(beans...) + if err != nil { + return err + } else if !exist { + return ErrNotExist + } else { + return nil + } +} + func isPtrOfTime(v interface{}) bool { if _, ok := v.(*time.Time); ok { return true