gostock/cmd/pnl/main.go

package main

import (
	"fmt"

	"git.apinb.com/bsm-sdk/core/utils"
	"git.apinb.com/quant/gostock/internal/config"
	"git.apinb.com/quant/gostock/internal/impl"
	"git.apinb.com/quant/gostock/internal/models"
	"github.com/jedib0t/go-pretty/v6/table"
)

var (
	ServiceKey = "gostock"
)

func main() {
	config.New(ServiceKey)
	impl.NewImpl()

	fmt.Println("")
	run()
	fmt.Println("")
}

func run() {
	tw := table.NewWriter()
	tw.SetStyle(table.StyleLight)
	tw.AppendHeader(table.Row{"ID", "Code", "Name", "OpenDate", "OpenPrice", "TodayPrice", "PNL/Per", "PNLRate(%)"})
	var data []models.MockPosition
	impl.DBService.Where("status=?", 0).Find(&data)
	var tPNL, tPNLR, cost, sell float64
	for _, item := range data {
		var stock models.StockBasic
		impl.DBService.Where("ts_code=?", item.Code).First(&stock)

		var daily models.StockDaily
		impl.DBService.Model(&models.StockDaily{}).Where("ts_code=?", item.Code).Order("id desc").Limit(1).First(&daily)

		high := utils.FloatRound(daily.High*0.997, 2)
		pnl := utils.FloatRound(high-item.OpenPrice, 2)
		pnlRate := utils.FloatRound(pnl/item.OpenPrice*100, 2)

		tPNL = tPNL + pnl
		cost = cost + item.OpenPrice
		sell = sell + high

		tw.AppendRow(table.Row{item.ID, item.Code, stock.Name, item.CreatedAt.Format("2006-01-02"), item.OpenPrice, high, pnl, pnlRate})
	}

	tPNLR = utils.FloatRound(((sell-cost)/cost)*100, 2)

	tw.AppendFooter(table.Row{"", "", "", "TOTAL", utils.FloatRound(cost, 2), utils.FloatRound(sell, 2), utils.FloatRound(tPNL, 2), tPNLR})

	fmt.Println(tw.Render())
}