AzureStorageFactory.cs 16 KB

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