package logic

import (
	"context"
	"fmt"

	"demo/internal/svc"
	"demo/internal/types"
	"demo/model/es"

	"github.com/zeromicro/go-zero/core/logx"
)

type EsPressureTestLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewEsPressureTestLogic(ctx context.Context, svcCtx *svc.ServiceContext) *EsPressureTestLogic {
	return &EsPressureTestLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *EsPressureTestLogic) EsPressureTest(req *types.EsQueryName) (resp *types.Response, err error) {
	// todo: add your logic here and delete this line
	// todo: add your logic here and delete this line
	esQuery := `
		{
			"query": {
			  "match_phrase": {
				"districtName": "%s"
			  }
			  
			}
		  }
		`

	esQueryStr := fmt.Sprintf(esQuery, req.Name)
	doc, err := es.EsServiceApp.EsSearchDocWithStr(l.ctx, esQueryStr, 0, 100)
	if err != nil {
		return nil, err
	}

	return &types.Response{
		Success: true,
		Message: "success",
		Result:  doc,
	}, nil

}