You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
1.2 KiB
Go
60 lines
1.2 KiB
Go
package main
|
|
|
|
import (
|
|
"net/http"
|
|
"runtime"
|
|
"time"
|
|
|
|
"github.com/tal-tech/go-zero/core/logx"
|
|
"github.com/tal-tech/go-zero/core/service"
|
|
"github.com/tal-tech/go-zero/core/stat"
|
|
"github.com/tal-tech/go-zero/core/syncx"
|
|
"github.com/tal-tech/go-zero/rest"
|
|
)
|
|
|
|
func main() {
|
|
logx.Disable()
|
|
stat.SetReporter(nil)
|
|
server := rest.MustNewServer(rest.RestConf{
|
|
ServiceConf: service.ServiceConf{
|
|
Name: "breaker",
|
|
Log: logx.LogConf{
|
|
Mode: "console",
|
|
},
|
|
},
|
|
Host: "0.0.0.0",
|
|
Port: 8080,
|
|
MaxConns: 1000,
|
|
Timeout: 3000,
|
|
})
|
|
latch := syncx.NewLimit(10)
|
|
server.AddRoute(rest.Route{
|
|
Method: http.MethodGet,
|
|
Path: "/heavy",
|
|
Handler: func(w http.ResponseWriter, r *http.Request) {
|
|
if latch.TryBorrow() {
|
|
defer latch.Return()
|
|
runtime.LockOSThread()
|
|
defer runtime.UnlockOSThread()
|
|
begin := time.Now()
|
|
for {
|
|
if time.Now().Sub(begin) > time.Millisecond*50 {
|
|
break
|
|
}
|
|
}
|
|
} else {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
}
|
|
},
|
|
})
|
|
server.AddRoute(rest.Route{
|
|
Method: http.MethodGet,
|
|
Path: "/good",
|
|
Handler: func(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusOK)
|
|
},
|
|
})
|
|
defer server.Stop()
|
|
server.Start()
|
|
}
|