123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268 |
- package api
- import (
- "github.com/gin-gonic/gin"
- )
- // import (
- // "box-cost/db/model"
- // "box-cost/db/repo"
- // randc "crypto/rand"
- // "encoding/json"
- // "fmt"
- // "log"
- // "math/big"
- // "math/rand"
- // "sort"
- // "sync"
- // "time"
- // "unsafe"
- // "github.com/gin-gonic/gin"
- // "go.mongodb.org/mongo-driver/bson"
- // "go.mongodb.org/mongo-driver/mongo"
- // )
- func Printr(c *gin.Context, apictx *ApiSession) (interface{}, error) {
- // https://sku3d-test.obs.cn-east-3.myhuaweicloud.com/image/jpg/1675757991547cnt7X9_2.jpg zhang
- // https://sku3d-test.obs.cn-east-3.myhuaweicloud.com/image/jpg/1675758027305uLKtda_1.jpg zhu
- // loc, _ := time.LoadLocation("Local")
- // date := time.Now().Format("2006-01-02")
- // startDate := date + " 00:00:00"
- // startTime, _ := time.ParseInLocation("2006-01-02 15:04:05", startDate, loc)
- // endDate := date + " 23:59:59"
- // endTime, _ := time.ParseInLocation("2006-01-02 15:04:05", endDate, loc)
- // fmt.Println(startTime.Unix())
- // fmt.Println(endTime.Unix())
- // print
- // return "success", nil
- // 去重
- // arr := []string{"a", "a", "a", "b", "a", "a", "a", "c", "c", "b"}
- // newArr := removeDuplicationSort(arr)
- // for k, v := range newArr {
- // fmt.Printf("k:%d,v:%s\n", k, v)
- // }
- return nil, nil
- }
- // // 去重相邻元素
- // func removeDuplicationSort(arr []string) []string {
- // length := len(arr)
- // if length == 0 {
- // return arr
- // }
- // j := 0
- // for i := 1; i < length; i++ {
- // if arr[i] != arr[j] {
- // j++
- // if j < i {
- // swap(arr, i, j)
- // }
- // }
- // }
- // return arr[:j+1]
- // }
- // func swap(arr []string, a, b int) {
- // arr[a], arr[b] = arr[b], arr[a]
- // }
- // // 生成百万数据
- // func GenData(_ *gin.Context, apictx *ApiSession) (interface{}, error) {
- // var wg sync.WaitGroup
- // for i := 0; i < 100; i++ {
- // wg.Add(1)
- // go insertData(apictx, &wg)
- // }
- // wg.Wait()
- // return true, nil
- // }
- // // 聚合查询百万数据
- // func SearchData(_ *gin.Context, apictx *ApiSession) (interface{}, error) {
- // dataStatResult := make(chan bson.M)
- // var output ResultSlice
- // var wg sync.WaitGroup
- // // 82个gorutine 执行聚合
- // for i := 18; i < 100; i++ {
- // wg.Add(1)
- // go DataAggregate(apictx, i, dataStatResult, &wg)
- // }
- // // 从管道中获取聚合结果 保存在数组中
- // for value := range dataStatResult {
- // output = append(output, OutPut{
- // Age: value["age"].(int32),
- // Sex: value["sex"].(int32),
- // Total: value["total"].(int32),
- // AvgSalary: value["avgSalary"].(float64),
- // })
- // if len(output) == 164 { // 如果大于164,不会跳出;如果小于164,wg.done != wg.add ,wg.wait阻塞
- // break
- // }
- // }
- // wg.Wait()
- // //倒序排列
- // sort.Sort(output)
- // for _, v := range output {
- // result, err := json.Marshal(&v)
- // if err != nil {
- // fmt.Println("json.marshal failed, err:", err)
- // return nil, err
- // }
- // fmt.Println(string(result))
- // }
- // return output, nil
- // }
- // func insertData(apictx *ApiSession, wg *sync.WaitGroup) {
- // collectName := "aggregate-test"
- // rowNum := 10000
- // for i := 0; i < rowNum; i++ {
- // repo.RepoAddDoc(apictx.CreateRepoCtx(), collectName, &model.AggregateTest{
- // Name: randName(6),
- // Age: randAge(),
- // Sex: randSex(),
- // Salary: randSalary(),
- // })
- // }
- // wg.Done()
- // }
- // func randSex() *int {
- // result, _ := randc.Int(randc.Reader, big.NewInt(2))
- // sex := int(result.Int64())
- // return &sex
- // }
- // func randAge() int {
- // result, _ := randc.Int(randc.Reader, big.NewInt(82))
- // return int(result.Int64() + 18)
- // }
- // func randSalary() int {
- // result, _ := randc.Int(randc.Reader, big.NewInt(10000))
- // return int(result.Int64() + 2000)
- // }
- // const letters = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
- // var src = rand.NewSource(time.Now().UnixNano())
- // const (
- // // 6 bits to represent a letter index
- // letterIdBits = 6
- // // All 1-bits as many as letterIdBits
- // letterIdMask = 1<<letterIdBits - 1
- // letterIdMax = 63 / letterIdBits
- // )
- // func randName(n int) string {
- // b := make([]byte, n)
- // // A rand.Int63() generates 63 random bits, enough for letterIdMax letters!
- // for i, cache, remain := n-1, src.Int63(), letterIdMax; i >= 0; {
- // if remain == 0 {
- // cache, remain = src.Int63(), letterIdMax
- // }
- // if idx := int(cache & letterIdMask); idx < len(letters) {
- // b[i] = letters[idx]
- // i--
- // }
- // cache >>= letterIdBits
- // remain--
- // }
- // return *(*string)(unsafe.Pointer(&b))
- // }
- // func genPipeline(age int) (bson.D, bson.D, bson.D) {
- // matchStage := bson.D{
- // {"$match", bson.D{
- // {"age",
- // bson.D{
- // {"$eq", age},
- // }},
- // }},
- // }
- // groupStage := bson.D{
- // {"$group", bson.D{
- // {"_id", bson.D{
- // {"age", "$age"},
- // {"sex", "$sex"},
- // }},
- // {"age", bson.D{
- // {"$first", "$age"},
- // }},
- // {"sex", bson.D{
- // {"$first", "$sex"},
- // }},
- // {"total", bson.D{
- // {"$sum", 1},
- // }},
- // {"avgSalary", bson.D{
- // {"$avg", "$salary"},
- // }},
- // }},
- // }
- // projectStage := bson.D{
- // {"$project", bson.D{
- // {"_id", 0},
- // {"age", 1},
- // {"sex", 1},
- // {"total", 1},
- // {"avgSalary", 1},
- // }},
- // }
- // return matchStage, groupStage, projectStage
- // }
- // func DataAggregate(apictx *ApiSession, age int, resultChan chan bson.M, wg *sync.WaitGroup) {
- // matchStage, groupStage, projectStage := genPipeline(age)
- // // opts := options.Aggregate().SetMaxTime(15 * time.Second)
- // cursor, err := apictx.Svc.Mongo.GetCollection("aggregate-test").Aggregate(apictx.CreateRepoCtx().Ctx, mongo.Pipeline{matchStage, groupStage, projectStage})
- // if err != nil {
- // log.Println(err)
- // }
- // //打印文档内容
- // var results []bson.M
- // if err = cursor.All(apictx.CreateRepoCtx().Ctx, &results); err != nil {
- // log.Println(err)
- // }
- // log.Printf("%#v\n", results)
- // for _, result := range results {
- // resultChan <- result
- // }
- // wg.Done()
- // }
- // type OutPut struct {
- // Age int32 `json:"age"`
- // Sex int32 `json:"sex"`
- // Total int32 `json:"total"`
- // AvgSalary float64 `json:"avg_salary"`
- // }
- // type ResultSlice []OutPut
- // func (a ResultSlice) Len() int { // 重写 Len() 方法
- // return len(a)
- // }
- // func (a ResultSlice) Swap(i, j int) { // 重写 Swap() 方法
- // a[i], a[j] = a[j], a[i]
- // }
- // func (a ResultSlice) Less(i, j int) bool { // 重写 Less() 方法, 从大到小排序
- // return a[j].Age < a[i].Age
- // }
|