From 53f68b06cb63637fdc5c998f306f9f5968edcb3d Mon Sep 17 00:00:00 2001 From: CareyWong Date: Sun, 8 Mar 2020 23:36:17 +0800 Subject: [PATCH] =?UTF-8?q?=E6=8C=87=E5=AE=9A=E5=8F=82=E6=95=B0=20domain?= =?UTF-8?q?=20=E5=BF=85=E5=A1=AB=E9=A1=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.go | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/main.go b/main.go index a1ece93..742167a 100644 --- a/main.go +++ b/main.go @@ -20,7 +20,6 @@ type Response struct { const letterBytes = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ" const defaultPort int = 8002 -const defaultDomain = "s.wcc.best" const defaultExpire = 90 const redisConfig = "127.0.0.1:6379" @@ -29,10 +28,15 @@ func main() { router := gin.Default() port := flag.Int("port", defaultPort, "服务端口") - domain := flag.String("domain", defaultDomain, "短链接域名") + domain := flag.String("domain", "", "短链接域名，必填项") ttl := flag.Int("ttl", defaultExpire, "短链接有效期，单位(天)，默认90天。") flag.Parse() + if *domain == "" { + flag.Usage() + return + } + router.POST("/short", func(context *gin.Context) { res := &Response{ Code: 1,