jyseo/internal/controller/enterprise.go

package controller

import (
	"fmt"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/net/ghttp"
	"github.com/gogf/gf/v2/util/gconv"
	"jyseo/internal/service"
	"jyseo/utility"
	"strings"
)

// 0 采购单位 1-中标单位
func getRequestType(r *ghttp.Request) (int, string, error) {
	if strings.HasPrefix(r.RequestURI, "/dw/") {
		return 0, "dw", nil
	} else if strings.HasPrefix(r.RequestURI, "/qy/") {
		return 1, "qy", nil
	} else {
		return -1, "", fmt.Errorf("未知路由")
	}
}

func EnterpriseList(r *ghttp.Request) {
	pageNum := r.Get("pageNum", 1).Int() //页码
	status, flag, err := getRequestType(r)
	if err != nil {
		service.HtmlRender.NotFound(r)
		return
	}
	res, err := service.JySeoEnterprises.GetEnterprisesList(r.Context(), status, pageNum, 50, 5000)
	if err != nil {
		g.Log().Errorf(r.Context(), err.Error())
		service.HtmlRender.RenderError(r, fmt.Errorf("获取列表数据异常"))
		return
	}
	service.HtmlRender.Render(r, utility.GetCommonRenderPatch(r.Request.UserAgent(), "enterprise_list.html"),
		g.Map{
			"flag": flag,
			"list": res.List,
			//"tdk":        service.JySeoTdk.GetAreaSTypeListTdk(r.Context(), sTypeNode.Code, areaNode.Name),
			"pagination": service.GetLetterPaging(pageNum, gconv.Int(res.Total), fmt.Sprintf("/%s/p%s", flag, "%d")),
		},
	)
}

func EnterpriseDetail(r *ghttp.Request) {
	seoId := r.Get("seoId").String() //页码
	status, flag, err := getRequestType(r)
	if err != nil {
		service.HtmlRender.NotFound(r)
		return
	}
	rData := g.Map{"flag": flag}
	switch status {
	case 0: //采购单位
		baseInfo := service.JySeoEnterprises.GetEntDetailMsg(r.Context(), seoId, false)
		if baseInfo == nil || len(baseInfo) == 0 {
			service.HtmlRender.NotFound(r)
			return
		}
		rData["entDetail"] = baseInfo
		dynamicNews, _ := service.JySeoEnterprises.GetDynamicNews(r.Context(), gconv.String(baseInfo["name"]), false)
		rData["dynamicNews"] = dynamicNews
		rData["otherEnt"] = service.JySeoEnterprises.RelevanceData(r.Context(), gconv.String(baseInfo["name"]), false)
		rData["recommendedEnt"] = service.JySeoEnterprises.RecommendedData(r.Context(), gconv.String(baseInfo["name"]), gconv.String(baseInfo["area_code"]), false) //其他企业推荐
	case 1: //中标企业
		baseInfo := service.JySeoEnterprises.GetEntDetailMsg(r.Context(), seoId, true)
		if baseInfo == nil || len(baseInfo) == 0 {
			service.HtmlRender.NotFound(r)
			return
		}
		rData["entDetail"] = baseInfo                                                                                                                                      //企业基本信息
		dynamicNews, _ := service.JySeoEnterprises.GetDynamicNews(r.Context(), gconv.String(baseInfo["company_id"]), true)                                                 //企业动态
		rData["dynamicNews"] = dynamicNews                                                                                                                                 //企业中标动态
		rData["otherEnt"] = service.JySeoEnterprises.RelevanceData(r.Context(), gconv.String(baseInfo["company_id"]), true)                                                //关联采购单位
		rData["recommendedEnt"] = service.JySeoEnterprises.RecommendedData(r.Context(), gconv.String(baseInfo["company_name"]), gconv.String(baseInfo["area_code"]), true) //其他企业推荐
	}
	service.HtmlRender.Render(r, utility.GetCommonRenderPatch(r.Request.UserAgent(), "enterprise_detail.html"), rData)
}