Put usage server under Spring

kelveny · kelveny · commit 1450547987e9 · 2013-01-25T23:00:55.000-08:00
diff --git a/engine/storage/volume/test/org/apache/cloudstack/storage/volume/test/TestConfiguration.java b/engine/storage/volume/test/org/apache/cloudstack/storage/volume/test/TestConfiguration.java
@@ -27,7 +27,6 @@
 import com.cloud.dc.dao.ClusterDao;
 import com.cloud.dc.dao.ClusterDaoImpl;
 
-
 @Configuration
 public class TestConfiguration {
     @Bean
diff --git a/server/src/com/cloud/service/dao/ServiceOfferingDaoImpl.java b/server/src/com/cloud/service/dao/ServiceOfferingDaoImpl.java
@@ -43,7 +43,7 @@ public class ServiceOfferingDaoImpl extends GenericDaoBase<ServiceOfferingVO, Lo
     protected final SearchBuilder<ServiceOfferingVO> ServiceOfferingsByKeywordSearch;
     protected final SearchBuilder<ServiceOfferingVO> PublicServiceOfferingSearch;
     
-    protected ServiceOfferingDaoImpl() {
+    public ServiceOfferingDaoImpl() {
         super();
         
         UniqueNameSearch = createSearchBuilder();
diff --git a/server/src/com/cloud/tags/dao/ResourceTagsDaoImpl.java b/server/src/com/cloud/tags/dao/ResourceTagsDaoImpl.java
@@ -34,7 +34,7 @@
 public class ResourceTagsDaoImpl extends GenericDaoBase<ResourceTagVO, Long> implements ResourceTagDao{
     final SearchBuilder<ResourceTagVO> AllFieldsSearch;
     
-    protected ResourceTagsDaoImpl() {
+    public ResourceTagsDaoImpl() {
         AllFieldsSearch = createSearchBuilder();
         AllFieldsSearch.and("resourceId", AllFieldsSearch.entity().getResourceId(), Op.EQ);
         AllFieldsSearch.and("uuid", AllFieldsSearch.entity().getResourceUuid(), Op.EQ);
diff --git a/server/src/com/cloud/user/dao/AccountDaoImpl.java b/server/src/com/cloud/user/dao/AccountDaoImpl.java
@@ -55,7 +55,7 @@ public class AccountDaoImpl extends GenericDaoBase<AccountVO, Long> implements A
     protected final SearchBuilder<AccountVO> CleanupForDisabledAccountsSearch;
     protected final SearchBuilder<AccountVO> NonProjectAccountSearch;
     
-    protected AccountDaoImpl() {
+    public AccountDaoImpl() {
         AllFieldsSearch = createSearchBuilder();
         AllFieldsSearch.and("accountName", AllFieldsSearch.entity().getAccountName(), SearchCriteria.Op.EQ);
         AllFieldsSearch.and("domainId", AllFieldsSearch.entity().getDomainId(), SearchCriteria.Op.EQ);
diff --git a/server/src/com/cloud/user/dao/UserAccountDaoImpl.java b/server/src/com/cloud/user/dao/UserAccountDaoImpl.java
@@ -32,7 +32,7 @@ public class UserAccountDaoImpl extends GenericDaoBase<UserAccountVO, Long> impl
 	
 	protected final SearchBuilder<UserAccountVO> userAccountSearch;
 	
-    protected UserAccountDaoImpl() {
+    public UserAccountDaoImpl() {
     	userAccountSearch = createSearchBuilder();
     	userAccountSearch.and("apiKey", userAccountSearch.entity().getApiKey(), SearchCriteria.Op.EQ);
         userAccountSearch.done();
diff --git a/server/src/com/cloud/vm/dao/NicDaoImpl.java b/server/src/com/cloud/vm/dao/NicDaoImpl.java
@@ -42,7 +42,7 @@ public class NicDaoImpl extends GenericDaoBase<NicVO, Long> implements NicDao {
     final GenericSearchBuilder<NicVO, Integer> CountBy;
 
     
-    protected NicDaoImpl() {
+    public NicDaoImpl() {
         super();
         
         AllFieldsSearch = createSearchBuilder();
diff --git a/server/src/com/cloud/vm/dao/UserVmDaoImpl.java b/server/src/com/cloud/vm/dao/UserVmDaoImpl.java
@@ -116,7 +116,7 @@ public class UserVmDaoImpl extends GenericDaoBase<UserVmVO, Long> implements Use
     @Inject protected UserVmDetailsDao _detailsDao;
     @Inject protected NicDao _nicDao;
     
-    protected UserVmDaoImpl() {
+    public UserVmDaoImpl() {
     }
     
     @PostConstruct
diff --git a/server/src/com/cloud/vm/dao/UserVmDetailsDaoImpl.java b/server/src/com/cloud/vm/dao/UserVmDetailsDaoImpl.java
@@ -36,7 +36,7 @@ public class UserVmDetailsDaoImpl extends GenericDaoBase<UserVmDetailVO, Long> i
     protected final SearchBuilder<UserVmDetailVO> VmSearch;
     protected final SearchBuilder<UserVmDetailVO> DetailSearch;
 
-	protected UserVmDetailsDaoImpl() {
+	public UserVmDetailsDaoImpl() {
 		VmSearch = createSearchBuilder();
 		VmSearch.and("vmId", VmSearch.entity().getVmId(), SearchCriteria.Op.EQ);
         VmSearch.done();
diff --git a/usage/conf/usageApplicationContext.xml.in b/usage/conf/usageApplicationContext.xml.in
@@ -0,0 +1,53 @@
+<!--
+  Licensed to the Apache Software Foundation (ASF) under one
+  or more contributor license agreements. See the NOTICE file
+  distributed with this work for additional information
+  regarding copyright ownership. The ASF licenses this file
+  to you under the Apache License, Version 2.0 (the
+  "License"); you may not use this file except in compliance
+  with the License. You may obtain a copy of the License at
+
+  http://www.apache.org/licenses/LICENSE-2.0
+
+  Unless required by applicable law or agreed to in writing,
+  software distributed under the License is distributed on an
+  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+  KIND, either express or implied. See the License for the
+  specific language governing permissions and limitations
+  under the License.
+-->
+<beans xmlns="http://www.springframework.org/schema/beans"
+  xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" 
+  xmlns:context="http://www.springframework.org/schema/context"
+  xmlns:tx="http://www.springframework.org/schema/tx" 
+  xmlns:aop="http://www.springframework.org/schema/aop"
+  xsi:schemaLocation="http://www.springframework.org/schema/beans
+                      http://www.springframework.org/schema/beans/spring-beans-3.0.xsd
+                      http://www.springframework.org/schema/tx 
+                      http://www.springframework.org/schema/tx/spring-tx-3.0.xsd
+                      http://www.springframework.org/schema/aop
+                      http://www.springframework.org/schema/aop/spring-aop-3.0.xsd
+                      http://www.springframework.org/schema/context
+                      http://www.springframework.org/schema/context/spring-context-3.0.xsd">                     
+
+  <context:annotation-config />
+  <context:component-scan base-package="com.cloud.usage" />
+  
+  <!--
+    @DB support
+  -->
+  <aop:config proxy-target-class="true">
+    <aop:aspect id="dbContextBuilder" ref="transactionContextBuilder">
+        <aop:pointcut id="captureAnyMethod"
+            expression="execution(* *(..))" 
+        />
+        <aop:around pointcut-ref="captureAnyMethod" method="AroundAnyMethod"/> 
+    </aop:aspect>
+
+  </aop:config>
+  
+  <bean id="transactionContextBuilder" class="com.cloud.utils.db.TransactionContextBuilder" />
+  <bean id="ComponentContext" class="com.cloud.utils.component.ComponentContext" />
+  <bean id="UsageServerConfig" class="com.cloud.usage.UsageServerComponentConfig" />
+ 
+</beans>
diff --git a/usage/src/com/cloud/usage/UsageManagerImpl.java b/usage/src/com/cloud/usage/UsageManagerImpl.java
@@ -144,7 +144,8 @@ public boolean configure(String name, Map<String, Object> params) throws Configu
         final Class<?> c = UsageServer.class;
         m_version = c.getPackage().getImplementationVersion();
         if (m_version == null) {
-            throw new CloudRuntimeException("Unable to find the implementation version of this usage server");
+            // TODO
+        	// throw new CloudRuntimeException("Unable to find the implementation version of this usage server");
         }
 
         if (s_logger.isInfoEnabled()) {
diff --git a/usage/src/com/cloud/usage/UsageServer.java b/usage/src/com/cloud/usage/UsageServer.java
@@ -16,31 +16,47 @@
 // under the License.
 package com.cloud.usage;
 
-import javax.inject.Inject;
+import java.io.File;
 
 import org.apache.log4j.Logger;
+import org.apache.log4j.PropertyConfigurator;
+import org.apache.log4j.xml.DOMConfigurator;
+import org.springframework.context.ApplicationContext;
+import org.springframework.context.support.ClassPathXmlApplicationContext;
 
-
+import com.cloud.utils.PropertiesUtil;
+import com.cloud.utils.component.ComponentContext;
 
 public class UsageServer {
     private static final Logger s_logger = Logger.getLogger(UsageServer.class.getName());
     public static final String Name = "usage-server";
-
-    @Inject UsageManager mgr; 
+    
+    UsageManager mgr; 
+    
     /**
      * @param args
      */
     public static void main(String[] args) {
+    	initLog4j();
+    	ApplicationContext appContext = new ClassPathXmlApplicationContext("usageApplicationContext.xml");
+	    
         UsageServer usage = new UsageServer();
         usage.init(args);
-        usage.start();
+        usage.start(appContext);
     }
 
     public void init(String[] args) {
-
     }
 
-    public void start() {
+    public void start(ApplicationContext appContext) {
+    	try {
+    		ComponentContext.initComponentsLifeCycle();
+    	} catch(Exception e) {
+    		e.printStackTrace();
+    	}
+    	
+    	mgr = appContext.getBean(UsageManager.class);
+    	
         if (mgr != null) {
             if (s_logger.isInfoEnabled()) {
                 s_logger.info("UsageServer ready...");
@@ -55,4 +71,18 @@ public void stop() {
     public void destroy() {
 
     }
+
+    static private void initLog4j() {
+    	File file = PropertiesUtil.findConfigFile("log4j-cloud.xml");
+    	if (file != null) {
+        s_logger.info("log4j configuration found at " + file.getAbsolutePath());
+        DOMConfigurator.configureAndWatch(file.getAbsolutePath());
+	    } else {
+	        file = PropertiesUtil.findConfigFile("log4j-cloud.properties");
+	        if (file != null) {
+	            s_logger.info("log4j configuration found at " + file.getAbsolutePath());
+	            PropertyConfigurator.configureAndWatch(file.getAbsolutePath());
+	        }
+	    }
+   }
 }
diff --git a/usage/src/com/cloud/usage/UsageServerComponentConfig.java b/usage/src/com/cloud/usage/UsageServerComponentConfig.java
@@ -0,0 +1,180 @@
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// the License.  You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+package com.cloud.usage;
+
+import org.springframework.context.annotation.Bean;
+
+import org.springframework.context.annotation.Configuration;
+
+import com.cloud.cluster.agentlb.dao.HostTransferMapDao;
+import com.cloud.cluster.agentlb.dao.HostTransferMapDaoImpl;
+import com.cloud.dc.dao.*;
+import com.cloud.service.dao.ServiceOfferingDaoImpl;
+import com.cloud.vm.dao.*;
+import com.cloud.network.dao.*;
+import com.cloud.host.dao.*;
+
+import com.cloud.utils.crypt.EncryptionSecretKeyChecker;  
+import com.cloud.vm.dao.VMInstanceDaoImpl; 
+import com.cloud.vm.dao.UserVmDaoImpl;
+import com.cloud.event.dao.EventDaoImpl; 
+import com.cloud.user.dao.UserStatisticsDaoImpl; 
+import com.cloud.network.dao.IPAddressDaoImpl; 
+import com.cloud.domain.dao.DomainDaoImpl; 
+import com.cloud.user.dao.AccountDaoImpl; 
+import com.cloud.user.dao.UserAccountDaoImpl; 
+import com.cloud.configuration.dao.ConfigurationDaoImpl; 
+import com.cloud.alert.dao.AlertDaoImpl;
+import com.cloud.event.dao.UsageEventDaoImpl; 
+import com.cloud.service.dao.ServiceOfferingDao;
+import com.cloud.event.dao.EventDao;
+import com.cloud.user.dao.UserStatisticsDao;
+import com.cloud.domain.dao.DomainDao;
+import com.cloud.user.dao.*;
+import com.cloud.configuration.dao.ConfigurationDao;
+import com.cloud.alert.dao.AlertDao;
+import com.cloud.event.dao.UsageEventDao;
+import com.cloud.tags.dao.*;
+
+@Configuration
+public class UsageServerComponentConfig {
+
+	@Bean
+	public HostTransferMapDao HostTransferDao() {
+		return new HostTransferMapDaoImpl();
+	}
+
+	@Bean
+	public ClusterDao ClusterDao() {
+		return new ClusterDaoImpl();
+	}
+
+	@Bean
+	public HostPodDao HostPodDao() {
+		return new HostPodDaoImpl();
+	}
+	
+	@Bean
+	public UserVmDetailsDao UserVmDetailsDao() {
+		return new UserVmDetailsDaoImpl();
+	}
+	
+	@Bean
+	public VlanDaoImpl VlanDaoImpl() {
+		return new VlanDaoImpl();
+	}
+	
+	@Bean
+	public PodVlanMapDao PodVlanMapDao() {
+		return new PodVlanMapDaoImpl();
+	}
+	
+	@Bean
+	public AccountVlanMapDao AccountVlanMapDao() {
+		return new AccountVlanMapDaoImpl();
+	}
+
+	@Bean
+	public EncryptionSecretKeyChecker EncryptionSecretKeyChecker() {
+		return new EncryptionSecretKeyChecker();
+	}
+	
+	@Bean
+	public VMInstanceDao VmInstanceDao() {
+		return new VMInstanceDaoImpl();
+	}
+
+	@Bean
+	public UserVmDao UserVmDao() {
+		return new UserVmDaoImpl();
+	}
+
+	@Bean
+	public ServiceOfferingDao ServiceOfferingDao() {
+		return new ServiceOfferingDaoImpl();
+	}
+	
+	@Bean
+	public EventDao EventDao() {
+		return new EventDaoImpl();
+	}
+
+	@Bean
+	public UserStatisticsDao UserStatisticsDao() {
+		return new UserStatisticsDaoImpl();
+	}
+
+	@Bean
+	public IPAddressDao IPAddressDao() {
+		return new IPAddressDaoImpl();
+	}
+	
+	@Bean
+	public DomainDao DomainDao() {
+		return new DomainDaoImpl();
+	}
+
+	@Bean
+	public AccountDao AccountDao() {
+		return new AccountDaoImpl();
+	}
+
+	@Bean
+	public UserAccountDao UserAccountDao() {
+		return new UserAccountDaoImpl();
+	}
+
+	@Bean
+	public ConfigurationDao ConfigurationDao() {
+		return new ConfigurationDaoImpl();
+	}
+
+	@Bean
+	public AlertDao AlertDao() {
+		return new AlertDaoImpl();
+	}
+
+	@Bean
+	public UsageEventDao UsageEventDao() {
+		return new UsageEventDaoImpl();
+	}
+
+	@Bean
+	public ResourceTagsDaoImpl ResourceTagsDaoImpl() {
+		return new ResourceTagsDaoImpl();
+	}
+
+	@Bean
+	public NicDao NicDao() {
+		return new NicDaoImpl();
+	}
+
+	@Bean
+	public HostDao HostDao() {
+		return new HostDaoImpl();
+	}
+
+	@Bean
+	public HostDetailsDao HostDetailsDao() {
+		return new HostDetailsDaoImpl();
+	}
+
+	@Bean
+	public HostTagsDao HostTagsDao() {
+		return new HostTagsDaoImpl();
+	}
+}

Original file line number	Diff line number	Diff line change
`@@ -116,7 +116,7 @@ public class UserVmDaoImpl extends GenericDaoBase<UserVmVO, Long> implements Use`
`116`	`116`	`@Inject protected UserVmDetailsDao _detailsDao;`
`117`	`117`	`@Inject protected NicDao _nicDao;`
`118`	`118`
`119`		`- protected UserVmDaoImpl() {`
	`119`	`+ public UserVmDaoImpl() {`
`120`	`120`	`}`
`121`	`121`
`122`	`122`	`@PostConstruct`
Original file line number	Diff line number	Diff line change
`@@ -144,7 +144,8 @@ public boolean configure(String name, Map<String, Object> params) throws Configu`
`144`	`144`	`final Class<?> c = UsageServer.class;`
`145`	`145`	`m_version = c.getPackage().getImplementationVersion();`
`146`	`146`	`if (m_version == null) {`
`147`		`- throw new CloudRuntimeException("Unable to find the implementation version of this usage server");`
	`147`	`+ // TODO`
	`148`	`+ // throw new CloudRuntimeException("Unable to find the implementation version of this usage server");`
`148`	`149`	`}`
`149`	`150`
`150`	`151`	`if (s_logger.isInfoEnabled()) {`