eta_crawler/cmd/processor_factory.go

// @Author gmy 2024/8/6 10:48:00
package main

import (
	"context"
	"eta/eta_crawler/models"
	"fmt"
)

type ReportProcessor interface {
	Process(ctx context.Context, product string, reportContent string, keywords []string, classifyId int) ([]models.BaseFromLyData, error)
}

func GetProcessor(product string, category string) (ReportProcessor, error) {
	if product == "大豆" {
		switch category {
		case "进口成本":
			return &ImportCostProcessor{}, nil
		case "加工利润":
			return &ProcessingProfitProcessor{}, nil
		case "加工报告":
			return &ProcessingReportProcessor{}, nil
		case "船运费用":
			return &ShippingCostProcessor{}, nil
		case "供需平衡":
			return &SupplyDemandBalanceProcessor{}, nil
		case "采购装船":
			return &PurchaseShippingProcessor{}, nil
		case "库存分析":
			return &InventoryAnalysisProcessor{}, nil
		default:
			return nil, fmt.Errorf("unknown category: %s", category)
		}
	} else if product == "豆粕" {
		switch category {
		case "库存分析":
			return &InventoryAnalysisProcessor{}, nil
		default:
			return nil, fmt.Errorf("unknown category: %s", category)
		}
	} else if product == "大豆油" {
		switch category {
		case "库存分析":
			return &InventoryAnalysisProcessor{}, nil
		case "价差套利":
			return &PriceSpreadArbitrageProcessor{}, nil
		case "每日成交":
			return &DailyTransactionProcessor{}, nil
		default:
			return nil, fmt.Errorf("unknown category: %s", category)
		}
	} else if product == "棕榈油" {
		switch category {
		case "库存分析":
			return &InventoryAnalysisProcessor{}, nil
		case "每日成交":
			return &DailyTransactionProcessor{}, nil
		default:
			return nil, fmt.Errorf("unknown category: %s", category)
		}
	} else if product == "油菜籽" {
		switch category {
		case "进口成本":
			return &ImportCostProcessor{}, nil
		case "库存分析":
			return &InventoryAnalysisProcessor{}, nil
		default:
			return nil, fmt.Errorf("unknown category: %s", category)
		}
	} else if product == "菜粕" {
		switch category {
		case "库存分析":
			return &InventoryAnalysisProcessor{}, nil
		default:
			return nil, fmt.Errorf("unknown category: %s", category)
		}
	} else if product == "菜籽油" {
		switch category {
		case "库存分析":
			return &InventoryAnalysisProcessor{}, nil
		default:
			return nil, fmt.Errorf("unknown category: %s", category)
		}
	}
	// 可以添加更多的逻辑来处理其他产品和类别
	return nil, fmt.Errorf("no processor found for product %s and category %s", product, category)
}