elasticutil.go 42 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501
  1. package elastic
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. "log"
  6. "net/url"
  7. "reflect"
  8. "runtime"
  9. "strconv"
  10. "strings"
  11. "sync"
  12. "time"
  13. es "app.yhyue.com/moapp/esv1/gopkg.in/olivere/elastic.v1"
  14. )
  15. //检索库服务地址
  16. var addrs []string
  17. var LocCity = map[string]string{}
  18. var SIZE = 30
  19. const (
  20. QStr = `{"query":{"bool":{"must":[$and],"must_not":[],
  21. "should":[$or],"minimum_should_match" : 1}}}`
  22. )
  23. var pool chan *es.Client
  24. var ntimeout int
  25. var syncPool sync.Pool
  26. //初始化全文检索
  27. func InitElastic(addr string) {
  28. InitElasticSize(addr, SIZE)
  29. }
  30. //自定义HttpClient
  31. /**
  32. var httpclient = &http.Client{Transport: &http.Transport{
  33. Dial: func(netw, addr string) (net.Conn, error) {
  34. deadline := time.Now().Add(5000 * time.Millisecond)
  35. c, err := net.DialTimeout(netw, addr, 10000*time.Millisecond)
  36. if err != nil {
  37. return nil, err
  38. }
  39. tcp_conn := c.(*net.TCPConn)
  40. tcp_conn.SetKeepAlive(false)
  41. tcp_conn.SetDeadline(deadline)
  42. return tcp_conn, nil
  43. },
  44. DisableKeepAlives: true, //不保持,这样才能释放
  45. }}
  46. **/
  47. //var op = es.SetHttpClient(httpclient)
  48. var poolsize = int32(20)
  49. //n倍的池
  50. func InitElasticSize(addr string, size int) {
  51. poolsize = int32(3 * size)
  52. pool = make(chan *es.Client, poolsize)
  53. for _, s := range strings.Split(addr, ",") {
  54. addrs = append(addrs, s)
  55. }
  56. for i := 0; i < size; i++ {
  57. client, _ := es.NewClient(es.SetURL(addrs...), es.SetMaxRetries(2), es.SetSniff(false))
  58. pool <- client
  59. }
  60. }
  61. //关闭连接
  62. func DestoryEsConn(client *es.Client) {
  63. select {
  64. case pool <- client:
  65. break
  66. case <-time.After(time.Second * 1):
  67. if client != nil {
  68. client.Stop()
  69. }
  70. client = nil
  71. }
  72. }
  73. var (
  74. lastTime = int64(0)
  75. lastTimeLock = &sync.Mutex{}
  76. )
  77. //获取连接
  78. func GetEsConn() *es.Client {
  79. select {
  80. case c := <-pool:
  81. if c == nil || !c.IsRunning() {
  82. log.Println("new esclient.", len(pool))
  83. client, err := es.NewClient(es.SetURL(addrs...),
  84. es.SetMaxRetries(2), es.SetSniff(false))
  85. if err == nil && client.IsRunning() {
  86. return client
  87. }
  88. }
  89. return c
  90. case <-time.After(time.Second * 4):
  91. //超时
  92. ntimeout++
  93. lastTimeLock.Lock()
  94. defer lastTimeLock.Unlock()
  95. //12秒后允许创建链接
  96. c := time.Now().Unix() - lastTime
  97. if c > 12 {
  98. lastTime = time.Now().Unix()
  99. log.Println("add client..", len(pool))
  100. c, _ := es.NewClient(es.SetURL(addrs...), es.SetMaxRetries(2), es.SetSniff(false))
  101. go func() {
  102. for i := 0; i < 2; i++ {
  103. client, _ := es.NewClient(es.SetURL(addrs...), es.SetMaxRetries(2), es.SetSniff(false))
  104. pool <- client
  105. }
  106. }()
  107. return c
  108. }
  109. return nil
  110. }
  111. }
  112. //保存对象
  113. func Save(index, itype string, obj interface{}) bool {
  114. client := GetEsConn()
  115. defer DestoryEsConn(client)
  116. defer func() {
  117. if r := recover(); r != nil {
  118. log.Println("[E]", r)
  119. for skip := 1; ; skip++ {
  120. _, file, line, ok := runtime.Caller(skip)
  121. if !ok {
  122. break
  123. }
  124. go log.Printf("%v,%v\n", file, line)
  125. }
  126. }
  127. }()
  128. _, err := client.Index().Index(index).Type(itype).BodyJson(objToMap(obj)).Do()
  129. if err != nil {
  130. log.Println("保存到ES出错", err.Error(), obj)
  131. return false
  132. } else {
  133. return true
  134. }
  135. }
  136. //通用查询
  137. //{"query": {"bool":{"must":[{"query_string":{"default_field":"name","query":"*"}}]}}}
  138. //{"query":{"bool":{"must":{"match":{"content":{"query":"fulltextsearch","operator":"and"}}},"should":[{"match":{"content":{"query":"Elasticsearch","boost":3}}},{"match":{"content":{"query":"Lucene","boost":2}}}]}}}
  139. //prefix
  140. //{"query":{"match":{"title":{"query":"brownfox","operator":"and"}}}} //默认为or
  141. //{"query":{"multi_match":{"query":"PolandStreetW1V","type":"most_fields","fields":["*_street","city^2","country","postcode"]}}}
  142. //{"query":{"wildcard":{"postcode":"W?F*HW"}}}
  143. //{"query":{"regexp":{"postcode":"W[0-9].+"}}}
  144. //{"query":{"filtered":{"filter":{"range":{"price":{"gte":10000}}}}},"aggs":{"single_avg_price":{"avg":{"field":"price"}}}}
  145. //{"query":{"match":{"make":"ford"}},"aggs":{"colors":{"terms":{"field":"color"}}}}//查fork有几种颜色
  146. //过滤器不会计算相关度的得分,所以它们在计算上更快一些
  147. //{"query":{"filtered":{"query":{"match_all":{}},"filter":{"range":{"balance":{"gte":20000,"lte":30000}}}}}}
  148. //{"query":{"match_all":{}},"from":10,"size":10,"_source":["account_number","balance"],"sort":{"balance":{"order":"desc"}}}
  149. //{"query":{"match_phrase":{"address":"milllane"}}}和match不同会去匹配整个短语,相当于must[]
  150. func Get(index, itype, query string) *[]map[string]interface{} {
  151. //log.Println("query -- ", query)
  152. client := GetEsConn()
  153. defer func() {
  154. go DestoryEsConn(client)
  155. }()
  156. var res []map[string]interface{}
  157. if client != nil {
  158. defer func() {
  159. if r := recover(); r != nil {
  160. log.Println("[E]", r)
  161. for skip := 1; ; skip++ {
  162. _, file, line, ok := runtime.Caller(skip)
  163. if !ok {
  164. break
  165. }
  166. go log.Printf("%v,%v\n", file, line)
  167. }
  168. }
  169. }()
  170. searchResult, err := client.Search().Index(index).Type(itype).Source(query).Do()
  171. if err != nil {
  172. log.Println("从ES查询出错", err.Error())
  173. return nil
  174. }
  175. if searchResult.Hits != nil {
  176. resNum := len(searchResult.Hits.Hits)
  177. if resNum <= 5000 {
  178. res = make([]map[string]interface{}, resNum)
  179. for i, hit := range searchResult.Hits.Hits {
  180. //d := json.NewDecoder(bytes.NewBuffer(*hit.Source))
  181. //d.UseNumber()
  182. //d.Decode(&res[i])
  183. parseErr := json.Unmarshal(*hit.Source, &res[i])
  184. if parseErr == nil && hit.Highlight != nil && res[i] != nil {
  185. res[i]["highlight"] = map[string][]string(hit.Highlight)
  186. }
  187. }
  188. } else {
  189. log.Println("查询结果太多,查询到:", resNum, "条")
  190. }
  191. }
  192. }
  193. return &res
  194. }
  195. // GetNew 返回查询结果数量总数
  196. func GetNew(index, itype, query string) (*[]map[string]interface{}, int64) {
  197. //log.Println("query -- ", query)
  198. client := GetEsConn()
  199. defer func() {
  200. go DestoryEsConn(client)
  201. }()
  202. var (
  203. res []map[string]interface{}
  204. count int64
  205. )
  206. if client != nil {
  207. defer func() {
  208. if r := recover(); r != nil {
  209. log.Println("[E]", r)
  210. for skip := 1; ; skip++ {
  211. _, file, line, ok := runtime.Caller(skip)
  212. if !ok {
  213. break
  214. }
  215. go log.Printf("%v,%v\n", file, line)
  216. }
  217. }
  218. }()
  219. searchResult, err := client.Search().Index(index).Type(itype).Source(query).Do()
  220. if err != nil {
  221. log.Println("从ES查询出错", err.Error())
  222. return nil, count
  223. }
  224. if searchResult.Hits != nil {
  225. count = searchResult.Hits.TotalHits
  226. resNum := len(searchResult.Hits.Hits)
  227. if resNum > 5000 {
  228. log.Println("查询结果太多,查询到:", resNum, "条")
  229. } else {
  230. res = make([]map[string]interface{}, resNum)
  231. for i, hit := range searchResult.Hits.Hits {
  232. parseErr := json.Unmarshal(*hit.Source, &res[i])
  233. if parseErr == nil && hit.Highlight != nil && res[i] != nil {
  234. res[i]["highlight"] = map[string][]string(hit.Highlight)
  235. }
  236. }
  237. }
  238. }
  239. }
  240. return &res, count
  241. }
  242. func GetOA(index, itype, query string) (*[]map[string]interface{}, int) {
  243. //log.Println("query -- ", query)
  244. client := GetEsConn()
  245. defer func() {
  246. go DestoryEsConn(client)
  247. }()
  248. var res []map[string]interface{}
  249. var resNum int
  250. if client != nil {
  251. defer func() {
  252. if r := recover(); r != nil {
  253. log.Println("[E]", r)
  254. for skip := 1; ; skip++ {
  255. _, file, line, ok := runtime.Caller(skip)
  256. if !ok {
  257. break
  258. }
  259. go log.Printf("%v,%v\n", file, line)
  260. }
  261. }
  262. }()
  263. searchResult, err := client.Search().Index(index).Type(itype).Source(query).Do()
  264. if err != nil {
  265. log.Println("从ES查询出错", err.Error())
  266. return nil, 0
  267. }
  268. if searchResult.Hits != nil {
  269. resNum = len(searchResult.Hits.Hits)
  270. if resNum <= 5000 {
  271. res = make([]map[string]interface{}, resNum)
  272. for i, hit := range searchResult.Hits.Hits {
  273. //d := json.NewDecoder(bytes.NewBuffer(*hit.Source))
  274. //d.UseNumber()
  275. //d.Decode(&res[i])
  276. parseErr := json.Unmarshal(*hit.Source, &res[i])
  277. if parseErr == nil && hit.Highlight != nil && res[i] != nil {
  278. res[i]["highlight"] = map[string][]string(hit.Highlight)
  279. }
  280. }
  281. } else {
  282. log.Println("查询结果太多,查询到:", resNum, "条")
  283. }
  284. }
  285. }
  286. return &res, resNum
  287. }
  288. func GetNoLimit(index, itype, query string) *[]map[string]interface{} {
  289. //log.Println("query -- ", query)
  290. client := GetEsConn()
  291. defer DestoryEsConn(client)
  292. var res []map[string]interface{}
  293. if client != nil {
  294. defer func() {
  295. if r := recover(); r != nil {
  296. log.Println("[E]", r)
  297. for skip := 1; ; skip++ {
  298. _, file, line, ok := runtime.Caller(skip)
  299. if !ok {
  300. break
  301. }
  302. go log.Printf("%v,%v\n", file, line)
  303. }
  304. }
  305. }()
  306. searchResult, err := client.Search().Index(index).Type(itype).Source(query).Do()
  307. if err != nil {
  308. log.Println("从ES查询出错", err.Error())
  309. return nil
  310. }
  311. if searchResult.Hits != nil {
  312. resNum := len(searchResult.Hits.Hits)
  313. res = make([]map[string]interface{}, resNum)
  314. for i, hit := range searchResult.Hits.Hits {
  315. json.Unmarshal(*hit.Source, &res[i])
  316. }
  317. }
  318. }
  319. return &res
  320. }
  321. //分页查询
  322. //{"name":"张三","$and":[{"age":{"$gt":10}},{"age":{"$lte":20}}]}
  323. //fields直接是 `"_id","title"`
  324. func GetPage(index, itype, query, order, field string, start, limit int) *[]map[string]interface{} {
  325. return Get(index, itype, MakeQuery(query, order, field, start, limit))
  326. }
  327. //openapi
  328. func GetOAPage(index, itype, query, order, field string, start, limit int) (*[]map[string]interface{}, int) {
  329. return GetOA(index, itype, MakeQuery(query, order, field, start, limit))
  330. }
  331. var SR = strings.Replace
  332. func MakeQuery(query, order, fileds string, start, limit int) string {
  333. res := AnalyQuery(query, "", QStr)
  334. if len(res) > 10 {
  335. if strings.Contains(res, "###剑鱼###") {
  336. res = strings.ReplaceAll(res, "###剑鱼###", "\\\"")
  337. }
  338. res = SR(SR(SR(SR(res, ",$and", "", -1), "$and", "", -1), ",$or", "", -1), "$or", "", -1)
  339. if len(fileds) > 0 {
  340. //"_source":["account_number","balance"]
  341. res = res[:len(res)-1] + `,"_source":[` + fileds + "]}"
  342. }
  343. //{"name":-1,"age":1}
  344. if len(order) > 0 {
  345. res = res[:len(res)-1] + `,"sort":[` + SR(SR(SR(SR(order, ",", "},{", -1), " ", "", -1), ":-1", `:"desc"`, -1), ":1", `:"asc"`, -1) + `]}`
  346. }
  347. if start > -1 {
  348. res = res[:len(res)-1] + `,"from":` + strconv.Itoa(start) + `,"size":` + strconv.Itoa(limit) + "}"
  349. }
  350. return res
  351. }
  352. return ""
  353. }
  354. //{"name":"aaa"}
  355. func AnalyQuery(query interface{}, parent string, result string) string {
  356. m := make(map[string]interface{})
  357. if q1, ok := query.(string); ok {
  358. json.Unmarshal([]byte(q1), &m)
  359. } else if q2, ok2 := query.(map[string]interface{}); ok2 {
  360. m = q2
  361. }
  362. if len(parent) == 0 {
  363. for k, v := range m {
  364. if k == "$and" || k == "$or" {
  365. temps := ""
  366. if map1, ok := v.([]interface{}); ok {
  367. for i := 0; i < len(map1); i++ {
  368. temps += "," + AnalyQuery(map1[i], k, "")
  369. }
  370. }
  371. if len(temps) > 0 {
  372. temps = temps[1:]
  373. }
  374. result = SR(result, k, temps+","+k, 1)
  375. } else {
  376. switch reflect.TypeOf(v).String() {
  377. case "string":
  378. if strings.Index(k, "TERM_") == 0 {
  379. result = SR(result, "$and", `{"term":{"`+SR(k, "TERM_", "", 1)+`":"`+fmt.Sprintf("%v", v)+`"}},$and`, 1)
  380. } else {
  381. result = SR(result, "$and", `{"query_string":{"default_field":"`+k+`","query":"`+fmt.Sprintf("%v", v)+`"}},$and`, 1)
  382. }
  383. case "int", "int8", "int32", "int64", "float32", "float64":
  384. if strings.Index(k, "TERM_") == 0 {
  385. result = SR(result, "$and", `{"term":{"`+SR(k, "TERM_", "", 1)+`":`+fmt.Sprintf("%v", v)+`}},$and`, 1)
  386. } else {
  387. result = SR(result, "$and", `{"query_string":{"default_field":"`+k+`","query":`+fmt.Sprintf("%v", v)+`}},$and`, 1)
  388. }
  389. default:
  390. result = SR(result, "$and", AnalyQuery(v, k, "")+",$and", 1)
  391. }
  392. }
  393. }
  394. return result
  395. } else {
  396. for k, v := range m {
  397. if k == "$in" {
  398. s := ""
  399. if map1, ok := v.([]interface{}); ok {
  400. for i := 0; i < len(map1); i++ {
  401. s += "," + `"` + fmt.Sprintf("%v", map1[i]) + `"`
  402. }
  403. }
  404. if len(s) > 0 {
  405. s = s[1:]
  406. }
  407. return `{"terms":{"` + parent + `":[` + s + `]}}`
  408. } else if strings.Contains(k, "$lt") || strings.Contains(k, "$gt") {
  409. return `{"range":{"` + parent + `":{"` + SR(k, "$", "", 1) + `":` + fmt.Sprintf("%v", v) + `}}}`
  410. } else {
  411. switch reflect.TypeOf(v).String() {
  412. case "string":
  413. if strings.Index(k, "TERM_") == 0 {
  414. return `{"term":{"` + SR(k, "TERM_", "", 1) + `":"` + fmt.Sprintf("%v", v) + `"}}`
  415. } else {
  416. return `{"query_string":{"default_field":"` + k + `","query":"` + fmt.Sprintf("%v", v) + `"}}`
  417. }
  418. case "int", "int8", "int32", "int64", "float32", "float64":
  419. if strings.Index(k, "TERM_") == 0 {
  420. return `{"term":{"` + SR(k, "TERM_", "", 1) + `":` + fmt.Sprintf("%v", v) + `}}`
  421. } else {
  422. return `{"query_string":{"default_field":"` + k + `","query":` + fmt.Sprintf("%v", v) + `}}`
  423. }
  424. default:
  425. return AnalyQuery(v, k, result)
  426. }
  427. }
  428. }
  429. }
  430. return result
  431. }
  432. func GetByIdField(index, itype, id, fields string) *map[string]interface{} {
  433. client := GetEsConn()
  434. defer DestoryEsConn(client)
  435. if client != nil {
  436. defer func() {
  437. if r := recover(); r != nil {
  438. log.Println("[E]", r)
  439. for skip := 1; ; skip++ {
  440. _, file, line, ok := runtime.Caller(skip)
  441. if !ok {
  442. break
  443. }
  444. go log.Printf("%v,%v\n", file, line)
  445. }
  446. }
  447. }()
  448. query := `{"query":{"term":{"_id":"` + id + `"}}`
  449. if len(fields) > 0 {
  450. query = query + `,"_source":[` + fields + `]`
  451. }
  452. query = query + "}"
  453. searchResult, err := client.Search().Index(index).Type(itype).Source(query).Do()
  454. if err != nil {
  455. log.Println("从ES查询出错", err.Error())
  456. return nil
  457. }
  458. var res map[string]interface{}
  459. if searchResult.Hits != nil {
  460. resNum := len(searchResult.Hits.Hits)
  461. if resNum == 1 {
  462. res = make(map[string]interface{})
  463. for _, hit := range searchResult.Hits.Hits {
  464. json.Unmarshal(*hit.Source, &res)
  465. }
  466. return &res
  467. }
  468. }
  469. }
  470. return nil
  471. }
  472. //根据id来查询文档
  473. func GetById(index, itype string, ids ...string) *[]map[string]interface{} {
  474. client := GetEsConn()
  475. defer DestoryEsConn(client)
  476. var res []map[string]interface{}
  477. if client != nil {
  478. defer func() {
  479. if r := recover(); r != nil {
  480. log.Println("[E]", r)
  481. for skip := 1; ; skip++ {
  482. _, file, line, ok := runtime.Caller(skip)
  483. if !ok {
  484. break
  485. }
  486. go log.Printf("%v,%v\n", file, line)
  487. }
  488. }
  489. }()
  490. query := es.NewIdsQuery().Ids(ids...)
  491. searchResult, err := client.Search().Index(index).Type(itype).Query(&query).Do()
  492. if err != nil {
  493. log.Println("从ES查询出错", err.Error())
  494. return nil
  495. }
  496. if searchResult.Hits != nil {
  497. resNum := len(searchResult.Hits.Hits)
  498. if resNum <= 5000 {
  499. res = make([]map[string]interface{}, resNum)
  500. for i, hit := range searchResult.Hits.Hits {
  501. json.Unmarshal(*hit.Source, &res[i])
  502. }
  503. } else {
  504. log.Println("查询结果太多,查询到:", resNum, "条")
  505. }
  506. }
  507. }
  508. return &res
  509. }
  510. //删除某个索引,根据查询
  511. func Del(index, itype string, query interface{}) bool {
  512. client := GetEsConn()
  513. defer DestoryEsConn(client)
  514. b := false
  515. if client != nil {
  516. defer func() {
  517. if r := recover(); r != nil {
  518. log.Println("[E]", r)
  519. for skip := 1; ; skip++ {
  520. _, file, line, ok := runtime.Caller(skip)
  521. if !ok {
  522. break
  523. }
  524. go log.Printf("%v,%v\n", file, line)
  525. }
  526. }
  527. }()
  528. var err error
  529. if qs, ok := query.(string); ok {
  530. temp := es.BoolQuery{
  531. QueryStrings: qs,
  532. }
  533. _, err = client.DeleteByQuery().Index(index).Type(itype).Query(temp).Do()
  534. } else if qi, ok2 := query.(es.Query); ok2 {
  535. _, err = client.DeleteByQuery().Index(index).Type(itype).Query(qi).Do()
  536. }
  537. if err != nil {
  538. log.Println("删除索引出错:", err.Error())
  539. } else {
  540. b = true
  541. }
  542. }
  543. return b
  544. }
  545. //根据语句更新对象
  546. func Update(index, itype, id string, updateStr string) bool {
  547. client := GetEsConn()
  548. defer DestoryEsConn(client)
  549. b := false
  550. if client != nil {
  551. defer func() {
  552. if r := recover(); r != nil {
  553. log.Println("[E]", r)
  554. for skip := 1; ; skip++ {
  555. _, file, line, ok := runtime.Caller(skip)
  556. if !ok {
  557. break
  558. }
  559. go log.Printf("%v,%v\n", file, line)
  560. }
  561. }
  562. }()
  563. var err error
  564. _, err = client.Update().Index(index).Type(itype).Id(id).Script(updateStr).ScriptLang("groovy").Do()
  565. if err != nil {
  566. log.Println("更新检索出错:", err.Error())
  567. } else {
  568. b = true
  569. }
  570. }
  571. return b
  572. }
  573. func BulkUpdate(index, itype string, ids []string, updateStr string) {
  574. client := GetEsConn()
  575. defer DestoryEsConn(client)
  576. if client != nil {
  577. defer func() {
  578. if r := recover(); r != nil {
  579. log.Println("[E]", r)
  580. for skip := 1; ; skip++ {
  581. _, file, line, ok := runtime.Caller(skip)
  582. if !ok {
  583. break
  584. }
  585. go log.Printf("%v,%v\n", file, line)
  586. }
  587. }
  588. }()
  589. for _, id := range ids {
  590. _, err := client.Update().Index(index).Type(itype).Id(id).Script(updateStr).ScriptLang("groovy").Do()
  591. if err != nil {
  592. log.Println("更新检索出错:", err.Error())
  593. }
  594. }
  595. }
  596. }
  597. //根据id删除索引对象
  598. func DelById(index, itype, id string) bool {
  599. client := GetEsConn()
  600. defer DestoryEsConn(client)
  601. b := false
  602. if client != nil {
  603. defer func() {
  604. if r := recover(); r != nil {
  605. log.Println("[E]", r)
  606. for skip := 1; ; skip++ {
  607. _, file, line, ok := runtime.Caller(skip)
  608. if !ok {
  609. break
  610. }
  611. go log.Printf("%v,%v\n", file, line)
  612. }
  613. }
  614. }()
  615. var err error
  616. _, err = client.Delete().Index(index).Type(itype).Id(id).Do()
  617. if err != nil {
  618. log.Println("更新检索出错:", err.Error())
  619. } else {
  620. b = true
  621. }
  622. }
  623. return b
  624. }
  625. //先删除后增
  626. func UpdateNewDoc(index, itype string, obj ...interface{}) bool {
  627. client := GetEsConn()
  628. defer DestoryEsConn(client)
  629. b := false
  630. if client != nil {
  631. defer func() {
  632. if r := recover(); r != nil {
  633. log.Println("[E]", r)
  634. for skip := 1; ; skip++ {
  635. _, file, line, ok := runtime.Caller(skip)
  636. if !ok {
  637. break
  638. }
  639. go log.Printf("%v,%v\n", file, line)
  640. }
  641. }
  642. }()
  643. var err error
  644. for _, v := range obj {
  645. tempObj := objToMap(v)
  646. id := fmt.Sprintf("%v", (*tempObj)["_id"])
  647. client.Delete().Index(index).Type(itype).Id(id).Do()
  648. _, err = client.Index().Index(index).Type(itype).BodyJson(tempObj).Do()
  649. if err != nil {
  650. log.Println("保存到ES出错", err.Error())
  651. } else {
  652. b = true
  653. }
  654. }
  655. }
  656. return b
  657. }
  658. //把地市代码转为地市
  659. func getLoc(code string, res *map[string]string) (loc string) {
  660. switch len(code) {
  661. case 6:
  662. loc = (*res)[code[:2]] + " " + (*res)[code[:4]] + " " + (*res)[code]
  663. break
  664. case 4:
  665. loc = (*res)[code[:2]] + " " + (*res)[code]
  666. break
  667. case 2:
  668. loc = (*res)[code]
  669. break
  670. }
  671. return
  672. }
  673. func ConverData(ent *map[string]interface{}) map[string]interface{} {
  674. tmp := *ent
  675. id64, _ := tmp["ID"].(int64)
  676. ids := fmt.Sprintf("%d", id64)
  677. tmp2 := make(map[string]interface{})
  678. tmp2["ID"] = ids
  679. tmp2["_id"] = tmp["_id"]
  680. tmp2["Area"] = tmp["Area"]
  681. tmp2["LeRep"] = tmp["LeRep"]
  682. tmp2["RegNo"] = tmp["RegNo"]
  683. tmp2["EntType"] = tmp["EntType"]
  684. tmp2["EntName"] = tmp["EntName"]
  685. tmp2["EntTypeName"] = tmp["EntTypeName"]
  686. tmp2["Dom"] = tmp["Dom"]
  687. tmp2["EstDate"] = tmp["EstDate"]
  688. tmp2["OpStateName"] = tmp["OpStateName"]
  689. tmp2["OpScope"] = tmp["OpScope"]
  690. tmp2["OpState"] = tmp["OpState"]
  691. tmp2["s_submitid"] = tmp["s_submitid"]
  692. tmp2["l_submittime"] = tmp["l_submittime"]
  693. tmp2["s_submitname"] = tmp["s_submitname"]
  694. tmp2["RegCapCurName"] = tmp["RegCapCurName"]
  695. //增加营业状态排序
  696. if tmp2["OpState"] == "06" {
  697. tmp2["OpSint"] = true
  698. } else {
  699. tmp2["OpSint"] = false
  700. }
  701. tmp2["OpLocDistrict"] = tmp["OpLocDistrict"]
  702. //增加代码转名称
  703. tmpLoc, _ := tmp["OpLocDistrict"].(string)
  704. tmp2["OpLocDistrictName"] = getLoc(tmpLoc, &LocCity)
  705. tmp2["RecCap"] = tmp["RecCap"]
  706. tmp2["RegCap"] = tmp["RegCap"]
  707. tmp2["IndustryPhy"] = tmp["IndustryPhy"]
  708. tmp2["IndustryPhyName"] = tmp["IndustryPhyName"]
  709. tmp2["RegOrg"] = tmp["RegOrg"]
  710. tmp2["RegOrgName"] = tmp["RegOrgName"]
  711. tmp2["Tel"] = tmp["Tel"]
  712. tmp2["CompForm"] = tmp["CompForm"]
  713. tmp2["CompFormName"] = tmp["CompFormName"]
  714. //增加异常名录标记 Ycml可能是bool也可能是string
  715. Ycmlb, _ := tmp["Ycml"].(bool)
  716. Ycmls, _ := tmp["Ycml"].(string)
  717. if Ycmlb || Ycmls == "1" {
  718. tmp2["Ycml"] = true
  719. } else {
  720. tmp2["Ycml"] = false
  721. }
  722. //增加年报联系信息
  723. if tmp["Nb_email"] != nil {
  724. tmp2["Nb_email"] = tmp["Nb_email"]
  725. }
  726. if tmp["Nb_tel"] != nil {
  727. tmp2["Nb_tel"] = tmp["Nb_tel"]
  728. }
  729. if tmp["Nb_addr"] != nil {
  730. tmp2["Nb_addr"] = tmp["Nb_addr"]
  731. }
  732. s_synopsis := tmp["s_synopsis"]
  733. if s_synopsis == nil {
  734. s_synopsis = ""
  735. }
  736. tmp2["s_synopsis"] = s_synopsis //企业简介
  737. //股东
  738. stock := getStock(tmp["investor"])
  739. tmp2["stock"] = stock
  740. tmp2["LegCerNO"] = tmp["LegCerNO"]
  741. if tmp["s_microwebsite"] != nil {
  742. tmp2["s_microwebsite"] = tmp["s_microwebsite"]
  743. }
  744. tmp2["SourceType"] = tmp["SourceType"] //数据来源
  745. s_servicenames := tmp["s_servicenames"]
  746. if s_servicenames == nil {
  747. s_servicenames = ""
  748. }
  749. tmp2["s_servicenames"] = s_servicenames //服务名称
  750. s_action := tmp["s_action"]
  751. if s_action == nil {
  752. s_action = "N"
  753. }
  754. tmp2["s_action"] = s_action
  755. tmp2["s_persion"] = tmp["s_persion"]
  756. tmp2["s_mobile"] = tmp["s_mobile"]
  757. tmp2["s_enturl"] = tmp["s_enturl"]
  758. tmp2["s_weixin"] = tmp["s_weixin"]
  759. tmp2["s_avatar"] = tmp["s_avatar"]
  760. return tmp2
  761. }
  762. func getStock(obj interface{}) string {
  763. stock := ""
  764. if ns, ok := obj.([]interface{}); ok {
  765. stock = " "
  766. for _, ns1 := range ns {
  767. if nn, ok1 := ns1.(map[string]interface{}); ok1 {
  768. tmp := fmt.Sprintf("%s", nn["Inv"])
  769. if strings.Index(stock, tmp) < 0 {
  770. stock += tmp + " "
  771. }
  772. }
  773. }
  774. }
  775. return stock
  776. }
  777. func BulkSave(index, itype string, obj *[]map[string]interface{}, isDelBefore bool) {
  778. client := GetEsConn()
  779. defer DestoryEsConn(client)
  780. if client != nil {
  781. defer func() {
  782. if r := recover(); r != nil {
  783. log.Println("[E]", r)
  784. for skip := 1; ; skip++ {
  785. _, file, line, ok := runtime.Caller(skip)
  786. if !ok {
  787. break
  788. }
  789. go log.Printf("%v,%v\n", file, line)
  790. }
  791. }
  792. }()
  793. req := client.Bulk()
  794. for _, v := range *obj {
  795. if isDelBefore {
  796. req = req.Add(es.NewBulkDeleteRequest().Index(index).Type(itype).Id(fmt.Sprintf("%v", v["_id"])))
  797. }
  798. req = req.Add(es.NewBulkIndexRequest().Index(index).Type(itype).Doc(v))
  799. }
  800. _, err := req.Do()
  801. if err != nil {
  802. log.Println("批量保存到ES出错", err.Error())
  803. }
  804. }
  805. }
  806. func Count(index, itype string, query interface{}) int64 {
  807. client := GetEsConn()
  808. defer DestoryEsConn(client)
  809. if client != nil {
  810. defer func() {
  811. if r := recover(); r != nil {
  812. log.Println("[E]", r)
  813. for skip := 1; ; skip++ {
  814. _, file, line, ok := runtime.Caller(skip)
  815. if !ok {
  816. break
  817. }
  818. go log.Printf("%v,%v\n", file, line)
  819. }
  820. }
  821. }()
  822. var qq es.Query
  823. if qs, ok := query.(string); ok {
  824. temp := es.BoolQuery{
  825. QueryStrings: qs,
  826. }
  827. qq = temp
  828. } else if qi, ok2 := query.(es.Query); ok2 {
  829. qq = qi
  830. }
  831. n, err := client.Count(index).Type(itype).Query(qq).Do()
  832. if err != nil {
  833. log.Println("统计出错", err.Error())
  834. }
  835. return n
  836. }
  837. return 0
  838. }
  839. //ngram精确查询
  840. /*
  841. {
  842. "query": {
  843. "bool": {
  844. "should": [
  845. {
  846. "bool":{
  847. "must":[
  848. { "multi_match": {
  849. "query": "智能",
  850. "type": "phrase",
  851. "fields": [
  852. "title"
  853. ],
  854. "analyzer": "my_ngram"
  855. }
  856. },{
  857. "multi_match": {
  858. "query": "机器",
  859. "type": "phrase",
  860. "fields": [
  861. "title"
  862. ],
  863. "analyzer": "my_ngram"
  864. }
  865. },{
  866. "multi_match": {
  867. "query": "2016",
  868. "type": "phrase",
  869. "fields": [
  870. "title"
  871. ],
  872. "analyzer": "my_ngram"
  873. }
  874. }
  875. ]
  876. }
  877. },
  878. {
  879. "bool":{
  880. "must":[
  881. { "multi_match": {
  882. "query": "河南",
  883. "type": "phrase",
  884. "fields": [
  885. "title"
  886. ],
  887. "analyzer": "my_ngram"
  888. }
  889. },{
  890. "multi_match": {
  891. "query": "工商",
  892. "type": "phrase",
  893. "fields": [
  894. "title"
  895. ],
  896. "analyzer": "my_ngram"
  897. }
  898. },{
  899. "multi_match": {
  900. "query": "2016",
  901. "type": "phrase",
  902. "fields": [
  903. "title"
  904. ],
  905. "analyzer": "my_ngram"
  906. }
  907. }
  908. ]
  909. }
  910. }
  911. ],"minimum_should_match": 1
  912. }
  913. },
  914. "_source": [
  915. "_id",
  916. "title"
  917. ],
  918. "from": 0,
  919. "size": 10,
  920. "sort": [{
  921. "publishtime": "desc"
  922. }]
  923. }
  924. */
  925. //"2016+智能+办公,"河南+工商"
  926. //["2016+智能+办公","河南+工商"]
  927. //QStr = `{"query":{"bool":{should":[$or],"minimum_should_match" : 1}}}`
  928. //{"bool":{"must":[]}}
  929. //{"multi_match": {"query": "$word","type": "phrase", "fields": [$field],"analyzer": "my_ngram"}}
  930. //"highlight": {"pre_tags": [""],"post_tags": [""],"fields": {"detail": {"fragment_size": 1,"number_of_fragments": 1},"title": {"fragment_size": 1,"number_of_fragments": 1}}}
  931. const (
  932. //此处最后少一个},正好NgramStr取[1:]多一个}
  933. FilterQuery = `{"query": {"filtered": {"filter": {"bool": {"must": [%s]}},%s}}`
  934. NgramStr = `{"query":{"bool":{"must":[%s],"should":[%s],"minimum_should_match": 1}}}`
  935. NgramMust = `{"bool":{"must":[%s]}}`
  936. NgramMustAndNot = `{"bool":{"must":[%s],"must_not":[%s]}}`
  937. minq = `{"multi_match": {"query": "%s","type": "phrase", "fields": [%s]}}`
  938. HL = `"highlight": {"pre_tags": [""],"post_tags": [""],"fields": {%s}}`
  939. highlightStr = `%s: {"fragment_size": %d,"number_of_fragments": 1}`
  940. FilterQuery_New = `{"query":{"bool":{"must": [%s%s%s],"should":[]}}}`
  941. MatchQueryString = `{"match": {%s: { "query":"%s", "operator": "and"}}}`
  942. HL_New = `"highlight": {"pre_tags": ["<HL>"],"post_tags": ["<HL>"],"fields": {%s}}`
  943. //数据查询高亮标记2019-07-10
  944. HL_MP = `"highlight": {"pre_tags": ["<HL>"],"post_tags": ["</HL>"],"fields": {%s}}`
  945. ik_highlightStr = `%s: {"fragment_size": %d,"number_of_fragments": 1,"require_field_match": true}`
  946. IK_pre_tags = `<font class=\"es-highlight\">`
  947. IK_post_tags = `</font>`
  948. HL_IK = `"highlight": {"pre_tags": ["` + IK_pre_tags + `"],"post_tags": ["` + IK_post_tags + `"],"fields": {%s}}`
  949. )
  950. //替换了"号
  951. func GetNgramQuery(query interface{}, mustquery, findfields string) (qstr string) {
  952. var words []string
  953. if q, ok := query.(string); ok {
  954. if q != "" {
  955. words = strings.Split(q, ",")
  956. }
  957. } else if q, ok := query.([]string); ok {
  958. words = q
  959. } else if q, ok := query.([]interface{}); ok {
  960. words = objArrToStringArr(q)
  961. }
  962. if words != nil {
  963. new_minq := fmt.Sprintf(minq, "%s", findfields)
  964. musts := []string{}
  965. for _, qs_words := range words {
  966. qws := strings.Split(qs_words, "+")
  967. mq := []string{}
  968. for _, qs_word := range qws {
  969. mq = append(mq, fmt.Sprintf(new_minq, ReplaceYH(qs_word)))
  970. }
  971. musts = append(musts, fmt.Sprintf(NgramMust, strings.Join(mq, ",")))
  972. }
  973. qstr = fmt.Sprintf(NgramStr, mustquery, strings.Join(musts, ","))
  974. //log.Println("ngram-query", qstr)
  975. } else {
  976. qstr = fmt.Sprintf(NgramStr, mustquery, "")
  977. }
  978. return
  979. }
  980. func GetNgramQuery_New(querystring, querymust interface{}, must, findfields string) (qstring string) {
  981. querymust_string := ""
  982. var wordsMust []string
  983. if q, ok := querymust.(string); ok {
  984. if q != "" {
  985. wordsMust = strings.Split(q, ",")
  986. }
  987. } else if q, ok := querymust.([]string); ok {
  988. wordsMust = q
  989. } else if q, ok := querymust.([]interface{}); ok {
  990. wordsMust = objArrToStringArr(q)
  991. }
  992. if wordsMust != nil {
  993. new_minq := fmt.Sprintf(minq, "%s", findfields)
  994. musts := []string{}
  995. for _, qs_wordsMust := range wordsMust {
  996. qws := strings.Split(qs_wordsMust, "+")
  997. mq := []string{}
  998. for _, qs_word := range qws {
  999. mq = append(mq, fmt.Sprintf(new_minq, qs_word))
  1000. }
  1001. musts = append(musts, fmt.Sprintf(NgramMust, strings.Join(mq, ",")))
  1002. }
  1003. querymust_string = strings.Join(musts, ",")
  1004. }
  1005. //log.Println("must", must, querymust_string)
  1006. //querystring---------------------------------------------
  1007. query_string := ""
  1008. var querysShold []string
  1009. if q, ok := querystring.(string); ok {
  1010. if q != "" {
  1011. querysShold = strings.Split(q, ",")
  1012. }
  1013. } else if q, ok := querystring.([]string); ok {
  1014. querysShold = q
  1015. } else if q, ok := querystring.([]interface{}); ok {
  1016. querysShold = objArrToStringArr(q)
  1017. }
  1018. if querysShold != nil {
  1019. for k, name := range strings.Split(findfields, ",") {
  1020. for _, qs_querysShold := range querysShold {
  1021. if k > 0 {
  1022. query_string = query_string + "," + fmt.Sprintf(MatchQueryString, fmt.Sprint(name), qs_querysShold)
  1023. } else {
  1024. query_string = query_string + fmt.Sprintf(MatchQueryString, fmt.Sprint(name), qs_querysShold)
  1025. }
  1026. }
  1027. }
  1028. }
  1029. //log.Println("querystring", query_string)
  1030. if querymust_string == "" {
  1031. qstring = fmt.Sprintf(FilterQuery_New, must, query_string, querymust_string)
  1032. } else {
  1033. qstring = fmt.Sprintf(FilterQuery_New, must, query_string, ","+querymust_string)
  1034. }
  1035. return
  1036. }
  1037. func GetByNgram(index, itype string, query interface{}, mustquery, findfields, order, fields string, start, limit int) *[]map[string]interface{} {
  1038. return GetByNgramAll(index, itype, query, mustquery, findfields, order, fields, start, limit, false, false)
  1039. }
  1040. //增加高亮、过滤查询、高亮截取字数
  1041. func GetByNgramOther(index, itype string, query interface{}, mustquery, findfields, order, fields string, start, limit int, highlight bool, filtermode bool, count int) *[]map[string]interface{} {
  1042. defer catch()
  1043. qstr := ""
  1044. if mustquery != "" && filtermode {
  1045. qstr = GetNgramQuery(query, "", findfields)
  1046. qstr = fmt.Sprintf(FilterQuery, mustquery, qstr[1:])
  1047. } else {
  1048. qstr = GetNgramQuery(query, mustquery, findfields)
  1049. }
  1050. if qstr != "" {
  1051. if highlight {
  1052. ws := []string{}
  1053. for _, w := range strings.Split(findfields, ",") {
  1054. ws = append(ws, fmt.Sprintf(highlightStr, w, count))
  1055. }
  1056. qstr = qstr[:len(qstr)-1] + `,` + fmt.Sprintf(HL, strings.Join(ws, ",")) + `}`
  1057. }
  1058. if len(fields) > 0 {
  1059. qstr = qstr[:len(qstr)-1] + `,"_source":[` + fields + "]}"
  1060. }
  1061. if len(order) > 0 {
  1062. qstr = qstr[:len(qstr)-1] + `,"sort":[` + SR(SR(SR(SR(order, ",", "},{", -1), " ", "", -1), ":-1", `:"desc"`, -1), ":1", `:"asc"`, -1) + `]}`
  1063. }
  1064. if start > -1 {
  1065. qstr = qstr[:len(qstr)-1] + `,"from":` + strconv.Itoa(start) + `,"size":` + strconv.Itoa(limit) + "}"
  1066. }
  1067. //log.Println("ngram-find", qstr)
  1068. return Get(index, itype, qstr)
  1069. } else {
  1070. return nil
  1071. }
  1072. }
  1073. //增加高亮、过滤查询
  1074. //替换了"号
  1075. func GetByNgramAll(index, itype string, query interface{}, mustquery, findfields, order, fields string, start, limit int, highlight bool, filtermode bool) *[]map[string]interface{} {
  1076. defer catch()
  1077. qstr := ""
  1078. if mustquery != "" && filtermode {
  1079. qstr = GetNgramQuery(query, "", findfields)
  1080. qstr = fmt.Sprintf(FilterQuery, mustquery, qstr[1:])
  1081. } else {
  1082. qstr = GetNgramQuery(query, mustquery, findfields)
  1083. }
  1084. if qstr != "" {
  1085. if highlight {
  1086. ws := []string{}
  1087. for _, w := range strings.Split(findfields, ",") {
  1088. ws = append(ws, fmt.Sprintf(highlightStr, w, 1))
  1089. }
  1090. qstr = qstr[:len(qstr)-1] + `,` + fmt.Sprintf(HL, strings.Join(ws, ",")) + `}`
  1091. }
  1092. if len(fields) > 0 {
  1093. qstr = qstr[:len(qstr)-1] + `,"_source":[` + fields + "]}"
  1094. }
  1095. if strings.HasPrefix(order, "CUSTOM_") {
  1096. qstr = qstr[:len(qstr)-1] + `,` + strings.TrimLeft(order, "CUSTOM_") + `}`
  1097. } else if len(order) > 0 {
  1098. qstr = qstr[:len(qstr)-1] + `,"sort":[` + SR(SR(SR(SR(order, ",", "},{", -1), " ", "", -1), ":-1", `:"desc"`, -1), ":1", `:"asc"`, -1) + `]}`
  1099. }
  1100. if start > -1 {
  1101. qstr = qstr[:len(qstr)-1] + `,"from":` + strconv.Itoa(start) + `,"size":` + strconv.Itoa(limit) + "}"
  1102. }
  1103. //log.Println("ngram-find", qstr)
  1104. return Get(index, itype, qstr)
  1105. } else {
  1106. return nil
  1107. }
  1108. }
  1109. //增加高亮、过滤查询
  1110. func GetByNgramAll_New(index, itype string, querystring, querymust interface{}, mustquery, findfields, order, fields string, start, limit int, highlight bool, filtermode bool) *[]map[string]interface{} {
  1111. defer catch()
  1112. qstr := ""
  1113. if filtermode {
  1114. qstr = GetNgramQuery_New(querystring, querymust, mustquery, findfields)
  1115. } else {
  1116. qstr = GetNgramQuery_New(querystring, "", mustquery, findfields)
  1117. }
  1118. if qstr != "" {
  1119. if highlight {
  1120. ws := []string{}
  1121. for _, w := range strings.Split(findfields, ",") {
  1122. ws = append(ws, w+`:{"force_source": true}`)
  1123. }
  1124. qstr = qstr[:len(qstr)-1] + `,` + fmt.Sprintf(HL_New, strings.Join(ws, ",")) + `}`
  1125. }
  1126. if len(fields) > 0 {
  1127. qstr = qstr[:len(qstr)-1] + `,"_source":[` + fields + "]}"
  1128. }
  1129. if len(order) > 0 {
  1130. qstr = qstr[:len(qstr)-1] + `,"sort":[` + SR(SR(SR(SR(order, ",", ",", -1), " ", "", -1), ":-1", `:"desc"`, -1), ":1", `:"asc"`, -1) + `]}`
  1131. }
  1132. if start > -1 {
  1133. qstr = qstr[:len(qstr)-1] + `,"from":` + strconv.Itoa(start) + `,"size":` + strconv.Itoa(limit) + "}"
  1134. }
  1135. //log.Println("ngram-find", order, qstr)
  1136. return Get(index, itype, qstr)
  1137. } else {
  1138. return nil
  1139. }
  1140. }
  1141. type KeyConfig struct {
  1142. Keys []string `json:"key"`
  1143. NotKeys []string `json:"notkey"`
  1144. InfoTypes []string `json:"infotype"`
  1145. Areas []string `json:"area"`
  1146. }
  1147. //替换了"号
  1148. func GetResForJY(index, itype string, keys []KeyConfig, allquery, findfields, SortQuery, fields string, start, limit int) *[]map[string]interface{} {
  1149. if len(keys) > 0 {
  1150. qstr := ""
  1151. new_minq := fmt.Sprintf(minq, "%s", findfields)
  1152. not_new_minq := fmt.Sprintf(minq, "%s", findfields) //排除词只查询标题
  1153. musts := []string{}
  1154. for _, qs_words := range keys {
  1155. mq := []string{}
  1156. notmq := []string{}
  1157. for _, qs_word := range qs_words.Keys {
  1158. mq = append(mq, fmt.Sprintf(new_minq, ReplaceYH(qs_word)))
  1159. /*
  1160. qs := AnalyzerWord("bidding", qs_word)
  1161. for _, qw := range qs {
  1162. mq = append(mq, fmt.Sprintf(new_minq, ReplaceYH(qw)))
  1163. }
  1164. */
  1165. }
  1166. for _, qs_word := range qs_words.NotKeys {
  1167. notmq = append(notmq, fmt.Sprintf(not_new_minq, ReplaceYH(qs_word)))
  1168. }
  1169. if len(qs_words.Areas) > 0 {
  1170. mq = append(mq, fmt.Sprintf(`{"terms":{"area":["%s"]}}`, strings.Join(qs_words.Areas, `","`)))
  1171. }
  1172. if len(qs_words.InfoTypes) > 0 {
  1173. mq = append(mq, fmt.Sprintf(`{"terms":{"toptype":["%s"]}}`, strings.Join(qs_words.InfoTypes, `","`)))
  1174. }
  1175. musts = append(musts, fmt.Sprintf(NgramMustAndNot, strings.Join(mq, ","), strings.Join(notmq, ",")))
  1176. }
  1177. qstr = fmt.Sprintf(NgramStr, "", strings.Join(musts, ","))
  1178. qstr = fmt.Sprintf(FilterQuery, allquery, qstr[1:])
  1179. ws := []string{}
  1180. for _, w := range strings.Split(findfields, ",") {
  1181. ws = append(ws, fmt.Sprintf(highlightStr, w, 1))
  1182. }
  1183. qstr = qstr[:len(qstr)-1] + `,` + fmt.Sprintf(HL, strings.Join(ws, ",")) + `}`
  1184. if len(fields) > 0 {
  1185. qstr = qstr[:len(qstr)-1] + `,"_source":[` + fields + "]}"
  1186. }
  1187. if len(SortQuery) > 0 {
  1188. qstr = qstr[:len(qstr)-1] + `,"sort":` + SortQuery + `}`
  1189. }
  1190. if start > -1 {
  1191. qstr = qstr[:len(qstr)-1] + `,"from":` + strconv.Itoa(start) + `,"size":` + strconv.Itoa(limit) + "}"
  1192. }
  1193. //log.Println("jy-ngram-find", qstr)
  1194. return Get(index, itype, qstr)
  1195. } else {
  1196. return nil
  1197. }
  1198. }
  1199. func ReplaceYH(src string) (rpl string) {
  1200. return strings.Replace(src, `"`, `\"`, -1)
  1201. }
  1202. //
  1203. func GetAllByNgramNew(index, itype, qstr, findfields, order, fields string, start, limit, count int, highlight bool) (*[]map[string]interface{}, int64) {
  1204. if qstr != "" {
  1205. if highlight {
  1206. ws := []string{}
  1207. for _, w := range strings.Split(findfields, ",") {
  1208. ws = append(ws, fmt.Sprintf(highlightStr, w, count))
  1209. }
  1210. qstr = qstr[:len(qstr)-1] + `,` + fmt.Sprintf(HL, strings.Join(ws, ",")) + `}`
  1211. }
  1212. if len(fields) > 0 {
  1213. qstr = qstr[:len(qstr)-1] + `,"_source":[` + fields + "]}"
  1214. }
  1215. if len(order) > 0 {
  1216. qstr = qstr[:len(qstr)-1] + `,"sort":[` + SR(SR(SR(SR(order, ",", "},{", -1), " ", "", -1), ":-1", `:"desc"`, -1), ":1", `:"asc"`, -1) + `]}`
  1217. }
  1218. if start > -1 {
  1219. qstr = qstr[:len(qstr)-1] + `,"from":` + strconv.Itoa(start) + `,"size":` + strconv.Itoa(limit) + "}"
  1220. }
  1221. //log.Println("GetAllByNgram:", qstr)
  1222. return GetNew(index, itype, qstr)
  1223. } else {
  1224. return nil, 0
  1225. }
  1226. }
  1227. //
  1228. func GetAllByNgram(index, itype, qstr, findfields, order, fields string, start, limit, count int, highlight bool) *[]map[string]interface{} {
  1229. if qstr != "" {
  1230. if highlight {
  1231. ws := []string{}
  1232. for _, w := range strings.Split(findfields, ",") {
  1233. ws = append(ws, fmt.Sprintf(highlightStr, w, count))
  1234. }
  1235. qstr = qstr[:len(qstr)-1] + `,` + fmt.Sprintf(HL, strings.Join(ws, ",")) + `}`
  1236. }
  1237. if len(fields) > 0 {
  1238. qstr = qstr[:len(qstr)-1] + `,"_source":[` + fields + "]}"
  1239. }
  1240. if len(order) > 0 {
  1241. qstr = qstr[:len(qstr)-1] + `,"sort":[` + SR(SR(SR(SR(order, ",", "},{", -1), " ", "", -1), ":-1", `:"desc"`, -1), ":1", `:"asc"`, -1) + `]}`
  1242. }
  1243. if start > -1 {
  1244. qstr = qstr[:len(qstr)-1] + `,"from":` + strconv.Itoa(start) + `,"size":` + strconv.Itoa(limit) + "}"
  1245. }
  1246. //log.Println("GetAllByNgram:", qstr)
  1247. return Get(index, itype, qstr)
  1248. } else {
  1249. return nil
  1250. }
  1251. }
  1252. //数据标记2019-07-10
  1253. func GetAllByNgram_MP(index, itype, qstr, findfields, order, fields string, start, limit, count int, highlight bool) *[]map[string]interface{} {
  1254. if qstr != "" {
  1255. if highlight {
  1256. ws := []string{}
  1257. for _, w := range strings.Split(findfields, ",") {
  1258. ws = append(ws, fmt.Sprintf(highlightStr, w, count))
  1259. }
  1260. qstr = qstr[:len(qstr)-1] + `,` + fmt.Sprintf(HL_MP, strings.Join(ws, ",")) + `}`
  1261. }
  1262. if len(fields) > 0 {
  1263. qstr = qstr[:len(qstr)-1] + `,"_source":[` + fields + "]}"
  1264. }
  1265. if len(order) > 0 {
  1266. qstr = qstr[:len(qstr)-1] + `,"sort":[` + SR(SR(SR(SR(order, ",", "},{", -1), " ", "", -1), ":-1", `:"desc"`, -1), ":1", `:"asc"`, -1) + `]}`
  1267. }
  1268. if start > -1 {
  1269. qstr = qstr[:len(qstr)-1] + `,"from":` + strconv.Itoa(start) + `,"size":` + strconv.Itoa(limit) + "}"
  1270. }
  1271. // log.Println("GetAllByNgram:", qstr)
  1272. return Get(index, itype, qstr)
  1273. } else {
  1274. return nil
  1275. }
  1276. }
  1277. //ik 分词
  1278. func GetAllByIk(index, itype, qstr, findfields, order, fields string, start, limit, count int, highlight bool) *[]map[string]interface{} {
  1279. if qstr != "" {
  1280. if highlight {
  1281. ws := []string{}
  1282. for _, w := range strings.Split(findfields, ",") {
  1283. ws = append(ws, fmt.Sprintf(ik_highlightStr, w, count))
  1284. }
  1285. qstr = qstr[:len(qstr)-1] + `,` + fmt.Sprintf(HL_IK, strings.Join(ws, ",")) + `}`
  1286. }
  1287. if len(fields) > 0 {
  1288. qstr = qstr[:len(qstr)-1] + `,"_source":[` + fields + "]}"
  1289. }
  1290. if len(order) > 0 {
  1291. qstr = qstr[:len(qstr)-1] + `,"sort":[` + SR(SR(SR(SR(order, ",", "},{", -1), " ", "", -1), ":-1", `:"desc"`, -1), ":1", `:"asc"`, -1) + `]}`
  1292. }
  1293. if start > -1 {
  1294. qstr = qstr[:len(qstr)-1] + `,"from":` + strconv.Itoa(start) + `,"size":` + strconv.Itoa(limit) + "}"
  1295. }
  1296. //log.Println("GetAllByNgram:", qstr)
  1297. return Get(index, itype, qstr)
  1298. } else {
  1299. return nil
  1300. }
  1301. }
  1302. //分词
  1303. func AnalyzerWord(index, word string) (result []string) {
  1304. client := GetEsConn()
  1305. defer DestoryEsConn(client)
  1306. result = []string{}
  1307. p := url.Values{}
  1308. p["text"] = []string{word}
  1309. p["analyzer"] = []string{"ik"}
  1310. by, err := client.PerformRequest("GET", "/"+index+"/_analyze", p, nil)
  1311. if err != nil {
  1312. log.Println("AnalyzerWord Error:", err)
  1313. return
  1314. }
  1315. b, err := by.Body.MarshalJSON()
  1316. if err != nil {
  1317. log.Println("AnalyzerWord MarshalJSON Error:", err)
  1318. return
  1319. }
  1320. var res map[string][]map[string]interface{}
  1321. err = json.Unmarshal(b, &res)
  1322. if err != nil {
  1323. log.Println("AnalyzerWord Unmarshal Error:", err)
  1324. return
  1325. }
  1326. if res == nil {
  1327. return
  1328. }
  1329. for _, v := range res["tokens"] {
  1330. token, _ := v["token"].(string)
  1331. if token != "" {
  1332. result = append(result, token)
  1333. }
  1334. }
  1335. return
  1336. }
  1337. func GetResForJYView(index, itype string, keys []KeyConfig, allquery, findfields, SortQuery, fields string, start, limit int) *[]map[string]interface{} {
  1338. if len(keys) > 0 {
  1339. qstr := ""
  1340. new_minq := fmt.Sprintf(minq, "%s", findfields)
  1341. not_new_minq := fmt.Sprintf(minq, "%s", findfields) //排除词只查询标题
  1342. musts := []string{}
  1343. for _, qs_words := range keys {
  1344. mq := []string{}
  1345. notmq := []string{}
  1346. for _, qs_word := range qs_words.Keys {
  1347. mq = append(mq, fmt.Sprintf(new_minq, ReplaceYH(qs_word)))
  1348. }
  1349. for _, qs_word := range qs_words.NotKeys {
  1350. notmq = append(notmq, fmt.Sprintf(not_new_minq, ReplaceYH(qs_word)))
  1351. }
  1352. if len(qs_words.Areas) > 0 {
  1353. mq = append(mq, fmt.Sprintf(`{"terms":{"area":["%s"]}}`, strings.Join(qs_words.Areas, `","`)))
  1354. }
  1355. if len(qs_words.InfoTypes) > 0 {
  1356. mq = append(mq, fmt.Sprintf(`{"terms":{"toptype":["%s"]}}`, strings.Join(qs_words.InfoTypes, `","`)))
  1357. }
  1358. musts = append(musts, fmt.Sprintf(NgramMustAndNot, strings.Join(mq, ","), strings.Join(notmq, ",")))
  1359. }
  1360. qstr = fmt.Sprintf(NgramStr, "", strings.Join(musts, ","))
  1361. qstr = fmt.Sprintf(FilterQuery, allquery, qstr[1:])
  1362. ws := []string{}
  1363. for _, w := range strings.Split(findfields, ",") {
  1364. ws = append(ws, fmt.Sprintf(highlightStr, w, 1))
  1365. }
  1366. qstr = qstr[:len(qstr)-1] + `,` + fmt.Sprintf(HL, strings.Join(ws, ",")) + `}`
  1367. if len(fields) > 0 {
  1368. qstr = qstr[:len(qstr)-1] + `,"_source":[` + fields + "]}"
  1369. }
  1370. if len(SortQuery) > 0 {
  1371. qstr = qstr[:len(qstr)-1] + `,"sort":` + SortQuery + `}`
  1372. }
  1373. if start > -1 {
  1374. qstr = qstr[:len(qstr)-1] + `,"from":` + strconv.Itoa(start) + `,"size":` + strconv.Itoa(limit) + "}"
  1375. }
  1376. return Get(index, itype, qstr)
  1377. } else {
  1378. return nil
  1379. }
  1380. }
  1381. //返回count 和 res
  1382. func GetWithCount(index, itype, query string) (int64, *[]map[string]interface{}) {
  1383. client := GetEsConn()
  1384. defer func() {
  1385. go DestoryEsConn(client)
  1386. }()
  1387. var res []map[string]interface{}
  1388. var count int64 = 0
  1389. if client != nil {
  1390. defer func() {
  1391. if r := recover(); r != nil {
  1392. log.Println("[E]", r)
  1393. for skip := 1; ; skip++ {
  1394. _, file, line, ok := runtime.Caller(skip)
  1395. if !ok {
  1396. break
  1397. }
  1398. go log.Printf("%v,%v\n", file, line)
  1399. }
  1400. }
  1401. }()
  1402. searchResult, err := client.Search().Index(index).Type(itype).Source(query).Do()
  1403. if err != nil {
  1404. log.Println("从ES查询出错", err.Error())
  1405. return count, nil
  1406. }
  1407. if searchResult.Hits != nil {
  1408. resNum := len(searchResult.Hits.Hits)
  1409. count = searchResult.Hits.TotalHits
  1410. if resNum > 5000 {
  1411. log.Println("查询结果太多,查询到:", resNum, "条")
  1412. } else {
  1413. res = make([]map[string]interface{}, resNum)
  1414. for i, hit := range searchResult.Hits.Hits {
  1415. parseErr := json.Unmarshal(*hit.Source, &res[i])
  1416. if parseErr == nil && hit.Highlight != nil && res[i] != nil {
  1417. res[i]["highlight"] = map[string][]string(hit.Highlight)
  1418. }
  1419. }
  1420. }
  1421. }
  1422. }
  1423. return count, &res
  1424. }