소스 검색

优化代码

dxc 1 년 전
부모
커밋
07d74ff820
5개의 변경된 파일13개의 추가작업 그리고 3개의 파일을 삭제
  1. 4 0
      internal/svc/servicecontext.go
  2. 2 2
      job/job.go
  3. 5 0
      pkg/storage/bolt.go
  4. 1 1
      transcoder.api
  5. 1 0
      transcoder.go

+ 4 - 0
internal/svc/servicecontext.go

@@ -22,3 +22,7 @@ func NewServiceContext(c config.Config) *ServiceContext {
 		BoltSdk:  storage.NewBoltSdk(),
 	}
 }
+
+func (c *ServiceContext) CloseHandler() {
+	c.BoltSdk.Close()
+}

+ 2 - 2
job/job.go

@@ -58,11 +58,11 @@ func AutoTranscoder(ctx *svc.ServiceContext) {
 			body, err := http.Post(req.CallbackUrl, "application/json", bytes.NewBuffer(bts))
 			if err != nil {
 				logx.Error(err)
-				return
+				continue
 			}
 			if body.StatusCode != http.StatusOK {
 				logx.Errorf("post %v fail, ErrorCode %v", req.CallbackUrl, body.StatusCode)
-				return
+				continue
 			}
 		}
 	}

+ 5 - 0
pkg/storage/bolt.go

@@ -60,9 +60,14 @@ func (s *BoltSdk) GetFirst() (key, value string, err error) {
 	})
 	return
 }
+
 func (s *BoltSdk) Delete(key string) (err error) {
 	err = s.db.Update(func(tx *bolt.Tx) error {
 		return tx.Bucket(s.bucket).Delete([]byte(key))
 	})
 	return
 }
+
+func (s *BoltSdk) Close() error {
+	return s.db.Close()
+}

+ 1 - 1
transcoder.api

@@ -27,7 +27,7 @@ service transcoder {
 	get /v1/transcoder(Request) returns (Response)
 
 	@handler TranscoderCallbackHandler
-	post /v1/transcoder/callback(Request) returns (CallbackResponse)
+	post /v1/transcoder/callback(CallbackRequest) returns (CallbackResponse)
 
 	@handler TestCallbackHandler
 	post /v1/transcoder/test(OriginDataResponse)

+ 1 - 0
transcoder.go

@@ -25,6 +25,7 @@ func main() {
 	defer server.Stop()
 
 	ctx := svc.NewServiceContext(c)
+	defer ctx.CloseHandler()
 	//处理任务
 	go job.AutoTranscoder(ctx)
 	//注册接口