优化性能

service/yggdrasil/texture.go

@@ -25,52 +25,53 @@ var (
 )
 
 func (y *Yggdrasil) delTexture(ctx context.Context, userProfileID int, textureType string) error {
-	err := utils.WithTx(ctx, y.client, func(tx *ent.Tx) error {
+	// 查找此用户该类型下是否已经存在皮肤
-		// 查找此用户该类型下是否已经存在皮肤
+	tl, err := y.client.UserTexture.Query().Where(usertexture.And(
-		tl, err := tx.UserTexture.Query().Where(usertexture.And(
+		usertexture.UserProfileID(userProfileID),
-			usertexture.UserProfileID(userProfileID),
+		usertexture.Type(textureType),
-			usertexture.Type(textureType),
+	)).All(ctx)
-		)).ForUpdate().All(ctx)
+	if err != nil {
+		return fmt.Errorf("delTexture: %w", err)
+	}
+	if len(tl) == 0 {
+		return nil
+	}
+	// 若存在，查找是否被引用
+	for _, v := range tl {
+		c, err := y.client.UserTexture.Query().Where(usertexture.TextureID(v.TextureID)).Count(ctx)
 		if err != nil {
-			return err
+			return fmt.Errorf("delTexture: %w", err)
 		}
-		if len(tl) == 0 {
+		if c == 1 {
-			return nil
+			// 若没有其他用户使用该皮肤，删除文件和记录
-		}
+			t, err := y.client.Texture.Query().Where(texture.ID(v.TextureID)).Only(ctx)
-		// 若存在，查找是否被引用
-		for _, v := range tl {
-			c, err := tx.UserTexture.Query().Where(usertexture.TextureID(v.TextureID)).Count(ctx)
 			if err != nil {
-				return err
+				var nf *ent.NotFoundError
+				if errors.As(err, &nf) {
+					continue
+				}
+				return fmt.Errorf("delTexture: %w", err)
 			}
-			if c == 1 {
+			path := filepath.Join(y.config.TexturePath, t.TextureHash[:2], t.TextureHash[2:4], t.TextureHash)
-				// 若没有其他用户使用该皮肤，删除文件和记录
+			err = os.Remove(path)
-				t, err := tx.Texture.Query().Where(texture.ID(v.TextureID)).Only(ctx)
+			if err != nil && !errors.Is(err, os.ErrNotExist) {
-				if err != nil {
+				return fmt.Errorf("delTexture: %w", err)
-					return err
+			}
-				}
+			// Texture 表中删除记录
-				path := filepath.Join(y.config.TexturePath, t.TextureHash[:2], t.TextureHash[2:4], t.TextureHash)
+			err = y.client.Texture.DeleteOneID(v.TextureID).Exec(ctx)
-				err = os.Remove(path)
+			if err != nil {
-				if err != nil && !errors.Is(err, os.ErrNotExist) {
+				return fmt.Errorf("delTexture: %w", err)
-					return err
-				}
-				// Texture 表中删除记录
-				err = tx.Texture.DeleteOneID(v.TextureID).Exec(ctx)
-				if err != nil {
-					return err
-				}
 			}
 		}
-		ids := lo.Map[*ent.UserTexture, int](tl, func(item *ent.UserTexture, index int) int {
+	}
-			return item.ID
+	ids := lo.Map[*ent.UserTexture, int](tl, func(item *ent.UserTexture, index int) int {
-		})
+		return item.ID
-		// 中间表删除记录
-		// UserProfile 上没有于此相关的字段，所以无需操作
-		_, err = tx.UserTexture.Delete().Where(usertexture.IDIn(ids...)).Exec(ctx)
-		return err
-		// 小概率皮肤上传后，高并发时被此处清理。问题不大重新上传一遍就行。
-		// 条件为使用一个独一无二的皮肤的用户，更换皮肤时，另一个用户同时更换自己的皮肤到这个独一无二的皮肤上。
 	})
+	// 中间表删除记录
+	// UserProfile 上没有于此相关的字段，所以无需操作
+	_, err = y.client.UserTexture.Delete().Where(usertexture.IDIn(ids...)).Exec(ctx)
+	// 小概率皮肤上传后，高并发时被此处清理。问题不大重新上传一遍就行。
+	// 条件为使用一个独一无二的皮肤的用户，更换皮肤时，另一个用户同时更换自己的皮肤到这个独一无二的皮肤上。
 	if err != nil {
 		return fmt.Errorf("delTexture: %w", err)
 	}