55
66package com .microsoft .azure .toolkit .intellij .sqlserver .task ;
77
8- import com .microsoft .azure .management .Azure ;
9- import com .microsoft .azure .management .resources .ResourceGroup ;
108import com .microsoft .azure .toolkit .intellij .common .Draft ;
9+ import com .microsoft .azure .toolkit .lib .Azure ;
1110import com .microsoft .azure .toolkit .lib .common .model .Region ;
1211import com .microsoft .azure .toolkit .lib .common .operation .AzureOperation ;
12+ import com .microsoft .azure .toolkit .lib .resource .AzureGroup ;
13+ import com .microsoft .azure .toolkit .lib .resource .ResourceGroupEntity ;
1314import com .microsoft .azure .toolkit .lib .sqlserver .SqlServerConfig ;
1415import com .microsoft .azure .toolkit .lib .sqlserver .model .SqlServerEntity ;
1516import com .microsoft .azure .toolkit .lib .sqlserver .service .AzureSqlServer ;
1617import com .microsoft .azure .toolkit .lib .sqlserver .service .ISqlServer ;
1718import com .microsoft .azuretools .ActionConstants ;
18- import com .microsoft .azuretools .authmanage .AuthMethodManager ;
1919import com .microsoft .azuretools .telemetry .TelemetryConstants ;
2020import com .microsoft .azuretools .telemetrywrapper .ErrorType ;
2121import com .microsoft .azuretools .telemetrywrapper .EventType ;
@@ -49,16 +49,13 @@ public ISqlServer execute() {
4949 EventUtil .logEvent (EventType .info , operation , Collections .singletonMap (TelemetryConstants .SUBSCRIPTIONID , subscriptionId ));
5050 // create resource group if necessary.
5151 if (config .getResourceGroup () instanceof Draft ) {
52- Azure azure = AuthMethodManager . getInstance (). getAzureClient ( subscriptionId );
53- ResourceGroup newResourceGroup = azure . resourceGroups (). define (config .getResourceGroup ().name ()). withRegion ( config .getRegion ().getName ()). create ( );
52+ ResourceGroupEntity newResourceGroup = Azure . az ( AzureGroup . class )
53+ . subscription ( subscriptionId ). create (config .getResourceGroup ().name (), config .getRegion ().getName ());
5454 config .setResourceGroup (newResourceGroup );
5555 }
5656 // create sql server
57- SqlServerEntity entity = SqlServerEntity .builder ().name (config .getServerName ()).subscriptionId (config .getSubscription ().subscriptionId ())
58- .resourceGroup (config .getResourceGroup ().name ()).region (Region .fromName (config .getRegion ().getName ())).administratorLoginName (config .getAdminUsername ())
59- .enableAccessFromAzureServices (config .isAllowAccessFromAzureServices ()).enableAccessFromLocalMachine (config .isAllowAccessFromLocalMachine ())
60- .build ();
61- return com .microsoft .azure .toolkit .lib .Azure .az (AzureSqlServer .class ).sqlServer (entity ).create ()
57+ SqlServerEntity entity = this .fromConfig (this .config );
58+ return Azure .az (AzureSqlServer .class ).sqlServer (entity ).create ()
6259 .withAdministratorLoginPassword (String .valueOf (config .getPassword ()))
6360 .commit ();
6461 } catch (final RuntimeException e ) {
@@ -69,4 +66,11 @@ public ISqlServer execute() {
6966 }
7067 }
7168
69+ private SqlServerEntity fromConfig (SqlServerConfig config ) {
70+ return SqlServerEntity .builder ().name (config .getServerName ()).subscriptionId (config .getSubscription ().subscriptionId ())
71+ .resourceGroup (config .getResourceGroup ().name ()).region (Region .fromName (config .getRegion ().getName ())).administratorLoginName (config .getAdminUsername ())
72+ .enableAccessFromAzureServices (config .isAllowAccessFromAzureServices ()).enableAccessFromLocalMachine (config .isAllowAccessFromLocalMachine ())
73+ .build ();
74+ }
75+
7276}
0 commit comments