BsicData添加公司信息

廖磊
1 parent 89dfc861
Showing 2 changed files with 23 additions and 1 deletions
src/main/java/com/bsth/data/BasicData.java
src/main/java/com/bsth/entity/oil/Ylb.java
@@ -25,6 +25,9 @@ import java.util.concurrent.TimeUnit;
  */
 @Component
 public class BasicData implements CommandLineRunner {
+	
+	//公司代码和公司名对照(K: 公司编码，V:公司名)
+	public static Map<String, String> businessCodeNameMap;
  
     //设备号和车辆自编号 （K: 设备编码 ，V：车辆自编号）
     public static BiMap<String, String> deviceId2NbbmMap;
@@ -107,6 +110,9 @@ public class BasicData implements CommandLineRunner {
  
         @Autowired
         PersonnelRepository personnelRepository;
+        
+        @Autowired
+        BusinessRepository businessRepository;
  
  
         @Override
@@ -132,6 +138,8 @@ public class BasicData implements CommandLineRunner {
                 loadStationRouteInfo();
                 //人员信息
                 loadPersonnelInfo();
+                //公司信息
+                loadBusinessInfo();
                 logger.info("加载基础数据成功！，");
             } catch (Exception e) {
                 logger.error("加载基础数据时出现异常，", e);
@@ -164,6 +172,20 @@ public class BasicData implements CommandLineRunner {
         }
  
         /**
+         * loadBusinessInfo
+         * (公司代码公司名对照)
+         */
+        public void loadBusinessInfo(){
+        	Map<String, String> businessMap=new HashMap<String,String>();
+        	Iterator<Business> busIter=businessRepository.findAll().iterator();
+        	Business t;
+        	while(busIter.hasNext()){
+        		t=busIter.next();
+        		businessMap.put(t.getBusinessCode(), t.getBusinessName());
+        	}
+        	businessCodeNameMap=businessMap;
+        }
+        /**
          * @Title: loadDeviceInfo
          * @Description: TODO(加载设备相关信息)
          */
@@ -291,7 +291,7 @@ public class Ylb {
 	}
  
 	public String getGsname() {
-		return 	BasicData.nbbm2CompanyCodeMap.get(this.nbbm);
+		return 	BasicData.businessCodeNameMap.get(this.ssgsdm);
 	}
  
 	public void setGsname(String gsname) {