func main() {func main() { var connString string if len(os.Args) < 2 { fmt.Fprintf(os.Stderr, "mongo <server connection string>") connString = "mongodb://127.0.0.1:27017/?connect=direct" } else { connString = os.Args[1] } clientOptions := options.Client().ApplyURI(connString) clientOptions.SetConnectTimeout(time.Duration(2000) * time.Millisecond) clientOptions.SetSocketTimeout(time.Duration(5000) * time.Millisecond) clientOptions.SetServerSelectionTimeout(time.Duration(2000) * time.Millisecond) ctx, _ := context.WithTimeout(context.Background(), time.Duration(5000)*time.Millisecond) client, err := mongo.Connect(ctx, clientOptions) if err != nil { fmt.Fprintf(os.Stderr, "connect mongo failed: %s\n", err.Error()) os.Exit(1) } defer client.Disconnect(ctx)
 err = client.Ping(context.TODO(), readpref.Primary()) if err != nil { fmt.Fprintf(os.Stderr, "ping mongo failed: %s\n", err.Error()) os.Exit(1) }
 coll := client.Database("maicai").Collection("station_order")
 var now = time.Now() timestamp0 := now.Truncate(30 * time.Minute) var unixTime int64 unixTime = timestamp0.Unix()
 projections := bson.D{ {"_id", 0}, {"order_number", 1}, {"station_id", 1}, {"status", 1}, {"finish_time", 1}, {"goods.origin_price", 1}, {"goods.price", 1}, {"goods.id", 1}, {"goods.count", 1}, {"goods.purchase_stockout", 1}, }
 findOptions := options.Find() findOptions.SetProjection(projections)
 cur, err := coll.Find( ctx, bson.D{ {"finish_time", bson.D{ {"$gte", unixTime}, }}, {"status", bson.D{ {"$in", bson.A{2}}, }}, }, findOptions, )
 if err != nil { fmt.Fprintf(os.Stderr, "mongo find failed: %s\n", err.Error()) os.Exit(1) } orders := make([]StationOrder, 0, 20000) err = cur.All(context.Background(), &orders) if err != nil { fmt.Fprintf(os.Stderr, "Query mongo station order failed: %s", err.Error()) os.Exit(1) }
 t := time.Unix(unixTime, 0) fmt.Fprintf(os.Stdout, "Orders from mongo DB: time = %s, count = %d", t.Format("2006-01-02 15:04:05"), len(orders))}