本地ip时间bug

src/main/java/com/edgec/browserbackend/browser/service/Impl/IpResourceServiceImpl.java

@@ -198,14 +198,18 @@ public class IpResourceServiceImpl implements IpResourceService {
                 ipResource.setIpType(IpType.VENDOR);
                 ipResource.setVendor(Vendor.valueOf(ipResourceRequestDto.getVendor()));
                 ipResource.setStatus(3);
+                ipResource.setValidTime(Instant.now().plusSeconds(3600*24*30).toEpochMilli());
             } else {
                 ipResource.setAddr("本地Ip未使用");
                 ipResource.setIpType(IpType.LOCAL);
                 ipResource.setVendor(Vendor.valueOf(ipResourceRequestDto.getVendor()));
                 ipResource.setStatus(4);
+                if (ipResourceRequestDto.getUnit().equals("week"))
+                    ipResource.setValidTime(Instant.now().atZone(ZoneOffset.UTC).plusWeeks(ipResourceRequestDto.getPeriod()).toInstant().toEpochMilli());
+                else
+                    ipResource.setValidTime(Instant.now().atZone(ZoneOffset.UTC).plusMonths(ipResourceRequestDto.getPeriod()).toInstant().toEpochMilli());
             }
             ipResource.setPurchasedTime(Instant.now().toEpochMilli());
-            ipResource.setValidTime(Instant.now().plusSeconds(3600*24*30).toEpochMilli());
             ipResource.setPort(port);
             ipResource.setUsername(USERNAME);
             if (account.getParent() != null)
@@ -384,8 +388,13 @@ public class IpResourceServiceImpl implements IpResourceService {
                 IpChargeRequestDto ipChargeRequestDto = buildIpChargeRequestDto(ipResourceRequestDto, 2, ipResourceRequestDto.getPayMethod());
                 accountService.chargeByMoney(username, newprice1, ipChargeRequestDto);
 
-                ipResource.setValidTime(Instant.ofEpochMilli(ipResource.getValidTime()).atZone(ZoneOffset.UTC)
-                        .plusMonths(ipResourceRequestDto.getPeriod()).toInstant().toEpochMilli());
+                if (ipResourceRequestDto.getUnit().equals("week")) {
+                    ipResource.setValidTime(Instant.ofEpochMilli(ipResource.getValidTime()).atZone(ZoneOffset.UTC)
+                            .plusMonths(ipResourceRequestDto.getPeriod()).toInstant().toEpochMilli());
+                } else {
+                    ipResource.setValidTime(Instant.ofEpochMilli(ipResource.getValidTime()).atZone(ZoneOffset.UTC)
+                            .plusWeeks(ipResourceRequestDto.getPeriod()).toInstant().toEpochMilli());
+                }
                 ipResourceRepository.save(ipResource);
                 ipOperationResultDto.getSuccessList().add(x);