Create & Init Project...

app/job/main/up/server/http/BUILD

@@ -0,0 +1,39 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_library",
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = [
+        "http.go",
+        "http_job.go",
+    ],
+    importpath = "go-common/app/job/main/up/server/http",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/job/main/up/conf:go_default_library",
+        "//app/job/main/up/model:go_default_library",
+        "//app/job/main/up/model/upcrmmodel:go_default_library",
+        "//app/job/main/up/service:go_default_library",
+        "//library/log:go_default_library",
+        "//library/net/http/blademaster:go_default_library",
+    ],
+)
+
+filegroup(
+    name = "package-srcs",
+    srcs = glob(["**"]),
+    tags = ["automanaged"],
+    visibility = ["//visibility:private"],
+)
+
+filegroup(
+    name = "all-srcs",
+    srcs = [":package-srcs"],
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+)

app/job/main/up/server/http/http.go

@@ -0,0 +1,50 @@
+package http
+
+import (
+	"net/http"
+
+	"go-common/app/job/main/up/conf"
+	"go-common/app/job/main/up/service"
+	"go-common/library/log"
+	bm "go-common/library/net/http/blademaster"
+)
+
+var (
+	svc *service.Service
+)
+
+// Init init
+func Init(c *conf.Config) {
+	svc = service.New(c)
+	engine := bm.DefaultServer(c.BM)
+	route(engine)
+	if err := engine.Start(); err != nil {
+		log.Error("bm Start error(%v)", err)
+		panic(err)
+	}
+}
+
+func route(e *bm.Engine) {
+	e.Ping(ping)
+	e.Register(register)
+	g := e.Group("/x/job/up")
+	{
+		g.GET("/job/run", runJob)
+		g.GET("/start")
+		g.GET("/job/warm-up", warmUp)
+		g.GET("/job/warm-up-mid", warmUpMid)
+		g.GET("/job/add-staff", addStaff)
+		g.GET("/job/delete-staff", deleteStaff)
+	}
+}
+
+func ping(c *bm.Context) {
+	if err := svc.Ping(c); err != nil {
+		log.Error("ping error(%v)", err)
+		c.AbortWithStatus(http.StatusServiceUnavailable)
+	}
+}
+
+func register(c *bm.Context) {
+	c.JSON(map[string]interface{}{}, nil)
+}

app/job/main/up/server/http/http_job.go

@@ -0,0 +1,146 @@
+package http
+
+import (
+	"context"
+	"go-common/app/job/main/up/model"
+	"go-common/app/job/main/up/model/upcrmmodel"
+	"go-common/library/log"
+	"go-common/library/net/http/blademaster"
+	"strings"
+	"time"
+)
+
+func runJob(c *blademaster.Context) {
+	var err error
+	var res interface{}
+	switch {
+	default:
+		var arg struct {
+			Date string `form:"date"`
+			Job  string `form:"job"`
+		}
+		var err = c.Bind(&arg)
+		if err != nil {
+			break
+		}
+		var date time.Time
+		if arg.Date == "" {
+			date = time.Now()
+		} else {
+			date, err = time.Parse(upcrmmodel.TimeFmtDate, arg.Date)
+			if err != nil {
+				log.Error("parse date err")
+				break
+			}
+		}
+
+		switch strings.ToLower(arg.Job) {
+		case "task":
+			svc.CheckTaskFinish(date)
+		case "due":
+			svc.CheckDateDueJob(date)
+		case "state":
+			svc.CheckStateJob(date)
+		case "tid":
+			svc.UpdateUpTidJob(date)
+		case "":
+			svc.UpdateUpTidJob(date)
+			svc.CheckStateJob(date)
+			svc.CheckDateDueJob(date)
+			svc.CheckTaskFinish(date)
+		}
+
+	}
+
+	if err != nil {
+		c.JSON(nil, err)
+	} else {
+		c.JSON(res, err)
+	}
+}
+
+func warmUp(c *blademaster.Context) {
+	var err error
+	var res interface{}
+	switch {
+	default:
+		var arg = &model.WarmUpReq{}
+		var err = c.Bind(arg)
+		if err != nil {
+			break
+		}
+
+		go func() {
+			res, err = svc.WarmUp(context.Background(), arg)
+		}()
+	}
+
+	if err != nil {
+		c.JSON(nil, err)
+	} else {
+		c.JSON(res, err)
+	}
+}
+
+func warmUpMid(c *blademaster.Context) {
+	var err error
+	var res interface{}
+	switch {
+	default:
+		var arg = &model.WarmUpReq{}
+		var err = c.Bind(arg)
+		if err != nil {
+			break
+		}
+
+		go func() {
+			res, err = svc.WarmUpMid(context.Background(), arg)
+		}()
+	}
+
+	if err != nil {
+		c.JSON(nil, err)
+	} else {
+		c.JSON(res, err)
+	}
+}
+
+func addStaff(c *blademaster.Context) {
+	var err error
+	var res interface{}
+	switch {
+	default:
+		var arg = &model.AddStaffReq{}
+		var err = c.Bind(arg)
+		if err != nil {
+			break
+		}
+		res, _ = svc.AddStaff(c, arg)
+	}
+
+	if err != nil {
+		c.JSON(nil, err)
+	} else {
+		c.JSON(res, err)
+	}
+}
+
+func deleteStaff(c *blademaster.Context) {
+	var err error
+	var res interface{}
+	switch {
+	default:
+		var arg = &model.AddStaffReq{}
+		var err = c.Bind(arg)
+		if err != nil {
+			break
+		}
+		res, _ = svc.DeleteStaff(c, arg)
+	}
+
+	if err != nil {
+		c.JSON(nil, err)
+	} else {
+		c.JSON(res, err)
+	}
+}