增加客户增加供应商操作

git-svn-id: svn+ssh://10.10.101.21/source/platform/platform-b2b@7169 f3bf4e98-0cf0-11e4-a00c-a99a8b9d557d

src/main/java/com/uas/platform/b2b/service/AddPartnerService.java

@@ -4,6 +4,7 @@ import org.springframework.data.domain.Page;
 
 import com.uas.platform.b2b.model.AddPartner;
 import com.uas.platform.b2b.model.SearchFilter;
+import com.uas.platform.b2b.model.Vendor;
 import com.uas.platform.core.model.PageInfo;
 
 public interface AddPartnerService {
@@ -63,4 +64,20 @@ public interface AddPartnerService {
 	 * @return
 	 */
 	public Object getMyMessageTodoCount();
+
+	/**
+	 * 添加为供应商
+	 * 
+	 * @param id
+	 * @return
+	 */
+	public Vendor addSupplier(Long id);
+
+	/**
+	 * 添加为客户
+	 * 
+	 * @param id
+	 * @return
+	 */
+	public Vendor addCust(Long id);
 }

src/main/java/com/uas/platform/b2b/service/impl/AddPartnerServiceImpl.java

@@ -154,4 +154,52 @@ public class AddPartnerServiceImpl implements AddPartnerService {
 				Status.NOT_REPLY.value());
 	}
 
+	@Override
+	public Vendor addSupplier(Long id) {
+		AddPartner adp = addPartnerDao.findOne(id);
+		// 先判断是否为供应商
+		List<Vendor> oldVendor = vendorDao.findByMyEnUUAndVendUU(adp.getClientuu(), adp.getSupplieruu());
+		if (!CollectionUtils.isEmpty(oldVendor)) {
+			throw new IllegalAccessError("该企业已经是您的供应商，请不要重复添加");
+		}
+		Vendor vendor = new Vendor();
+		Enterprise vendenter = enterpriseDao.findEnterpriseByUu(adp.getSupplieruu());
+		Enterprise myenter = enterpriseDao.findEnterpriseByUu(adp.getClientuu());
+		UserBaseInfo venduser = userBaseInfoDao.findByUserUU(adp.getAuditoruu());
+		UserBaseInfo myuser = userBaseInfoDao.findByUserUU(adp.getApplicantuu());
+		vendor.setMyEnterprise(myenter);
+		vendor.setMyEnUU(adp.getClientuu());
+		vendor.setMyUser(myuser);
+		vendor.setMyUserUU(adp.getApplicantuu());
+		vendor.setVendorEnterprise(vendenter);
+		vendor.setVendorUser(venduser);
+		vendor.setVendUserUU(adp.getAuditoruu());
+		vendor.setVendEnUU(adp.getSupplieruu());
+		return vendorDao.save(vendor);
+	}
+
+	@Override
+	public Vendor addCust(Long id) {
+		AddPartner adp = addPartnerDao.findOne(id);
+		// 先判断是否为客户
+		List<Vendor> oldVendor = vendorDao.findByMyEnUUAndVendUU(adp.getSupplieruu(), adp.getClientuu());
+		if (!CollectionUtils.isEmpty(oldVendor)) {
+			throw new IllegalAccessError("该企业已经是您的客户，请不要重复添加");
+		}
+		Vendor vendor = new Vendor();
+		Enterprise vendenter = enterpriseDao.findEnterpriseByUu(adp.getSupplieruu());
+		Enterprise myenter = enterpriseDao.findEnterpriseByUu(adp.getClientuu());
+		UserBaseInfo venduser = userBaseInfoDao.findByUserUU(adp.getAuditoruu());
+		UserBaseInfo myuser = userBaseInfoDao.findByUserUU(adp.getApplicantuu());
+		vendor.setMyEnterprise(vendenter);
+		vendor.setMyEnUU(adp.getSupplieruu());
+		vendor.setMyUser(venduser);
+		vendor.setMyUserUU(adp.getAuditoruu());
+		vendor.setVendorEnterprise(myenter);
+		vendor.setVendorUser(myuser);
+		vendor.setVendUserUU(adp.getApplicantuu());
+		vendor.setVendEnUU(adp.getClientuu());
+		return vendorDao.save(vendor);
+	}
+
 }