Ben 1 year ago
parent
commit
47688dfdb6
2 changed files with 2 additions and 2 deletions
  1. 1 1
      internal/server/server.go
  2. 1 1
      internal/storage/user.go

+ 1 - 1
internal/server/server.go

@@ -49,7 +49,7 @@ func AddUsedDuration(c *gin.Context) {
 		return
 	} else {
 		freeTrialDuration := totalFreeDuration - existed
-		if freeTrialDuration < 0 {
+		if totalFreeDuration <= existed || freeTrialDuration > totalFreeDuration {
 			freeTrialDuration = 0
 		}
 		c.JSON(http.StatusOK, dto.ConfigResponse{

+ 1 - 1
internal/storage/user.go

@@ -28,7 +28,7 @@ func AddUsedDuration(deviceId string, duration uint64) (uint64, error) {
 	}
 	ret := existed + duration
 	data := make([]byte, 8)
-	binary.BigEndian.PutUint64(data, duration)
+	binary.BigEndian.PutUint64(data, ret)
 	if err := set(generateDeviceIdUsedKey(deviceId), data); err != nil {
 		return 0, err
 	}