zhangzengfei
2024-10-11 539a78196da60eb97cf7057c5c85dfaa9b240741
pkg/floor.go
@@ -4,33 +4,33 @@
   "fmt"
   "strconv"
   "strings"
   "gat1400Exchange/pkg/snowflake"
)
// 已弃用
// 生成一个包含楼层的人脸id,解析楼层
// 使用48位源id, 其中前41位是imageid, 不可以修改  +99 + 3位楼层(第一位0表示正,1表示负) + 2位随机数
// 使用48位源id, 其中前41位是imageid, 不可以修改 41-43位填 06 代表图像, +99 + 3位楼层(第一位0表示正,1表示负)
func GenerateFaceIdContainFloor(srcId, floorStr string) string {
   floorNum, _ := parseFloor(floorStr)
   newId := srcId[0:41] + "99" + floorNum + snowflake.CreateRandomNumber(2)
   floorNum, _ := ParseFloor("0", floorStr)
   newId := srcId[0:43] + "99" + floorNum
   //newId := srcId[0:43] + "99" + floorNum + snowflake.CreateRandomNumber(1)
   return newId
}
func ParseFloorFromId(srcId string) (string, error) {
func ParseFloorFromId(srcId string) (string, string, error) {
   if len(srcId) != 48 {
      return "", fmt.Errorf("invalid id %s", srcId)
      return "", "", fmt.Errorf("invalid id %s", srcId)
   }
   if srcId[41:43] != "99" {
      return "", fmt.Errorf("invalid flag %s", srcId[41:43])
   if srcId[43:45] != "99" {
      return "", "", fmt.Errorf("invalid flag %s", srcId[43:45])
   }
   return restoreFloor(srcId[43:46])
   return RestoreFloor(srcId[45:48])
}
// parseFloor parses the floor string and returns a three-character string
func parseFloor(floor string) (string, error) {
// ParseFloor parses the floor string and returns a three-character string
func ParseFloor(direction, floor string) (string, error) {
   var sign string
   var number string
@@ -51,32 +51,45 @@
   // Format the number to be two digits
   formattedNumber := fmt.Sprintf("%02s", number)
   return sign + formattedNumber, nil
   return direction + sign + formattedNumber, nil
}
// restoreFloor restores the three-character string back to the original floor string
func restoreFloor(encoded string) (string, error) {
   if len(encoded) != 3 {
      return "", fmt.Errorf("encoded string must be 3 characters long")
   }
// RestoreFloor restores the three-character string back to the original floor string
func RestoreFloor(encoded string) (string, string, error) {
   var direction, sign uint8
   var floorNumber, floorStr, directionStr string
   sign := encoded[0]
   number := encoded[1:]
   // 3位是旧的格式, 为了兼容, 暂时保留
   if len(encoded) == 3 {
      sign = encoded[0]
      floorNumber = encoded[1:]
   } else if len(encoded) == 4 {
      direction = encoded[0]
      sign = encoded[1]
      floorNumber = encoded[2:]
   } else {
      return "", "", fmt.Errorf("encoded string must be 3 characters long")
   }
   // Convert the number back to integer to remove any leading zeros
   parsedNumber, err := strconv.Atoi(number)
   parsedNumber, err := strconv.Atoi(floorNumber)
   if err != nil {
      return "", err
      return "", "", err
   }
   var restoredFloor string
   if sign == '1' {
      restoredFloor = fmt.Sprintf("-%dF", parsedNumber)
      floorStr = fmt.Sprintf("-%dF", parsedNumber)
   } else if sign == '0' {
      restoredFloor = fmt.Sprintf("%dF", parsedNumber)
      floorStr = fmt.Sprintf("%dF", parsedNumber)
   } else {
      return "", fmt.Errorf("invalid sign character in encoded string")
      return "", "", fmt.Errorf("invalid sign character in encoded string")
   }
   return restoredFloor, nil
   if direction == '1' {
      directionStr = "in"
   } else if direction == '2' {
      directionStr = "out"
   }
   return directionStr, floorStr, nil
}