Trans2MongoDB.cs 5.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Linq;
  4. using System.Text;
  5. using System.Threading.Tasks;
  6. using CSharpUtil;
  7. using StackExchange.Redis;
  8. using MongoDB;
  9. using MongoDB.Bson;
  10. using MongoDB.Driver;
  11. using MongoDB.Bson.Serialization;
  12. using Newtonsoft.Json;
  13. using Newtonsoft.Json.Linq;
  14. namespace RedisTransfer
  15. {
  16. internal class Trans2MongoDB : ITransfer
  17. {
  18. public void Trans(int db = 0)
  19. {
  20. var redis = Redis.Rdb(db);
  21. var mdb = Mongo.Mdb();
  22. var collection = mdb.GetCollection<BsonDocument>("ylsj2019taptap");
  23. Console.WriteLine("start");
  24. var s = File.ReadAllLines("keys.txt");
  25. s.ToList().ForEach(x =>
  26. {
  27. var t = redis.KeyType(x);
  28. switch (t)
  29. {
  30. case RedisType.String:
  31. {
  32. var v = redis.StringGet(x).ToString();
  33. var bson = new BsonDocument("key",x); // (x, BsonDocument.Parse());
  34. if (BsonDocument.TryParse(v, out var bo))
  35. {
  36. bson.Add("value", bo);
  37. }
  38. else
  39. {
  40. bson.Add("value", v);
  41. }
  42. collection.InsertOne(bson);
  43. break;
  44. }
  45. case RedisType.Hash:
  46. {
  47. var jo = new BsonDocument();
  48. redis.HashGetAll(x).ToList().ForEach(x =>
  49. {
  50. if (BsonDocument.TryParse(x.Value.ToString(), out var bo))
  51. {
  52. jo.Add(x.Name, bo);
  53. }
  54. else
  55. {
  56. jo.Add(x.Name, x.Value.ToString());
  57. }
  58. //jo.Add(x.Name, BsonDocument.Parse(x.Value.ToString()))
  59. }
  60. );
  61. var bson = new BsonDocument("key", x);
  62. bson.Add("value", jo);
  63. collection.InsertOne(bson);
  64. break;
  65. }
  66. case RedisType.List:
  67. {
  68. var ja = new BsonArray();
  69. foreach (var r in redis.ListRange(x, 0, -1))
  70. {
  71. if (BsonDocument.TryParse(r.ToString(), out var bo))
  72. {
  73. ja.Add(bo);
  74. }
  75. else
  76. {
  77. ja.Add(r.ToString());
  78. }
  79. }
  80. var bson = new BsonDocument("key", x);
  81. bson.Add("value", ja);
  82. collection.InsertOne(bson);
  83. break;
  84. }
  85. case RedisType.Set:
  86. {
  87. var ja = new BsonArray();
  88. foreach (var r in redis.SetMembers(x))
  89. {
  90. if (BsonDocument.TryParse(r.ToString(), out var bo))
  91. {
  92. ja.Add(bo);
  93. }
  94. else
  95. {
  96. ja.Add(r.ToString());
  97. }
  98. }
  99. var bson = new BsonDocument("key", x);
  100. bson.Add("value", ja);
  101. collection.InsertOne(bson);
  102. //Console.WriteLine(new JProperty(x, ja).ToString() + ",");
  103. break;
  104. }
  105. case RedisType.SortedSet:
  106. {
  107. var ja = new BsonArray();
  108. foreach (var r in redis.SortedSetRangeByScoreWithScores(x))
  109. {
  110. if (BsonDocument.TryParse(r.ToString(), out var bo))
  111. {
  112. ja.Add(bo);
  113. }
  114. else
  115. {
  116. ja.Add(r.ToString());
  117. }
  118. }
  119. var bson = new BsonDocument("key", x);
  120. bson.Add("value", ja);
  121. collection.InsertOne(bson);
  122. //Console.WriteLine(new JProperty(x, ja).ToString() + ",");
  123. break;
  124. }
  125. default:
  126. Console.WriteLine("不支持的redis数据类型");
  127. break;
  128. }
  129. });
  130. Console.WriteLine("end");
  131. }
  132. }
  133. }