GoEthemineTelegramBot/update_data.go

package main

import (
	"fmt"
	"math"
	"strings"

	tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api"

	ethmineapi "GoEthemineTelegramBot/ethermineapi"
	"GoEthemineTelegramBot/botsettings"
)

func UpdateData() {
	LogInfo.Println("Updating data!")

	walletList, err := dBConnector.UniqueWallets()
	if err != nil {
		LogError.Println(err); return
	}

	for _, wallet := range walletList {
		UpdateWallet(wallet)
	}
}

func UpdateWallet(wallet string) {
	LogInfo.Printf(">> Getting data for wallet %s\n", wallet)
	LogInfo.Printf("From: %s/miner/%s/currentStats\n", botsettings.ApiUrl(), wallet)
	
	// Get current stats for miner
	currentStats, err := ethmineapi.MinerStats(botsettings.ApiUrl(), wallet)
	if err != nil {
		if strings.Contains(err.Error(), "no data for miner") {
			LogWarn.Println(err)
		} else {
			LogError.Println(err)
		}
		return
	}

	if currentStats.Status != "OK" {
		LogError.Printf("Error getting response from pool for %s!\n", wallet); 
		return
	}

	// Line 42 of UpdateData.cs
	// Search for the latest record for miner
	lastMinerRecord, err := dBConnector.LatestMinerRecord(wallet)
	if err != nil {
		LogError.Println(err)
		return
	}

	// Exist previous record with the same time
	if lastMinerRecord.Time == currentStats.Data.Time {
		LogInfo.Printf("Row with wallet: %s and time: %d already exists!", wallet, currentStats.Data.Time)
		return
	}

	LogInfo.Printf("Creating new record for miner %s, time %d\n", wallet, currentStats.Data.Time)
	// Line 28 of UpdateData.cs
	newMinerRecord := Miner {
	    Wallet:           wallet,
	    Time:             currentStats.Data.Time,
	    ReportedHashrate: currentStats.Data.ReportedHashrate,
	    CurrentHashrate:  currentStats.Data.CurrentHashrate,
	    ValidShares:      currentStats.Data.ValidShares,
	    StaleShares:      currentStats.Data.StaleShares,
	    InvalidShares:    currentStats.Data.InvalidShares,
	    Workers:          currentStats.Data.ActiveWorkers,
	    Unpaid:           currentStats.Data.Unpaid,
	}

	result := dBConnector.GormDb.Create(&newMinerRecord)
	if result.Error != nil {
		LogError.Println(result.Error)
		return
	}
	LogInfo.Printf("Added new row for miner %s\n", newMinerRecord.Wallet)

	if (newMinerRecord.Unpaid < lastMinerRecord.Unpaid) {
		err = NotifyChats(wallet)
		if err != nil {
			LogError.Println(err)
		}
	}
	
	dBConnector.UpdateWorkers(newMinerRecord, lastMinerRecord)
}

// at main... or somewhere else
func NotifyChats(wallet string) (err error) {
	walletChats, err := dBConnector.ChatsWithWallet(wallet)
	if err != nil {
		return err
	} 
	for _, chat := range walletChats {
		LogInfo.Println("Sended 'Payout detected!' message to chat ", chat)
		go myBotClient.Send(tgbotapi.NewMessage(chat, "Payout detected!"))
	}
	return nil
}

// UpdateWorkers updates data on workers of newMinerRecord
func (c *Connection) UpdateWorkers(newMinerRecord Miner, lastMinerRecord Miner) {
	// Line 64 of UpdateData.cs
	LogInfo.Printf("Getting workers data for wallet %s\n", newMinerRecord.Wallet)
	LogInfo.Printf("From: %s/miner/%s", botsettings.ApiUrl(), newMinerRecord.Wallet)

	currentWorker, err := ethmineapi.Workers(botsettings.ApiUrl(), newMinerRecord.Wallet)
	if err != nil {
		LogWarn.Println(err); return
	}

	if currentWorker.Status != "OK" {
		return
	}

	// Unsure is it occurs
	if len(currentWorker.Data) == 0 { 
		LogInfo.Printf("data in response from %s for workers of wallet %s not contains workers!\n",
			botsettings.ApiUrl(), newMinerRecord.Wallet)
		return
	}

	//Update for each individual worker
	for _, workerData := range currentWorker.Data {
		// In perfection:
		// worker, err := workerData.Update()
		// if err != nil {...}
		// if worker.Payout {
		//	c.getPayout()
		//	...
		// } 
		//
		// In reality
		c.updateWorker(workerData, newMinerRecord, lastMinerRecord)
	}
}

// updateWorker updates data on workerData 
func (c *Connection) updateWorker(workerData ethmineapi.WorkerData, 
	newMinerRecord Miner, lastMinerRecord Miner) {

	LogInfo.Printf("Creating new worker record for %s, time %d\n", workerData.Worker, workerData.Time)

	newWorkerRecord := Worker{
	    Wallet:            newMinerRecord.Wallet,
	    Time:              workerData.Time,
	    Worker:            workerData.Worker,
	    ReportedHashrate:  workerData.ReportedHashrate,
	    CurrentHashrate:   workerData.CurrentHashrate,
	    ValidShares:       workerData.ValidShares,
	    StaleShares:       workerData.StaleShares,
	    InvalidShares:     workerData.InvalidShares,
	}

	LogInfo.Printf("New worker record creating complete %s", newWorkerRecord.Worker)

	//Search for latest record for worker
	lastWorkerRecord, err := c.LatestWorkerRecord(workerData.Worker)
	if err != nil {
		LogError.Println(err);
		return
	}

	// Line 196 of UpdateData.cs
	if lastWorkerRecord == (Worker{}) {
		// TODO check default value of WorkerUnpaid 
		LogDebug.Println("Setting WorkerUnpaid to 0")
		newWorkerRecord.WorkerUnpaid = 0
	}
	
	// Line 85 of UpdateData.cs
	// check for payout
	// maybe newWorkerRecord.WorkerUnpaid, err := c.{}payout{}(newMinerRecord, lastMinerRecord)

	//Repeating of line 75 due to update of individual worker 
	if lastWorkerRecord != (Worker{}) && newMinerRecord.Unpaid < lastMinerRecord.Unpaid {
		// Adding new payout field to db
		payouts, err := ethmineapi.Payouts(botsettings.ApiUrl(), newWorkerRecord.Wallet)
		if err != nil {
			LogError.Println(err); return
		}

		if (len(payouts.Data) == 0) {
			LogWarn.Println("No actual payouts for wallet", newWorkerRecord.Wallet)
			return
		}

		lastPayout := payouts.Data[0]

		LogInfo.Println("Last payout time =", lastPayout.PaidOn)

		newPayoutRecord := Payout{
		    Wallet:  newWorkerRecord.Wallet,
		    Time:    lastPayout.PaidOn,
		    Amount:  lastPayout.Amount,
		    Worker:  workerData.Worker,
		}

		workerAmount := float64(lastWorkerRecord.WorkerUnpaid+
			(float64(lastPayout.Amount-lastMinerRecord.Unpaid))) *
			(float64(workerData.ReportedHashrate) / float64(newMinerRecord.ReportedHashrate))
		if math.IsNaN(workerAmount) || math.IsInf(workerAmount, 0) {
			workerAmount = lastWorkerRecord.WorkerUnpaid
		}
		newPayoutRecord.WorkerAmount = workerAmount

		workerUnpaid := float64(newMinerRecord.Unpaid) *
			(float64(workerData.ReportedHashrate) / float64(newMinerRecord.ReportedHashrate))
		LogDebug.Println("newMinerRecord.Unpaid:", newMinerRecord.Unpaid)
		LogDebug.Println("lastMinerRecord.Unpaid:", lastMinerRecord.Unpaid)
		LogDebug.Println("workerData.ReportedHashrate:", workerData.ReportedHashrate)
		LogDebug.Println("newMinerRecord.ReportedHashrate:", newMinerRecord.ReportedHashrate)
		LogDebug.Println("workerUnpaid:", workerUnpaid)
		if math.IsNaN(workerUnpaid) || math.IsInf(workerUnpaid, 0) {
			LogDebug.Println("workerUnpaid is NaN or Inf")
			workerUnpaid = 0
		}
		newWorkerRecord.WorkerUnpaid = workerUnpaid

		if result := c.GormDb.Create(&newPayoutRecord); result.Error != nil {
			LogError.Println(result.Error)
			return
		}
		LogInfo.Println("Added newPayoutRecord, time =", newPayoutRecord.Time)

		//removing old records
		result := c.GormDb.
			Where("wallet = ? AND time < ?", newPayoutRecord.Wallet, newMinerRecord.Time).
			Delete(Worker{})
		if result.Error != nil {
			LogError.Println(result.Error)
		}

		result = c.GormDb.
			Where("wallet = ? AND time < ?", newPayoutRecord.Wallet, newMinerRecord.Time).
			Delete(Miner{})
		if result.Error != nil {
			LogError.Println(result.Error)
		}

		result = c.GormDb.
			Where("wallet = ? AND time < ?", newPayoutRecord.Wallet, newMinerRecord.Time).
			Delete(Payout{})
		if result.Error != nil {
			LogError.Println(result.Error)
		}

	//Line 135 of UpdateData.cs
	} else {
		//Payout not happened
		//no check that last balance and prev balance are the same
		//don't sure > or >=
		//TODO rewrite this
		var max int64
		// max, err := c.LastPayoutTime(wallet)
		result := c.GormDb.Model(&Payout{}).
			Where("wallet = ?", lastWorkerRecord.Wallet).
			Order("time desc").
			Limit(1).
			Select("time").
			Scan(&max)
		if result.Error != nil {
			LogWarn.Printf("No payouts data for %s! Time is set to 0!\n", lastWorkerRecord.Wallet)
			LogError.Println(result.Error)
			max = 0
		}

		if lastWorkerRecord.Time > max || lastWorkerRecord.WorkerUnpaid == 0 {
			workerUnpaid := float64(lastWorkerRecord.WorkerUnpaid+
				float64(newMinerRecord.Unpaid-lastMinerRecord.Unpaid)) *
				(float64(workerData.ReportedHashrate) / float64(newMinerRecord.ReportedHashrate))
			if math.IsNaN(workerUnpaid) || math.IsInf(workerUnpaid, 0) {
				workerUnpaid = lastWorkerRecord.WorkerUnpaid
			}
			newWorkerRecord.WorkerUnpaid = workerUnpaid
			// Line 169 of UpdateData.cs
		} else {
			workerUnpaid := float64((newMinerRecord.Unpaid - lastMinerRecord.Unpaid)) *
				(float64(workerData.ReportedHashrate) / float64(newMinerRecord.ReportedHashrate))
			if math.IsNaN(workerUnpaid) || math.IsInf(workerUnpaid, 0) {
				workerUnpaid = lastWorkerRecord.WorkerUnpaid
			}
			newWorkerRecord.WorkerUnpaid = workerUnpaid

			targetedChats, err := c.ChatsWithWallet(newMinerRecord.Wallet)
			if err != nil {
				LogError.Println(err)
			}

			if err == nil {
				msg := fmt.Sprintf("Debug info: Your worker %s hasn't been zeroed on payout!", 
					workerData.Worker)
				for _, chatId := range targetedChats {
					go myBotClient.Send(tgbotapi.NewMessage(int64(chatId), msg))
				}
				LogDebug.Printf("Worker %s on address %s hasn't been zeroed on payout!\n", 
					workerData.Worker, newMinerRecord.Wallet)
			}
		}
	}

	rowExists, err := c.IsWorkerRowExists(newWorkerRecord.Worker, newWorkerRecord.Time)
	if err != nil {
		LogError.Println(err); return
	}

	if rowExists {
		LogInfo.Printf("Error adding new row: row with worker: %s, time: %d already exists!\n", 
		newWorkerRecord.Worker, newWorkerRecord.Time)
		return
	}

	if result := c.GormDb.Create(&newWorkerRecord); result.Error != nil {
		LogError.Println(result.Error)
		return
	}
	
	LogInfo.Printf("Added new row for worker %s\n", newWorkerRecord.Worker)
}