Browse Source

添加支持服务shutdown

fancl 2 years ago
parent
commit
7d0b299319
4 changed files with 10 additions and 0 deletions
  1. 1 0
      go.mod
  2. 2 0
      go.sum
  3. 5 0
      service.go
  4. 2 0
      vendor/modules.txt

+ 1 - 0
go.mod

@@ -4,6 +4,7 @@ go 1.15
 
 require (
 	github.com/google/btree v1.0.1
+	github.com/judwhite/go-svc v1.2.1 // indirect
 	github.com/peterh/liner v1.2.1
 	github.com/prometheus/client_golang v1.11.0
 	golang.org/x/net v0.0.0-20210226172049-e18ecbb05110

+ 2 - 0
go.sum

@@ -44,6 +44,8 @@ github.com/jpillora/backoff v1.0.0/go.mod h1:J/6gKK9jxlEcS3zixgDgUAsiuZ7yrSoa/FX
 github.com/json-iterator/go v1.1.6/go.mod h1:+SdeFBvtyEkXs7REEP0seUULqWtbJapLOCVDaaPEHmU=
 github.com/json-iterator/go v1.1.10/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=
 github.com/json-iterator/go v1.1.11/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=
+github.com/judwhite/go-svc v1.2.1 h1:a7fsJzYUa33sfDJRF2N/WXhA+LonCEEY8BJb1tuS5tA=
+github.com/judwhite/go-svc v1.2.1/go.mod h1:mo/P2JNX8C07ywpP9YtO2gnBgnUiFTHqtsZekJrUuTk=
 github.com/julienschmidt/httprouter v1.2.0/go.mod h1:SYymIcj16QtmaHHD7aYtjjsJG7VTCxuUUipMqKk8s4w=
 github.com/julienschmidt/httprouter v1.3.0/go.mod h1:JR6WtHb+2LUe8TCKY3cZOxFyyO8IZAc4RVcycCCAKdM=
 github.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=

+ 5 - 0
service.go

@@ -530,6 +530,11 @@ func (svr *Service) Run() (err error) {
 	return svr.destroy()
 }
 
+//Shutdown close service
+func (svr *Service) Shutdown() {
+	svr.cancelFunc()
+}
+
 func New(opts ...Option) *Service {
 	o := NewOptions()
 	for _, opt := range opts {

+ 2 - 0
vendor/modules.txt

@@ -11,6 +11,8 @@ github.com/golang/protobuf/ptypes/timestamp
 # github.com/google/btree v1.0.1
 ## explicit
 github.com/google/btree
+# github.com/judwhite/go-svc v1.2.1
+## explicit
 # github.com/mattn/go-runewidth v0.0.3
 github.com/mattn/go-runewidth
 # github.com/matttproud/golang_protobuf_extensions v1.0.1