feat：完善项目管理

2021-12-14 09:05:47 +08:00
parent 4e1f179af7
commit 3bb8f96d84
9 changed files with 312 additions and 4 deletions
--- a/app/api/enterprise/controller/manage/agent.go
+++ b/app/api/enterprise/controller/manage/agent.go
@ -0,0 +1,129 @@
+package manage
+
+import (
+	"SciencesServer/app/api/enterprise/model"
+	"SciencesServer/app/basic/controller"
+	model2 "SciencesServer/app/common/model"
+	"SciencesServer/app/session"
+	"errors"
+)
+
+type Agent struct {
+	*session.Enterprise
+	local string
+}
+
+type AgentHandle func(session *session.Enterprise, local string) *Agent
+
+type (
+	// AgentCompany 经纪人公司信息
+	AgentCompany struct {
+		ID string `json:"id"`
+	}
+)
+
+// Company 公司信息
+func (c *Agent) Company(name string, page, pageSize int) (*controller.ReturnPages, error) {
+	mUserAgent := model.NewUserAgent()
+
+	isExist, err := model2.FirstField(mUserAgent.UserAgent, []string{"id", "agent_id"},
+		model2.NewWhere("uid", c.UID), model2.NewWhere("invalid_status", model2.InvalidStatusForNot))
+
+	if err != nil {
+		return nil, err
+	} else if !isExist {
+		return nil, errors.New("操作错误，无权限操作")
+	}
+	mManageAgentCompany := model.NewManageAgentCompany()
+
+	var count int64
+
+	out := make([]*model.ManageAgentCompanyInfo, 0)
+
+	where := []*model2.ModelWhere{model2.NewWhere("c.agent_id", mUserAgent.ID)}
+
+	if name != "" {
+		where = append(where, model2.NewWhereLike("c.name", name))
+	}
+	if out, err = mManageAgentCompany.Company(page, pageSize, &count, where...); err != nil {
+		return nil, err
+	}
+	list := make([]*AgentCompany, 0)
+
+	for _, v := range out {
+		list = append(list, &AgentCompany{ID: v.GetEncodeID()})
+	}
+	return nil, nil
+}
+
+// Form 数据操作
+func (c *Agent) Form(id uint64, name string, industrys []string) error {
+	mUserAgent := model.NewUserAgent()
+
+	isExist, err := model2.FirstField(mUserAgent.UserAgent, []string{"id", "agent_id"},
+		model2.NewWhere("uid", c.UID), model2.NewWhere("invalid_status", model2.InvalidStatusForNot))
+
+	if err != nil {
+		return err
+	} else if !isExist {
+		return errors.New("操作错误，无权限操作")
+	}
+	mManageAgentCompany := model.NewManageAgentCompany()
+
+	if id > 0 {
+		mManageAgentCompany.ID = id
+
+		if isExist, err = model2.FirstField(mManageAgentCompany.ManageAgentCompany, []string{"id", "agent_id",
+			"name", "created_at"}); err != nil {
+			return err
+		} else if !isExist {
+			return errors.New("操作错误，无企业信息")
+		}
+		if mUserAgent.AgentID != mManageAgentCompany.AgentID {
+			return errors.New("无权限操作")
+		}
+	}
+	mManageAgentCompany.Name = name
+	mManageAgentCompany.SetIndustryAttribute(industrys)
+
+	if mManageAgentCompany.ID > 0 {
+		return model2.Updates(mManageAgentCompany.ManageAgentCompany, mManageAgentCompany.ManageAgentCompany)
+	}
+	return model2.Create(mManageAgentCompany.ManageAgentCompany)
+}
+
+// Delete 删除操作
+func (c *Agent) Delete(id uint64) error {
+	mUserAgent := model.NewUserAgent()
+
+	isExist, err := model2.FirstField(mUserAgent.UserAgent, []string{"id", "agent_id"},
+		model2.NewWhere("uid", c.UID), model2.NewWhere("invalid_status", model2.InvalidStatusForNot))
+
+	if err != nil {
+		return err
+	} else if !isExist {
+		return errors.New("操作错误，无权限操作")
+	}
+	mManageAgentCompany := model.NewManageAgentCompany()
+	mManageAgentCompany.ID = id
+
+	if isExist, err = model2.FirstField(mManageAgentCompany.ManageAgentCompany, []string{"id", "agent_id",
+		"name", "created_at"}); err != nil {
+		return err
+	} else if !isExist {
+		return errors.New("操作错误，无企业信息")
+	}
+	if mUserAgent.AgentID != mManageAgentCompany.AgentID {
+		return errors.New("无权限操作")
+	}
+	return model2.Delete(mManageAgentCompany.ManageAgentCompany)
+}
+
+func NewAgent() AgentHandle {
+	return func(session *session.Enterprise, local string) *Agent {
+		return &Agent{
+			Enterprise: session,
+			local:      local,
+		}
+	}
+}