main.go 2.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107
  1. package main
  2. import (
  3. "data_ai/tool"
  4. "data_ai/udp"
  5. "data_ai/ul"
  6. log "github.com/donnie4w/go-logger/logger"
  7. "github.com/gogf/gf/v2/util/gconv"
  8. "strings"
  9. "sync"
  10. "unicode/utf8"
  11. )
  12. func init() {
  13. ul.IsLocal = false //是否本地
  14. ul.IsTool = false //是否工具
  15. ul.IsFull = false //是否全量
  16. if ul.IsTool {
  17. ul.InitToolVar()
  18. } else {
  19. ul.InitGlobalVar()
  20. if !ul.IsFull {
  21. udp.InitProcessVar()
  22. }
  23. }
  24. if !ul.IsTool {
  25. ul.InitOss(ul.IsLocal)
  26. }
  27. log.Debug("测试一下转换MD方法···")
  28. detail := `<table border=\"1\"><tbody><tr><th>采购人名称</th><td>云南中医药大学</td></tr><tr><th>中标(成交)供应商名称</th><td>云南智蓝云鸽信息科技有限公司</td></tr><tr><th>合同金额</th><td>224.99万元 人民币</td></tr><tr><th>合同期限</th><td>年</td></tr><tr><th>合同签署时间</th><td>2023-12-29 00:00:00</td></tr></tbody></table>`
  29. new_detail := ul.HttpConvertToMarkdown(detail)
  30. log.Debug(new_detail)
  31. }
  32. func main() {
  33. if ul.IsTool {
  34. tool.StartToolInfo()
  35. return
  36. }
  37. //extract.TestSingleFieldInfo("bidding", "67763c6b3309c0998ba25811")
  38. lock := make(chan bool)
  39. <-lock
  40. }
  41. // 测试调试数据
  42. func test1() {
  43. q, total := map[string]interface{}{
  44. "_id": map[string]interface{}{
  45. "$gt": ul.StringTOBsonId("677638b13309c0998ba2488f"),
  46. "$lte": ul.StringTOBsonId("6777a3d23309c0998ba89797"),
  47. },
  48. }, 0
  49. count, _ := ul.BidMgo.Count("bidding", q)
  50. log.Debug("查询数量:", count)
  51. pool_mgo := make(chan bool, 10)
  52. wg_mgo := &sync.WaitGroup{}
  53. sess := ul.BidMgo.GetMgoConn()
  54. defer ul.BidMgo.DestoryMongoConn(sess)
  55. it := sess.DB(ul.BidMgo.DbName).C(ul.Bid_Name).Find(&q).Iter()
  56. for tmp := make(map[string]interface{}); it.Next(&tmp); total++ {
  57. if total%1000 == 0 {
  58. log.Debug("cur ai index ", total)
  59. }
  60. pool_mgo <- true
  61. wg_mgo.Add(1)
  62. go func(tmp map[string]interface{}) {
  63. defer func() {
  64. <-pool_mgo
  65. wg_mgo.Done()
  66. }()
  67. //tmpid := ul.BsonTOStringId(tmp["_id"])
  68. s1, s2 := gconv.String(tmp["budget"]), gconv.String(tmp["bidamount"])
  69. r1, r2 := strings.Split(s1, "."), strings.Split(s2, ".")
  70. if len(r1) == 2 {
  71. l1 := utf8.RuneCountInString(r1[1])
  72. if l1 > 3 {
  73. //log.Debug("问题金额:", tmpid, "~", s1)
  74. ul.BidMgo.Save("zktest_err_0106", tmp)
  75. return
  76. }
  77. }
  78. if len(r2) == 2 {
  79. l2 := utf8.RuneCountInString(r2[1])
  80. if l2 > 3 {
  81. //log.Debug("问题金额:", tmpid, "~", s2)
  82. ul.BidMgo.Save("zktest_err_0106", tmp)
  83. return
  84. }
  85. }
  86. }(tmp)
  87. tmp = make(map[string]interface{})
  88. }
  89. wg_mgo.Wait()
  90. log.Debug("is over ...", total)
  91. }
  92. func c(s string) string {
  93. s = strings.ReplaceAll(s, "(", "(")
  94. s = strings.ReplaceAll(s, ")", ")")
  95. s = strings.ReplaceAll(s, ",", ",")
  96. return s
  97. }