@@ -829,21 +829,6 @@ public VMSnapshotResponse createVMSnapshotResponse(VMSnapshot vmSnapshot) {
829829 }
830830 }
831831 populateOwner (vmSnapshotResponse , vmSnapshot );
832- Project project = ApiDBUtils .findProjectByProjectAccountId (vmSnapshot .getAccountId ());
833- if (project != null ) {
834- vmSnapshotResponse .setProjectId (project .getUuid ());
835- vmSnapshotResponse .setProjectName (project .getName ());
836- }
837- Account account = ApiDBUtils .findAccountById (vmSnapshot .getAccountId ());
838- if (account != null ) {
839- vmSnapshotResponse .setAccountName (account .getAccountName ());
840- }
841- DomainVO domain = ApiDBUtils .findDomainById (vmSnapshot .getDomainId ());
842- if (domain != null ) {
843- vmSnapshotResponse .setDomainId (domain .getUuid ());
844- vmSnapshotResponse .setDomainName (domain .getName ());
845- vmSnapshotResponse .setDomainPath (domain .getPath ());
846- }
847832
848833 List <? extends ResourceTag > tags = _resourceTagDao .listBy (vmSnapshot .getId (), ResourceObjectType .VMSnapshot );
849834 List <ResourceTagResponse > tagResponses = new ArrayList <ResourceTagResponse >();
@@ -2357,18 +2342,7 @@ public SecurityGroupResponse createSecurityGroupResponseFromSecurityGroupRule(Li
23572342 response .setName (securityGroup .getName ());
23582343 response .setDescription (securityGroup .getDescription ());
23592344
2360- Account account = securiytGroupAccounts .get (securityGroup .getAccountId ());
2361-
2362- if (securityGroup .getAccountType () == Account .Type .PROJECT ) {
2363- response .setProjectId (securityGroup .getProjectUuid ());
2364- response .setProjectName (securityGroup .getProjectName ());
2365- } else {
2366- response .setAccountName (securityGroup .getAccountName ());
2367- }
2368-
2369- response .setDomainId (securityGroup .getDomainUuid ());
2370- response .setDomainName (securityGroup .getDomainName ());
2371- response .setDomainPath (securityGroup .getDomainPath ());
2345+ populateOwner (response , securityGroup );
23722346
23732347 for (SecurityRule securityRule : securityRules ) {
23742348 SecurityGroupRuleResponse securityGroupData = new SecurityGroupRuleResponse ();
@@ -2765,29 +2739,18 @@ public NetworkResponse createNetworkResponse(ResponseView view, Network network)
27652739 // get domain from network_domain table
27662740 Pair <Long , Boolean > domainNetworkDetails = ApiDBUtils .getDomainNetworkDetails (network .getId ());
27672741 if (domainNetworkDetails .first () != null ) {
2768- Domain domain = ApiDBUtils .findDomainById (domainNetworkDetails .first ());
2769- if (domain != null ) {
2770- response .setDomainId (domain .getUuid ());
2771- response .setDomainPath (getPrettyDomainPath (domain .getPath ()));
2772- }
2742+ populateDomain (response , domainNetworkDetails .first ());
27732743 }
27742744 response .setSubdomainAccess (domainNetworkDetails .second ());
27752745 }
27762746
27772747 Long dedicatedDomainId = ApiDBUtils .getDedicatedNetworkDomain (network .getId ());
27782748 if (dedicatedDomainId != null ) {
2779- Domain domain = ApiDBUtils .findDomainById (dedicatedDomainId );
2780- if (domain != null ) {
2781- response .setDomainId (domain .getUuid ());
2782- response .setDomainName (domain .getName ());
2783- response .setDomainPath (domain .getPath ());
2784- }
2785-
2749+ populateDomain (response , dedicatedDomainId );
27862750 }
27872751
27882752 response .setSpecifyIpRanges (network .getSpecifyIpRanges ());
27892753
2790-
27912754 setVpcIdInResponse (network .getVpcId (), response ::setVpcId , response ::setVpcName );
27922755
27932756 setResponseAssociatedNetworkInformation (response , network .getId ());
@@ -3049,14 +3012,10 @@ private void populateOwner(ControlledEntityResponse response, ControlledEntity o
30493012 } else {
30503013 response .setAccountName (account .getAccountName ());
30513014 }
3052-
3053- Domain domain = ApiDBUtils .findDomainById (object .getDomainId ());
3054- response .setDomainId (domain .getUuid ());
3055- response .setDomainName (domain .getName ());
3056- response .setDomainPath (getPrettyDomainPath (domain .getPath ()));
3015+ populateDomain (response , object .getDomainId ());
30573016 }
30583017
3059- private void populateOwner (ControlledViewEntityResponse response , ControlledEntity object ) {
3018+ public static void populateOwner (ControlledViewEntityResponse response , ControlledEntity object ) {
30603019 Account account = ApiDBUtils .findAccountById (object .getAccountId ());
30613020
30623021 if (account .getType () == Account .Type .PROJECT ) {
@@ -3068,10 +3027,7 @@ private void populateOwner(ControlledViewEntityResponse response, ControlledEnti
30683027 response .setAccountName (account .getAccountName ());
30693028 }
30703029
3071- Domain domain = ApiDBUtils .findDomainById (object .getDomainId ());
3072- response .setDomainId (domain .getUuid ());
3073- response .setDomainName (domain .getName ());
3074- response .setDomainPath (domain .getPath ());
3030+ populateDomain (response , object .getDomainId ());
30753031 }
30763032
30773033 public static void populateOwner (ControlledViewEntityResponse response , ControlledViewEntity object ) {
@@ -3085,7 +3041,7 @@ public static void populateOwner(ControlledViewEntityResponse response, Controll
30853041
30863042 response .setDomainId (object .getDomainUuid ());
30873043 response .setDomainName (object .getDomainName ());
3088- response .setDomainPath (object .getDomainPath ());
3044+ response .setDomainPath (getPrettyDomainPath ( object .getDomainPath () ));
30893045 }
30903046
30913047 private void populateAccount (ControlledEntityResponse response , long accountId ) {
@@ -3109,10 +3065,22 @@ private void populateAccount(ControlledEntityResponse response, long accountId)
31093065
31103066 private void populateDomain (ControlledEntityResponse response , long domainId ) {
31113067 Domain domain = ApiDBUtils .findDomainById (domainId );
3068+ if (domain == null ) {
3069+ return ;
3070+ }
3071+ response .setDomainId (domain .getUuid ());
3072+ response .setDomainName (domain .getName ());
3073+ response .setDomainPath (getPrettyDomainPath (domain .getPath ()));
3074+ }
31123075
3076+ private static void populateDomain (ControlledViewEntityResponse response , long domainId ) {
3077+ Domain domain = ApiDBUtils .findDomainById (domainId );
3078+ if (domain == null ) {
3079+ return ;
3080+ }
31133081 response .setDomainId (domain .getUuid ());
31143082 response .setDomainName (domain .getName ());
3115- response .setDomainPath (domain .getPath ());
3083+ response .setDomainPath (getPrettyDomainPath ( domain .getPath () ));
31163084 }
31173085
31183086 @ Override
@@ -4104,12 +4072,7 @@ public UsageRecordResponse createUsageResponse(Usage usageRecord, Map<String, Se
41044072 usageRecResponse .setAccountName (account .getAccountName ());
41054073 }
41064074
4107- Domain domain = ApiDBUtils .findDomainById (usageRecord .getDomainId ());
4108- if (domain != null ) {
4109- usageRecResponse .setDomainId (domain .getUuid ());
4110- usageRecResponse .setDomainName (domain .getName ());
4111- usageRecResponse .setDomainPath (domain .getPath ());
4112- }
4075+ populateDomain (usageRecResponse , account .getDomainId ());
41134076
41144077 if (usageRecord .getZoneId () != null ) {
41154078 DataCenter zone = ApiDBUtils .findZoneById (usageRecord .getZoneId ());
@@ -4896,18 +4859,11 @@ public AffinityGroupResponse createAffinityGroupResponse(AffinityGroup group) {
48964859
48974860 AffinityGroupResponse response = new AffinityGroupResponse ();
48984861
4899- Account account = ApiDBUtils .findAccountById (group .getAccountId ());
49004862 response .setId (group .getUuid ());
4901- response .setAccountName (account .getAccountName ());
49024863 response .setName (group .getName ());
49034864 response .setType (group .getType ());
49044865 response .setDescription (group .getDescription ());
4905- Domain domain = ApiDBUtils .findDomainById (account .getDomainId ());
4906- if (domain != null ) {
4907- response .setDomainId (domain .getUuid ());
4908- response .setDomainName (domain .getName ());
4909- response .setDomainPath (domain .getPath ());
4910- }
4866+ populateOwner (response , group );
49114867
49124868 response .setObjectName ("affinitygroup" );
49134869 return response ;
0 commit comments