Create & Init Project...

app/admin/openplatform/sug/dao/season_es.go

@@ -0,0 +1,93 @@
+package dao
+
+import (
+	"bytes"
+	"context"
+	"encoding/json"
+	"fmt"
+	"strconv"
+
+	"go-common/app/admin/openplatform/sug/model"
+	"go-common/library/log"
+
+	"gopkg.in/olivere/elastic.v5"
+	"io/ioutil"
+)
+
+const (
+	_seasonIndex = "%s_sug_job_season"
+	_seasonType  = "sug_job_season"
+)
+
+// GetSeason get season from es.
+func (d *Dao) GetSeason(ctx context.Context, seasonID int64) (season model.Season, err error) {
+	seasonTermQuery := elastic.NewTermQuery("id", seasonID)
+	searchResult, err := d.es.Search().Index(fmt.Sprintf(_seasonIndex, d.c.Env)).Type(_seasonType).Query(seasonTermQuery).From(0).Size(1).Timeout("1s").Do(ctx)
+	if err != nil {
+		log.Error("es search error(%v)", err)
+		return
+	}
+	if searchResult.Hits.TotalHits > 0 {
+		for _, hit := range searchResult.Hits.Hits {
+			err = json.Unmarshal(*hit.Source, &season)
+			if err != nil {
+				log.Error("json.Unmarshal err(%v)", err)
+				return season, err
+			}
+		}
+	}
+	return
+}
+
+// SeasonList search season list.
+func (d *Dao) SeasonList(ctx context.Context, params *model.SourceSearch) (list []model.Season, err error) {
+	query := elastic.NewBoolQuery()
+	matchQuery := elastic.NewMatchQuery("title", params.Keyword).Fuzziness("40")
+	sid, _ := strconv.Atoi(params.Keyword)
+	termQuery := elastic.NewMatchQuery("id", sid).Boost(100)
+	query.Should(matchQuery)
+	query.Should(termQuery)
+	searchResult, err := d.es.Search().Index(fmt.Sprintf(_seasonIndex, d.c.Env)).Type(_seasonType).Query(query).From(0).Size(10).Timeout("1s").Do(ctx)
+	if err != nil {
+		return
+	}
+	list = []model.Season{}
+	if searchResult.Hits.TotalHits > 0 {
+		var season model.Season
+		for _, hit := range searchResult.Hits.Hits {
+			err := json.Unmarshal(*hit.Source, &season)
+			if err != nil {
+				log.Error("json.Unmarshal error(%v)", err)
+				continue
+			}
+			list = append(list, season)
+		}
+	}
+	return
+}
+
+// ItemList mall items list from http.
+func (d *Dao) ItemList(ctx context.Context, params *model.SourceSearch) (itemsList []model.Items, err error) {
+	query := make(map[string]interface{})
+	query["pageNum"] = params.PageNum
+	query["pageSize"] = params.PageSize
+	query["shopId"] = 0
+	query["keyword"] = params.Keyword
+	jsonQuery, _ := json.Marshal(query)
+	resp, err := d.client.Post(d.c.URL.ItemSearch, "application/json", bytes.NewReader(jsonQuery))
+	if err != nil {
+		log.Error("Request error(%v)", err)
+		return
+	}
+	HTTPResponse := model.HTTPResponse{}
+	bodyJSON, _ := ioutil.ReadAll(resp.Body)
+	if err = json.Unmarshal(bodyJSON, &HTTPResponse); err != nil {
+		log.Error("json.Unmarshal error(%v)", err)
+	}
+	if HTTPResponse.Code != 0 {
+		log.Error("Request (%s) search error(%v)", d.c.URL.ItemSearch, err)
+		return
+	}
+	itemsList = HTTPResponse.Data.List
+	return
+}