feat: MCPサーバーとレース時間ツールの仮実装

Makefile

@@ -1,18 +1,29 @@
 GOBASE=$(shell pwd)
 GOBIN=$(GOBASE)/bin
 
-.PHONY: gen-wire build
 
+.PHONY: install
+install:
+	go get github.com/mark3labs/mcp-go
+	go install golang.org/x/tools/cmd/goimports@latest
+	go install github.com/google/wire/cmd/wire@latest
+	go install github.com/go-delve/delve/cmd/dlv@v1.24.2
+	go install github.com/air-verse/air@v1.62.0
+
+.PHONY: gen-wire
 gen-wire:
 	wire gen di/wire.go
 
+.PHONY: lint
 lint:
 	goimports -w .
 
+.PHONY: build
 build:
 	go mod download
-	go build -o bin/ipat-aggreagtor cmd/main.go
+	go build -o bin/mcp-server cmd/mcp/main.go
 
+.PHONY: cache-clear
 cache-clear:
 	rm -rf ./cache/colly/*
 

app/controller/analysis.go

@@ -111,3 +111,17 @@ func (a *Analysis) Beta(ctx context.Context, input *AnalysisInput) {
 	}
 	a.logger.Info("fetching analysis beta end")
 }
+
+func (a *Analysis) Usm(ctx context.Context, input *AnalysisInput) {
+	a.logger.Info("fetching analysis usm start")
+	if err := a.analysisUseCase.Usm(ctx, &analysis_usecase.AnalysisInput{
+		Markers: input.Master.AnalysisMarkers,
+		Races:   input.Master.Races,
+		Odds: &analysis_usecase.AnalysisOddsInput{
+			Win:   input.Master.WinOdds,
+			Place: input.Master.PlaceOdds,
+		},
+	}); err != nil {
+		a.logger.Errorf("analysis usm error: %v", err)
+	}
+}

app/controller/master.go

@@ -18,7 +18,7 @@ type MasterInput struct {
 type MasterOutput struct {
 	Tickets           []*ticket_csv_entity.RaceTicket
 	Races             []*data_cache_entity.Race
-	RaceTimes         []*data_cache_entity.RaceTime
+	RaceTimes         []*data_cache_entity.RaceTimeV2
 	Jockeys           []*data_cache_entity.Jockey
 	WinOdds           []*data_cache_entity.Odds
 	PlaceOdds         []*data_cache_entity.Odds

app/controller/mcp.go

@@ -0,0 +1,54 @@
+package controller
+
+import (
+	"context"
+
+	"github.com/mapserver2007/ipat-aggregator/app/usecase/mcp_usecase"
+	"github.com/mark3labs/mcp-go/server"
+	"github.com/sirupsen/logrus"
+)
+
+const (
+	serverName    = "yamato"
+	serverVersion = "0.0.1"
+)
+
+type MCPServer struct {
+	mcpServer   *server.MCPServer
+	toolUseCase mcp_usecase.Tool
+	logger      *logrus.Logger
+}
+
+func NewMCPServer(
+	toolUseCase mcp_usecase.Tool,
+	logger *logrus.Logger,
+) *MCPServer {
+	mcpServer := server.NewMCPServer(
+		serverName,
+		serverVersion,
+		server.WithResourceCapabilities(true, true),
+		server.WithLogging(),
+		server.WithRecovery(),
+	)
+
+	return &MCPServer{
+		mcpServer:   mcpServer,
+		toolUseCase: toolUseCase,
+		logger:      logger,
+	}
+}
+
+func (s *MCPServer) Start() error {
+	if err := server.ServeStdio(s.mcpServer); err != nil {
+		s.logger.Fatalf("server start error: %v", err)
+		return err
+	}
+
+	return nil
+}
+
+func (s *MCPServer) Load(ctx context.Context) error {
+	s.mcpServer.AddTool(s.toolUseCase.GetRaceTimeTool(), s.toolUseCase.GetRaceTimeToolHandler)
+
+	return nil
+}

app/domain/entity/analysis_entity/jockey_result.go

@@ -0,0 +1,7 @@
+package analysis_entity
+
+import "github.com/mapserver2007/ipat-aggregator/app/domain/types"
+
+type JockeyResult struct {
+	raceId types.RaceId
+}

app/domain/entity/data_cache_entity/jockey_result.go

@@ -0,0 +1,86 @@
+package data_cache_entity
+
+import (
+	"github.com/mapserver2007/ipat-aggregator/app/domain/types"
+	"github.com/shopspring/decimal"
+)
+
+type JockeyResult struct {
+	jockeyResultId types.JockeyResultId
+	jockeyId       types.JockeyId
+	raceId         types.RaceId
+	raceDate       types.RaceDate
+	raceCourseId   types.RaceCourse
+	odds           decimal.Decimal
+	orderNo        int
+	horseId        types.HorseId
+	courseCategory types.CourseCategory
+	distance       int
+}
+
+func NewJockeyResult(
+	jockeyResultId string,
+	jockeyId string,
+	raceId string,
+	raceDate int,
+	raceCourseId string,
+	odds string,
+	orderNo int,
+	horseId string,
+	courseCategory int,
+	distance int,
+) *JockeyResult {
+	decimalOdds, _ := decimal.NewFromString(odds)
+	return &JockeyResult{
+		jockeyResultId: types.JockeyResultId(jockeyResultId),
+		jockeyId:       types.JockeyId(jockeyId),
+		raceId:         types.RaceId(raceId),
+		raceDate:       types.RaceDate(raceDate),
+		raceCourseId:   types.RaceCourse(raceCourseId),
+		odds:           decimalOdds,
+		orderNo:        orderNo,
+		horseId:        types.HorseId(horseId),
+		courseCategory: types.CourseCategory(courseCategory),
+		distance:       distance,
+	}
+}
+
+func (j *JockeyResult) JockeyResultId() types.JockeyResultId {
+	return j.jockeyResultId
+}
+
+func (j *JockeyResult) JockeyId() types.JockeyId {
+	return j.jockeyId
+}
+
+func (j *JockeyResult) RaceId() types.RaceId {
+	return j.raceId
+}
+
+func (j *JockeyResult) RaceDate() types.RaceDate {
+	return j.raceDate
+}
+
+func (j *JockeyResult) RaceCourseId() types.RaceCourse {
+	return j.raceCourseId
+}
+
+func (j *JockeyResult) Odds() decimal.Decimal {
+	return j.odds
+}
+
+func (j *JockeyResult) OrderNo() int {
+	return j.orderNo
+}
+
+func (j *JockeyResult) HorseId() types.HorseId {
+	return j.horseId
+}
+
+func (j *JockeyResult) CourseCategory() types.CourseCategory {
+	return j.courseCategory
+}
+
+func (j *JockeyResult) Distance() int {
+	return j.distance
+}

app/domain/entity/data_cache_entity/race_time.go

@@ -1,93 +1,64 @@
 package data_cache_entity
 
 import (
-	"time"
-
 	"github.com/mapserver2007/ipat-aggregator/app/domain/types"
+	"github.com/mapserver2007/ipat-aggregator/app/domain/vo/data_cache_vo"
 )
 
-type RaceTime struct {
-	raceId     types.RaceId
-	raceDate   types.RaceDate
-	time       string
-	timeIndex  int
-	trackIndex int
-	rapTimes   []time.Duration
-	first3f    time.Duration
-	first4f    time.Duration
-	last3f     time.Duration
-	last4f     time.Duration
-	rap5f      time.Duration
+type RaceTimeV2 struct {
+	raceTimeId  types.RaceTime
+	raceId      types.RaceId
+	raceDate    types.RaceDate
+	time        *data_cache_vo.RaceTime
+	timeIndex   int
+	trackIndex  int
+	raceTimeRap *data_cache_vo.RaceTimeRap
 }
 
-func NewRaceTime(
+func NewRaceTimeV2(
+	raceTimeId types.RaceTime,
 	raceId types.RaceId,
 	raceDate types.RaceDate,
-	time string,
+	time *data_cache_vo.RaceTime,
 	timeIndex int,
 	trackIndex int,
-	rapTimes []time.Duration,
-	first3f time.Duration,
-	first4f time.Duration,
-	last3f time.Duration,
-	last4f time.Duration,
-	rap5f time.Duration,
-) *RaceTime {
-	return &RaceTime{
-		raceId:     raceId,
-		raceDate:   raceDate,
-		time:       time,
-		timeIndex:  timeIndex,
-		trackIndex: trackIndex,
-		rapTimes:   rapTimes,
-		first3f:    first3f,
-		first4f:    first4f,
-		last3f:     last3f,
-		last4f:     last4f,
-		rap5f:      rap5f,
+	raceTimeRap *data_cache_vo.RaceTimeRap,
+) *RaceTimeV2 {
+	return &RaceTimeV2{
+		raceTimeId:  raceTimeId,
+		raceId:      raceId,
+		raceDate:    raceDate,
+		time:        time,
+		timeIndex:   timeIndex,
+		trackIndex:  trackIndex,
+		raceTimeRap: raceTimeRap,
 	}
 }
 
-func (r *RaceTime) RaceId() types.RaceId {
+func (r *RaceTimeV2) RaceTimeId() types.RaceTime {
+	return r.raceTimeId
+}
+
+func (r *RaceTimeV2) RaceId() types.RaceId {
 	return r.raceId
 }
 
-func (r *RaceTime) RaceDate() types.RaceDate {
+func (r *RaceTimeV2) RaceDate() types.RaceDate {
 	return r.raceDate
 }
 
-func (r *RaceTime) Time() string {
+func (r *RaceTimeV2) Time() *data_cache_vo.RaceTime {
 	return r.time
 }
 
-func (r *RaceTime) TimeIndex() int {
+func (r *RaceTimeV2) TimeIndex() int {
 	return r.timeIndex
 }
 
-func (r *RaceTime) TrackIndex() int {
+func (r *RaceTimeV2) TrackIndex() int {
 	return r.trackIndex
 }
 
-func (r *RaceTime) RapTimes() []time.Duration {
-	return r.rapTimes
-}
-
-func (r *RaceTime) First3f() time.Duration {
-	return r.first3f
-}
-
-func (r *RaceTime) First4f() time.Duration {
-	return r.first4f
-}
-
-func (r *RaceTime) Last3f() time.Duration {
-	return r.last3f
-}
-
-func (r *RaceTime) Last4f() time.Duration {
-	return r.last4f
-}
-
-func (r *RaceTime) Rap5f() time.Duration {
-	return r.rap5f
+func (r *RaceTimeV2) RaceTimeRap() *data_cache_vo.RaceTimeRap {
+	return r.raceTimeRap
 }

app/domain/entity/mcp_result_entity/mcp_result.go

@@ -0,0 +1,63 @@
+package mcp_result_entity
+
+type MCPResult struct {
+	Races     []*Race    `json:"races"`
+	RaceNote  string     `json:"race_note"`
+	Condition *Condition `json:"condition"`
+}
+
+type Race struct {
+	RaceId              string        `json:"race_id"`
+	RaceDate            string        `json:"race_date"`
+	RaceName            string        `json:"race_name"`
+	RaceNumber          int           `json:"race_number"`
+	RaceCourse          string        `json:"race_course"`
+	RaceUrl             string        `json:"race_url"`
+	Entries             int           `json:"entries"`
+	Distance            int           `json:"distance"`
+	Class               string        `json:"class"`
+	CourseCategory      string        `json:"course_category"`
+	TrackCondition      string        `json:"track_condition"`
+	RaceSexCondition    string        `json:"race_sex_condition"`
+	RaceWeightCondition string        `json:"race_weight_condition"`
+	RaceAgeCondition    string        `json:"race_age_condition"`
+	RaceResults         []*RaceResult `json:"race_results"`
+}
+
+type RaceResult struct {
+	OrderNo        int    `json:"order_no"`
+	HorseId        string `json:"horse_id"`
+	HorseName      string `json:"horse_name"`
+	BracketNumber  int    `json:"bracket_number"`
+	HorseNumber    int    `json:"horse_number"`
+	JockeyId       string `json:"jockey_id"`
+	Odds           string `json:"odds"`
+	PopularNumber  int    `json:"popular_number"`
+	JockeyWeight   string `json:"jockey_weight"`
+	HorseWeight    int    `json:"horse_weight"`
+	HorseWeightAdd int    `json:"horse_weight_add"`
+}
+
+type Condition struct {
+	Odds          *Odds  `json:"odds"`
+	ConditionNote string `json:"condition_note"`
+}
+
+// type Jockey struct {
+// 	JockeyId   string `json:"jockey_id"`
+// 	JockeyName string `json:"jockey_name"`
+// }
+
+type Odds struct {
+	OddsList []string `json:"odds_list"`
+	RaceIds  []string `json:"race_ids"`
+	OddsNote string   `json:"odds_note"`
+}
+
+// type RaceUrl struct {
+// 	Url  string `json:"url"`
+// 	Note string `json:"note"`
+// }
+
+// type RaceTime struct {
+// }