Feat/rollback (#74)

Author: lwnmengjing

Makefile

@@ -15,9 +15,12 @@ SRC_DIR=$(GOPATH)/src
 
 .PHONY: proto
 proto:
-	protoc --go_out=. --go_opt=paths=source_relative --go-grpc_out=. --go-grpc_opt=paths=source_relative proto/*.proto
+	#protoc-go-inject-tag -I ./proto -I ${GOPATH}/src  --go_out=plugins=grpc: proto/${W}/${V}/*;
+	find proto/ -name '*.proto' -exec protoc --proto_path=$(PROTO_PATH) $(PROTO_FLAGS) --go_out=plugins=grpc:. {} \;
 
 
 .PHONY: lint
 lint:
 	golangci-lint run -v ./...
+fix-lint:
+	goimports -w .

pkg/response/actions/get_gorm.go

@@ -33,7 +33,12 @@ func NewGetGorm(m schema.Tabler, key string) *Get {
 func (e *Get) getGorm(c *gin.Context, key string) {
 	api := response.Make(c)
 	m := pkg.TablerDeepCopy(e.ModelGorm)
-	err := gormdb.DB.First(m, "id = ?", c.Param(key)).Error
+	preloads := c.QueryArray("preloads[]")
+	query := gormdb.DB.Model(m).Where("id = ?", c.Param(key))
+	for _, preload := range preloads {
+		query = query.Preload(preload)
+	}
+	err := query.First(m).Error
 	if err != nil {
 		if errors.Is(err, gorm.ErrRecordNotFound) {
 			api.Err(http.StatusNotFound)

pkg/response/model.go

@@ -92,19 +92,19 @@ func (e *response) OK(c *gin.Context, data interface{}) {
 	res := Default.Clone()
 	res.SetList(data)
 	res.SetTraceID(pkg.GenerateMsgIDFromContext(c))
+	status := http.StatusOK
 	switch c.Request.Method {
 	case http.MethodDelete:
-		res.SetCode(http.StatusNoContent)
-		c.AbortWithStatusJSON(http.StatusNoContent, data)
-		return
+		status = http.StatusNoContent
 	case http.MethodPost:
-		res.SetCode(http.StatusCreated)
-		c.AbortWithStatusJSON(http.StatusCreated, data)
+		status = http.StatusCreated
+	}
+	res.SetCode(status)
+	if data == nil {
+		c.AbortWithStatus(status)
 		return
-	default:
-		res.SetCode(http.StatusOK)
-		c.AbortWithStatusJSON(http.StatusOK, data)
 	}
+	c.AbortWithStatusJSON(status, data)
 }
 
 // PageOK page ok