|
@@ -1,76 +1,59 @@
|
|
|
package services
|
|
|
|
|
|
import (
|
|
|
- "context"
|
|
|
- "encoding/json"
|
|
|
- "fmt"
|
|
|
"log"
|
|
|
"strconv"
|
|
|
- "time"
|
|
|
-
|
|
|
- "go.mod/models"
|
|
|
|
|
|
"github.com/kataras/iris/v12"
|
|
|
"go.mod/comm"
|
|
|
"go.mod/conf"
|
|
|
- rpc "go.mod/proto"
|
|
|
+ "go.mod/dao"
|
|
|
+ "go.mod/datasource"
|
|
|
+ "go.mod/models"
|
|
|
"go.mod/web/viewmodels"
|
|
|
- "google.golang.org/grpc"
|
|
|
)
|
|
|
|
|
|
type SafeService interface {
|
|
|
- Get(id int, RpcConnect *grpc.ClientConn) (int32, string, []viewmodels.Safe)
|
|
|
- Post(bidsectionId int, code string, position string, createTime string, inspection string, RpcConnect *grpc.ClientConn) (int32, string)
|
|
|
+ Get(id int) []viewmodels.SafeList
|
|
|
+ Post(data models.CmSafe) error
|
|
|
+ Del(id int) error
|
|
|
+
|
|
|
ValidRule(ctx iris.Context) (viewmodels.Safe, error)
|
|
|
}
|
|
|
|
|
|
// //返回service操作类
|
|
|
type safeService struct {
|
|
|
- valid string
|
|
|
- validCreate string
|
|
|
+ daoSafe *dao.SafeDao
|
|
|
+ daoSafeAudit *dao.SafeAuditDao
|
|
|
+ daoSafeFile *dao.SafeFileDao
|
|
|
+ valid string
|
|
|
+ validCreate string
|
|
|
}
|
|
|
|
|
|
//创建项目用户service
|
|
|
func NewSafeService() SafeService {
|
|
|
return &safeService{
|
|
|
- valid: "/api/safe",
|
|
|
- validCreate: "/api/safe/create",
|
|
|
+ daoSafe: dao.NewSafeDao(datasource.InstanceDbMaster()),
|
|
|
+ daoSafeAudit: dao.NewSafeAuditDao(datasource.InstanceDbMaster()),
|
|
|
+ daoSafeFile: dao.NewSafeFileDao(datasource.InstanceDbMaster()),
|
|
|
+ valid: "/api/safe",
|
|
|
+ validCreate: "/api/safe/create",
|
|
|
}
|
|
|
}
|
|
|
-
|
|
|
-func (s *safeService) Get(id int, RpcConnect *grpc.ClientConn) (int32, string, []viewmodels.Safe) {
|
|
|
- // 1.结束后关闭
|
|
|
- // defer s.rpcClient.Close()
|
|
|
- // 使用连接,创建HelloService实例
|
|
|
- safeCmList := make([]models.CmSafe, 0)
|
|
|
- safeList := make([]viewmodels.Safe, 0)
|
|
|
-
|
|
|
- rpcClient := rpc.NewSafeClient(RpcConnect)
|
|
|
-
|
|
|
- ctx, cancel := context.WithTimeout(context.Background(), time.Second)
|
|
|
- defer cancel()
|
|
|
- res, err := rpcClient.GetSafeList(ctx, &rpc.SafeGetRequest{Id: int32(id)})
|
|
|
- if err != nil {
|
|
|
- log.Fatalf("could not greet: %v", err)
|
|
|
- return -1, "", safeList
|
|
|
- }
|
|
|
-
|
|
|
- err = json.Unmarshal([]byte(res.GetData()), &safeCmList)
|
|
|
- if err != nil {
|
|
|
- msg := fmt.Sprintf("Json转换出错:err=", err)
|
|
|
- log.Println("Json转换出错:err=", err)
|
|
|
- return -1, msg, safeList
|
|
|
- }
|
|
|
-
|
|
|
- for _, item := range safeCmList {
|
|
|
- safeVM := viewmodels.Safe{}
|
|
|
+func (s *safeService) Get(id int) []viewmodels.SafeList {
|
|
|
+ datalist := s.daoSafe.GetListByBid(id)
|
|
|
+ safeList := make([]viewmodels.SafeList, 0)
|
|
|
+ for _, item := range datalist {
|
|
|
+ safeVM := viewmodels.SafeList{}
|
|
|
safeVM.Code = item.Code
|
|
|
-
|
|
|
- bidId, _ := comm.AesEncrypt(strconv.Itoa(item.BidsectionId), conf.SignSecret)
|
|
|
- safeVM.BidsectionId = bidId
|
|
|
- // safeVM.CreateTime = item.CreateTime
|
|
|
+ // fmt.Println("safe_id: %d", item.Id)
|
|
|
+ auditRecord := s.daoSafeAudit.GetRecordBySid(item.Id)
|
|
|
+ safeVM.AuditName = auditRecord.Name
|
|
|
+ // bidId, _ := comm.AesEncrypt(strconv.Itoa(item.BidsectionId), conf.SignSecret)
|
|
|
+ // safeVM.BidsectionId = bidId
|
|
|
+ safeVM.CreateTime = auditRecord.CreateTime
|
|
|
safeVM.Demand = item.Demand
|
|
|
- safeVM.EndTime = item.EndTime.Format("2006-01-02 15:04:05")
|
|
|
+ // safeVM.EndTime = item.EndTime.Format("2006-01-02 15:04:05")
|
|
|
|
|
|
id, _ := comm.AesEncrypt(strconv.Itoa(item.Id), conf.SignSecret)
|
|
|
safeVM.Id = id
|
|
@@ -79,44 +62,57 @@ func (s *safeService) Get(id int, RpcConnect *grpc.ClientConn) (int32, string, [
|
|
|
safeVM.InspectionDetail = item.InspectionDetail
|
|
|
safeVM.Position = item.Position
|
|
|
safeVM.Status = item.Status
|
|
|
+ safeVM.FileCounts = s.daoSafeFile.GetCount(item.Id)
|
|
|
safeList = append(safeList, safeVM)
|
|
|
}
|
|
|
-
|
|
|
- return res.GetCode(), res.GetMsg(), safeList
|
|
|
+ return safeList
|
|
|
}
|
|
|
|
|
|
-func (s *safeService) Post(bidsectionId int, code string, position string, createTime string, inspection string, RpcConnect *grpc.ClientConn) (int32, string) {
|
|
|
- rpcClient := rpc.NewSafeClient(RpcConnect)
|
|
|
-
|
|
|
- ctx, cancel := context.WithTimeout(context.Background(), time.Second)
|
|
|
- defer cancel()
|
|
|
- res, err := rpcClient.PostSafe(ctx, &rpc.SafePostRequest{BidsectionId: int32(bidsectionId), Code: code, Position: position, CreateTime: createTime, Inspection: inspection})
|
|
|
- if err != nil {
|
|
|
- log.Fatalf("could not greet: %v", err)
|
|
|
- return -1, ""
|
|
|
+// post请求,插入单条数据
|
|
|
+func (s *safeService) Post(data models.CmSafe) error {
|
|
|
+ Inserted, err := s.daoSafe.InsertRecord(data)
|
|
|
+ if Inserted == true {
|
|
|
+ return nil
|
|
|
}
|
|
|
- return res.GetCode(), res.GetMsg()
|
|
|
+ return err
|
|
|
}
|
|
|
|
|
|
+// delete请求,删除数据
|
|
|
+func (s *safeService) Del(id int) error {
|
|
|
+ Deleted, err := s.daoSafe.DeleteRecord(id)
|
|
|
+ if Deleted == true {
|
|
|
+ return nil
|
|
|
+ }
|
|
|
+ return err
|
|
|
+}
|
|
|
func (s *safeService) ValidRule(ctx iris.Context) (viewmodels.Safe, error) {
|
|
|
safeVaild := viewmodels.Safe{}
|
|
|
- if ctx.Path() == s.valid {
|
|
|
+ if ctx.Method() == "GET" {
|
|
|
err := ctx.ReadForm(&safeVaild)
|
|
|
if err != nil {
|
|
|
- log.Println("safe-ValidRule-ReadJson转换异常, error=", err)
|
|
|
+ log.Println("safe-ValidRule-ReadForm转换异常, error=", err)
|
|
|
return safeVaild, err
|
|
|
}
|
|
|
err = safeVaild.ValidateList()
|
|
|
}
|
|
|
|
|
|
- if ctx.Path() == s.validCreate {
|
|
|
+ if ctx.Method() == "POST" {
|
|
|
err := ctx.ReadJSON(&safeVaild)
|
|
|
if err != nil {
|
|
|
- log.Println("safe-ValidRule-ReadForm转换异常, error=", err)
|
|
|
+ log.Println("safe-ValidRule-ReadJson转换异常, error=", err)
|
|
|
return safeVaild, err
|
|
|
}
|
|
|
err = safeVaild.ValidateCreate()
|
|
|
}
|
|
|
+
|
|
|
+ if ctx.Method() == "PUT" {
|
|
|
+ err := ctx.ReadForm(&safeVaild)
|
|
|
+ if err != nil {
|
|
|
+ log.Println("safe-ValidRule-ReadForm转换异常, error=", err)
|
|
|
+ return safeVaild, err
|
|
|
+ }
|
|
|
+ err = safeVaild.ValidateDelete()
|
|
|
+ }
|
|
|
return safeVaild, nil
|
|
|
|
|
|
}
|