feat(users): merge promotion into revenue center

2026-08-03 15:38:17 +08:00
parent 156060d2ec
commit 49c12b761f
20 changed files with 369 additions and 413 deletions
@@ -54,13 +54,52 @@ class Revenue {
    public function settle(int $revenue_id): bool {
        global $wpdb;

-        return $wpdb->update(
-            $this->table(),
-            ['status' => 'settled', 'settled_at' => current_time('mysql')],
-            ['id' => $revenue_id, 'status' => 'pending'],
-            ['%s', '%s'],
-            ['%d', '%s']
-        ) !== false;
+        if (!class_exists('\AKSRT\Credits\Balance')) {
+            return false;
+        }
+
+        $wpdb->query('START TRANSACTION');
+        try {
+            $revenue = $wpdb->get_row($wpdb->prepare(
+                "SELECT id, user_id, amount, order_number FROM {$this->table()} WHERE id = %d AND status = %s FOR UPDATE",
+                $revenue_id,
+                'pending'
+            ), ARRAY_A);
+            if (!is_array($revenue)) {
+                $wpdb->query('ROLLBACK');
+                return false;
+            }
+
+            $credited = \AKSRT\Credits\Balance::add(
+                (int) $revenue['user_id'],
+                (float) $revenue['amount'],
+                'creator_income',
+                ['type' => 'revenue', 'id' => $revenue_id],
+                sprintf(__('创作收益结算（订单 %s）', 'aksrt'), (string) $revenue['order_number'])
+            );
+            if (is_wp_error($credited)) {
+                $wpdb->query('ROLLBACK');
+                return false;
+            }
+
+            $settled = $wpdb->update(
+                $this->table(),
+                ['status' => 'settled', 'settled_at' => current_time('mysql')],
+                ['id' => $revenue_id, 'status' => 'pending'],
+                ['%s', '%s'],
+                ['%d', '%s']
+            );
+            if ($settled !== 1) {
+                $wpdb->query('ROLLBACK');
+                return false;
+            }
+
+            $wpdb->query('COMMIT');
+            return true;
+        } catch (\Throwable $e) {
+            $wpdb->query('ROLLBACK');
+            return false;
+        }
    }

    /**