AzureStorageFactory.cs 16 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392
  1. using Microsoft.Azure.Cosmos.Table;
  2. using Microsoft.Extensions.Options;
  3. using Microsoft.Extensions.Logging;
  4. using System;
  5. using System.Collections.Generic;
  6. using System.Text;
  7. using Microsoft.Extensions.DependencyInjection;
  8. using Azure.Storage.Blobs;
  9. using Azure.Storage.Blobs.Models;
  10. using Azure.Storage.Blobs.Specialized;
  11. using Azure.Storage.Sas;
  12. using Azure.Storage;
  13. using TEAMModelOS.SDK.Extension;
  14. using Azure.Storage.Queues;
  15. using TEAMModelOS.SDK.Context.Attributes.Azure;
  16. using System.Threading.Tasks;
  17. using TEAMModelOS.SDK.Module.AzureBlob.Configuration;
  18. namespace TEAMModelOS.SDK.DI
  19. {
  20. public class AzureStorageFactory
  21. {
  22. private readonly IServiceProvider _services;
  23. private readonly IOptionsMonitor<AzureStorageFactoryOptions> _optionsMonitor;
  24. private readonly ILogger _logger;
  25. public AzureStorageFactory(IServiceProvider services, IOptionsMonitor<AzureStorageFactoryOptions> optionsMonitor, ILogger<AzureStorageFactory> logger)
  26. {
  27. if (services == null) throw new ArgumentNullException(nameof(services));
  28. if (optionsMonitor == null) throw new ArgumentNullException(nameof(optionsMonitor));
  29. _services = services;
  30. _optionsMonitor = optionsMonitor;
  31. _logger = logger;
  32. }
  33. public BlobServiceClient GetBlobServiceClient(string name = "Default")
  34. {
  35. try
  36. {
  37. var options = _optionsMonitor.Get(name);
  38. return new BlobServiceClient(options.StorageAccountConnectionString);
  39. }
  40. catch (OptionsValidationException e)
  41. {
  42. _logger?.LogWarning(e, e.Message);
  43. return null;
  44. }
  45. }
  46. public BlobContainerClient GetBlobContainerClient(string containerName, string name = "Default")
  47. {
  48. try
  49. {
  50. var options = _optionsMonitor.Get(name);
  51. return new BlobContainerClient(options.StorageAccountConnectionString, containerName.ToLower());
  52. }
  53. catch (OptionsValidationException e)
  54. {
  55. _logger?.LogWarning(e, e.Message);
  56. return null;
  57. }
  58. }
  59. public BlobBatchClient GetBlobBatchClient(string name = "Default")
  60. {
  61. try
  62. {
  63. var options = _optionsMonitor.Get(name);
  64. BlobServiceClient blobServiceClient = new BlobServiceClient(options.StorageAccountConnectionString);
  65. return blobServiceClient.GetBlobBatchClient();
  66. }
  67. catch (OptionsValidationException e)
  68. {
  69. _logger?.LogWarning(e, e.Message);
  70. return null;
  71. }
  72. }
  73. /// <summary>
  74. /// 取得Blob Container SAS (有效期預設99年)
  75. /// </summary>
  76. /// <param name="containerName">容器名稱</param>
  77. /// <param name="blobContainerSasPermissions">權限(可多選)Flags</param>
  78. /// <param name="name"></param>
  79. /// <returns></returns>
  80. public (string uri, string sas) GetBlobContainerSAS99Year(string containerName, BlobContainerSasPermissions blobContainerSasPermissions, string name = "Default")
  81. {
  82. try
  83. {
  84. var keys = Utils.ParseConnectionString(_optionsMonitor.Get(name).StorageAccountConnectionString);
  85. var accountname = keys["AccountName"];
  86. var accountkey = keys["AccountKey"];
  87. var endpoint = keys["EndpointSuffix"];
  88. var blobSasBuilder = new BlobSasBuilder
  89. {
  90. StartsOn = DateTimeOffset.UtcNow.Subtract(new TimeSpan(0, 15, 0)),
  91. ExpiresOn = DateTimeOffset.UtcNow.Add(new TimeSpan(36135, 0, 0, 0)),
  92. BlobContainerName = containerName.ToLower()
  93. };
  94. blobSasBuilder.SetPermissions(blobContainerSasPermissions);
  95. var sskc = new StorageSharedKeyCredential(accountname, accountkey);
  96. BlobSasQueryParameters sasQueryParameters = blobSasBuilder.ToSasQueryParameters(sskc);
  97. UriBuilder fullUri = new UriBuilder()
  98. {
  99. Scheme = "https",
  100. Host = $"{accountname}.blob.{endpoint}",
  101. Path = containerName.ToLower()
  102. //Query = sasQueryParameters.ToString()
  103. };
  104. return (fullUri.Uri.ToString(), sasQueryParameters.ToString());
  105. }
  106. catch
  107. {
  108. return (null, null);
  109. }
  110. }
  111. /// <summary>
  112. /// 取得Blob Container SAS (有效期預設一天)
  113. /// </summary>
  114. /// <param name="containerName">容器名稱</param>
  115. /// <param name="blobContainerSasPermissions">權限(可多選)Flags</param>
  116. /// <param name="name"></param>
  117. /// <returns></returns>
  118. public (string uri, string sas) GetBlobContainerSAS(string containerName, BlobContainerSasPermissions blobContainerSasPermissions, string name = "Default")
  119. {
  120. try
  121. {
  122. var keys = Utils.ParseConnectionString(_optionsMonitor.Get(name).StorageAccountConnectionString);
  123. var accountname = keys["AccountName"];
  124. var accountkey = keys["AccountKey"];
  125. var endpoint = keys["EndpointSuffix"];
  126. var blobSasBuilder = new BlobSasBuilder
  127. {
  128. StartsOn = DateTimeOffset.UtcNow.Subtract(new TimeSpan(0, 15, 0)),
  129. ExpiresOn = DateTimeOffset.UtcNow.Add(new TimeSpan(1, 0, 15, 0)),
  130. BlobContainerName = containerName.ToLower()
  131. };
  132. blobSasBuilder.SetPermissions(blobContainerSasPermissions);
  133. var sskc = new StorageSharedKeyCredential(accountname, accountkey);
  134. BlobSasQueryParameters sasQueryParameters = blobSasBuilder.ToSasQueryParameters(sskc);
  135. UriBuilder fullUri = new UriBuilder()
  136. {
  137. Scheme = "https",
  138. Host = $"{accountname}.blob.{endpoint}",
  139. Path = containerName.ToLower()
  140. //Query = sasQueryParameters.ToString()
  141. };
  142. return (fullUri.Uri.ToString(), sasQueryParameters.ToString());
  143. }
  144. catch
  145. {
  146. return (null, null);
  147. }
  148. }
  149. /// <summary>
  150. /// 取得Blob Container SAS (有效期預設一天)
  151. /// </summary>
  152. /// <param name="containerName">容器名稱</param>
  153. /// <param name="blobName"></param>
  154. /// <param name="blobContainerSasPermissions"></param>
  155. /// <param name="name"></param>
  156. /// <returns></returns>
  157. public BlobAuth GetContainerSasUri(BlobSas blobSas, bool isRead, string name = "Default")
  158. {
  159. try
  160. {
  161. string containerName = null;
  162. var keys = Utils.ParseConnectionString(_optionsMonitor.Get(name).StorageAccountConnectionString);
  163. var accountname = keys["AccountName"];
  164. var accountkey = keys["AccountKey"];
  165. var endpoint = keys["EndpointSuffix"];
  166. if (blobSas.role.Equals("system"))
  167. {
  168. containerName = "teammodelos";
  169. }
  170. else
  171. {
  172. containerName = blobSas.name.ToLower().Replace("#", "");
  173. }
  174. DateTimeOffset dateTime = DateTimeOffset.UtcNow.Add(new TimeSpan(1, 0, 15, 0));
  175. long time = dateTime.ToUnixTimeMilliseconds();
  176. var blobSasBuilder = new BlobSasBuilder
  177. {
  178. StartsOn = DateTimeOffset.UtcNow.Subtract(new TimeSpan(0, 15, 0)),
  179. ExpiresOn = dateTime,
  180. BlobContainerName = containerName
  181. };
  182. BlobContainerSasPermissions blobContainerSasPermissions = BlobContainerSasPermissions.Read;
  183. if (isRead)
  184. {
  185. blobContainerSasPermissions = BlobContainerSasPermissions.Read | BlobContainerSasPermissions.List;
  186. }
  187. else
  188. {
  189. blobContainerSasPermissions = BlobContainerSasPermissions.Delete | BlobContainerSasPermissions.Write | BlobContainerSasPermissions.Create | BlobContainerSasPermissions.Read | BlobContainerSasPermissions.List;
  190. }
  191. blobSasBuilder.SetPermissions(blobContainerSasPermissions);
  192. var sskc = new StorageSharedKeyCredential(accountname, accountkey);
  193. BlobSasQueryParameters sasQueryParameters = blobSasBuilder.ToSasQueryParameters(sskc);
  194. UriBuilder fullUri = new UriBuilder()
  195. {
  196. Scheme = "https",
  197. Host = $"{accountname}.blob.{endpoint}",
  198. Path = containerName
  199. //Query = sasQueryParameters.ToString()
  200. };
  201. return new BlobAuth { url = fullUri.Uri.ToString().Replace(fullUri.Uri.LocalPath, ""), sas = sasQueryParameters.ToString(), timeout = time, name = containerName };
  202. // return (fullUri.Uri.ToString(), sasQueryParameters.ToString());
  203. }
  204. catch
  205. {
  206. return null;
  207. }
  208. }
  209. /// <summary>
  210. /// 取得Blob SAS (有效期預設一天)
  211. /// </summary>
  212. /// <param name="containerName">容器名稱</param>
  213. /// <param name="blobName"></param>
  214. /// <param name="blobSasPermissions">權限(可多選)Flags</param>
  215. /// <param name="name"></param>
  216. /// <returns></returns>
  217. public string GetBlobSAS(string containerName, string blobName, BlobSasPermissions blobSasPermissions, string name = "Default")
  218. {
  219. try
  220. {
  221. var keys = Utils.ParseConnectionString(_optionsMonitor.Get(name).StorageAccountConnectionString);
  222. var accountname = keys["AccountName"];
  223. var accountkey = keys["AccountKey"];
  224. var endpoint = keys["EndpointSuffix"];
  225. var blobSasBuilder = new BlobSasBuilder
  226. {
  227. StartsOn = DateTimeOffset.UtcNow.Subtract(new TimeSpan(0, 15, 0)),
  228. ExpiresOn = DateTimeOffset.UtcNow.Add(new TimeSpan(1, 0, 15, 0)),
  229. BlobContainerName = containerName.ToLower(),
  230. BlobName = blobName
  231. };
  232. blobSasBuilder.SetPermissions(blobSasPermissions);
  233. var sskc = new StorageSharedKeyCredential(accountname, accountkey);
  234. BlobSasQueryParameters sasQueryParameters = blobSasBuilder.ToSasQueryParameters(sskc);
  235. UriBuilder fullUri = new UriBuilder()
  236. {
  237. Scheme = "https",
  238. Host = $"{accountname}.blob.{endpoint}",
  239. Path = $"{containerName.ToLower()}/{blobName}",
  240. Query = sasQueryParameters.ToString()
  241. };
  242. return fullUri.Uri.ToString();
  243. }
  244. catch
  245. {
  246. return null;
  247. }
  248. }
  249. /// <summary>
  250. /// 取得Blob SAS (有效期預設一天)
  251. /// </summary>
  252. /// <param name="containerName">容器名稱</param>
  253. /// <param name="blobName"></param>
  254. /// <param name="blobSasPermissions"></param>
  255. /// <param name="name"></param>
  256. /// <returns></returns>
  257. public BlobAuth GetBlobSasUriRead(string containerName, string blobName, string name = "Default")
  258. {
  259. try
  260. {
  261. var keys = Utils.ParseConnectionString(_optionsMonitor.Get(name).StorageAccountConnectionString);
  262. var accountname = keys["AccountName"];
  263. var accountkey = keys["AccountKey"];
  264. var endpoint = keys["EndpointSuffix"];
  265. DateTimeOffset dateTime = DateTimeOffset.UtcNow.Add(new TimeSpan(365 * 99, 0, 15, 0));
  266. long time = dateTime.ToUnixTimeMilliseconds();
  267. var blobSasBuilder = new BlobSasBuilder
  268. {
  269. StartsOn = DateTimeOffset.UtcNow.Subtract(new TimeSpan(0, 15, 0)),
  270. ExpiresOn = dateTime,
  271. BlobContainerName = containerName.ToLower(),
  272. BlobName = blobName
  273. };
  274. blobSasBuilder.SetPermissions(BlobSasPermissions.Read);
  275. var sskc = new StorageSharedKeyCredential(accountname, accountkey);
  276. BlobSasQueryParameters sasQueryParameters = blobSasBuilder.ToSasQueryParameters(sskc);
  277. UriBuilder fullUri = new UriBuilder()
  278. {
  279. Scheme = "https",
  280. Host = $"{accountname}.blob.{endpoint}",
  281. Path = $"{containerName.ToLower()}/{blobName}",
  282. Query = sasQueryParameters.ToString()
  283. };
  284. return new BlobAuth { url = fullUri.Uri.ToString(), sas = sasQueryParameters.ToString(), timeout = time };
  285. // return fullUri.Uri.ToString();
  286. }
  287. catch
  288. {
  289. return null;
  290. }
  291. }
  292. public CloudTableClient GetCloudTableClient(string name = "Default")
  293. {
  294. try
  295. {
  296. var options = _optionsMonitor.Get(name);
  297. CloudStorageAccount storageAccount = CloudStorageAccount.Parse(options.StorageAccountConnectionString);
  298. return storageAccount.CreateCloudTableClient();
  299. }
  300. catch (OptionsValidationException e)
  301. {
  302. _logger?.LogWarning(e, e.Message);
  303. return null;
  304. }
  305. }
  306. /// <summary>
  307. /// 可讓您管理儲存體帳戶中的所有佇列
  308. /// </summary>
  309. /// <param name="name"></param>
  310. /// <returns></returns>
  311. public QueueServiceClient GetQueueServiceClient(string name = "Default")
  312. {
  313. try
  314. {
  315. var options = _optionsMonitor.Get(name);
  316. return new QueueServiceClient(options.StorageAccountConnectionString);
  317. }
  318. catch (OptionsValidationException e)
  319. {
  320. _logger?.LogWarning(e, e.Message);
  321. return null;
  322. }
  323. }
  324. /// <summary>
  325. /// 可讓您管理和操作個別佇列及其訊息
  326. /// </summary>
  327. /// <param name="name"></param>
  328. /// <param name="queueName"></param>
  329. /// <returns></returns>
  330. public QueueClient GetQueueClient(string queueName, string name = "Default")
  331. {
  332. if (name == null) throw new ArgumentNullException(nameof(name));
  333. try
  334. {
  335. var options = _optionsMonitor.Get(name);
  336. return new QueueClient(options.StorageAccountConnectionString, queueName);
  337. }
  338. catch (OptionsValidationException e)
  339. {
  340. _logger?.LogWarning(e, e.Message);
  341. return null;
  342. }
  343. }
  344. public async Task<CloudTable> InitializeTable<T>()
  345. {
  346. string TableName = GetTableSpace<T>();
  347. CloudTable cloudTable = GetCloudTableClient().GetTableReference(TableName);
  348. await cloudTable.CreateIfNotExistsAsync();
  349. return cloudTable;
  350. }
  351. private string GetTableSpace<T>()
  352. {
  353. Type type = typeof(T);
  354. string Name = type.Name;
  355. object[] attributes = type.GetCustomAttributes(true);
  356. foreach (object attribute in attributes) //2.通过映射,找到成员属性上关联的特性类实例,
  357. {
  358. if (attribute is TableNameAttribute tableSpace)
  359. {
  360. Name = tableSpace.Name;
  361. }
  362. }
  363. return Name;
  364. }
  365. }
  366. }