102 lines
3.2 KiB
JavaScript
102 lines
3.2 KiB
JavaScript
// src/services/stockService.js
|
||
// 股票数据服务
|
||
|
||
import { logger } from '../utils/logger';
|
||
|
||
const API_BASE_URL = process.env.REACT_APP_API_BASE_URL || '';
|
||
|
||
/**
|
||
* 股票数据服务
|
||
*/
|
||
export const stockService = {
|
||
/**
|
||
* 获取所有股票列表
|
||
* @returns {Promise<{success: boolean, data: Array<{code: string, name: string}>}>}
|
||
*/
|
||
async getAllStocks() {
|
||
try {
|
||
const response = await fetch(`${API_BASE_URL}/api/stocklist`, {
|
||
method: 'GET',
|
||
headers: {
|
||
'Content-Type': 'application/json',
|
||
},
|
||
credentials: 'include'
|
||
});
|
||
|
||
if (!response.ok) {
|
||
throw new Error(`HTTP error! status: ${response.status}`);
|
||
}
|
||
|
||
const data = await response.json();
|
||
|
||
logger.debug('stockService', 'getAllStocks 成功', {
|
||
count: data?.length || 0
|
||
});
|
||
|
||
return {
|
||
success: true,
|
||
data: data || []
|
||
};
|
||
} catch (error) {
|
||
logger.error('stockService', 'getAllStocks', error);
|
||
return {
|
||
success: false,
|
||
data: [],
|
||
error: error.message
|
||
};
|
||
}
|
||
},
|
||
|
||
/**
|
||
* 模糊搜索股票（匹配 code 或 name）
|
||
* @param {string} query - 搜索关键词
|
||
* @param {Array<{code: string, name: string}>} stockList - 股票列表
|
||
* @param {number} limit - 返回结果数量限制
|
||
* @returns {Array<{code: string, name: string}>}
|
||
*/
|
||
fuzzySearch(query, stockList, limit = 10) {
|
||
if (!query || !stockList || stockList.length === 0) {
|
||
return [];
|
||
}
|
||
|
||
const lowerQuery = query.toLowerCase();
|
||
|
||
// 模糊匹配 code 或 name
|
||
const results = stockList.filter(stock => {
|
||
const code = (stock.code || '').toString().toLowerCase();
|
||
const name = (stock.name || '').toLowerCase();
|
||
return code.includes(lowerQuery) || name.includes(lowerQuery);
|
||
});
|
||
|
||
// 优先级排序：
|
||
// 1. code 精确匹配
|
||
// 2. name 精确匹配
|
||
// 3. code 开头匹配
|
||
// 4. name 开头匹配
|
||
// 5. 其他包含匹配
|
||
results.sort((a, b) => {
|
||
const aCode = (a.code || '').toString().toLowerCase();
|
||
const aName = (a.name || '').toLowerCase();
|
||
const bCode = (b.code || '').toString().toLowerCase();
|
||
const bName = (b.name || '').toLowerCase();
|
||
|
||
// 精确匹配
|
||
if (aCode === lowerQuery) return -1;
|
||
if (bCode === lowerQuery) return 1;
|
||
if (aName === lowerQuery) return -1;
|
||
if (bName === lowerQuery) return 1;
|
||
|
||
// 开头匹配
|
||
if (aCode.startsWith(lowerQuery) && !bCode.startsWith(lowerQuery)) return -1;
|
||
if (!aCode.startsWith(lowerQuery) && bCode.startsWith(lowerQuery)) return 1;
|
||
if (aName.startsWith(lowerQuery) && !bName.startsWith(lowerQuery)) return -1;
|
||
if (!aName.startsWith(lowerQuery) && bName.startsWith(lowerQuery)) return 1;
|
||
|
||
// 字母顺序
|
||
return aCode.localeCompare(bCode);
|
||
});
|
||
|
||
return results.slice(0, limit);
|
||
}
|
||
};
|