소스 검색

add api context to DB

abhishek9686 4 달 전
부모
커밋
9aa9b0e484
1개의 변경된 파일4개의 추가작업 그리고 4개의 파일을 삭제
  1. 4 4
      controllers/egress.go

+ 4 - 4
controllers/egress.go

@@ -75,7 +75,7 @@ func createEgress(w http.ResponseWriter, r *http.Request) {
 		logic.ReturnErrorResponse(w, r, logic.FormatError(errors.New("invalid egress request"), "badrequest"))
 		return
 	}
-	err = e.Create(db.WithContext(context.TODO()))
+	err = e.Create(db.WithContext(r.Context()))
 	if err != nil {
 		logic.ReturnErrorResponse(
 			w,
@@ -105,7 +105,7 @@ func listEgress(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 	e := schema.Egress{Network: network}
-	list, err := e.ListByNetwork(db.WithContext(context.TODO()))
+	list, err := e.ListByNetwork(db.WithContext(r.Context()))
 	if err != nil {
 		logic.ReturnErrorResponse(
 			w,
@@ -148,7 +148,7 @@ func updateEgress(w http.ResponseWriter, r *http.Request) {
 	}
 
 	e := schema.Egress{ID: req.ID}
-	err = e.Get(db.WithContext(context.TODO()))
+	err = e.Get(db.WithContext(r.Context()))
 	if err != nil {
 		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "badrequest"))
 		return
@@ -213,7 +213,7 @@ func deleteEgress(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 	e := schema.Egress{ID: id}
-	err := e.Delete(db.WithContext(context.TODO()))
+	err := e.Delete(db.WithContext(r.Context()))
 	if err != nil {
 		logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
 		return