Преглед на файлове

feat(go): restore method;

Vishal Dalwadi преди 4 месеца
родител
ревизия
a7fd5ccfe0
променени са 1 файла, в които са добавени 4 реда и са изтрити 0 реда
  1. 4 0
      db/db.go

+ 4 - 0
db/db.go

@@ -75,6 +75,10 @@ func Middleware(next http.Handler) http.Handler {
 //
 // The function panics, if a connection does not exist.
 func FromContext(ctx context.Context) *gorm.DB {
+	db, ok := ctx.Value(dbCtxKey).(*gorm.DB)
+	if !ok {
+		panic(ErrDBNotFound)
+	}
 
 	return db
 }