db_companies.go

package main

import "fmt"

type DbCompany struct {
	Id     uint   `json:"id"`
	IgdbId uint   `json:"igdb_id"`
	Name   string `json:"name"`
}

func (db DbClient) GetCompany(igdbId int) DbCompany {
	var company DbCompany
	err := db.instance.QueryRow(
		"SELECT id, name FROM companies WHERE igdb_id = ?", igdbId,
	).Scan(
		&company.Id,
		&company.Name,
	)
	if err != nil {
		panic(err.Error())
	}
	return company
}

func (db DbClient) GetCompanyById(id int64) DbCompany {
	var company DbCompany
	err := db.instance.QueryRow("SELECT name FROM companies WHERE id = ?", id).Scan(&company.Name)
	if err != nil {
		panic(err.Error())
	}
	return company
}

func (db DbClient) AddCompany(company DbCompany) {
	query := fmt.Sprintf("INSERT INTO companies (igdb_id, name, created_at) VALUES (%d, \"%s\", NOW())",
		company.IgdbId,
		company.Name,
	)
	insert, err := db.instance.Query(query)
	if err != nil {
		panic(err.Error())
	}
	defer insert.Close()
}

func (db DbClient) CompanyExists(igdbId int) bool {
	var company DbCompany
	err := db.instance.QueryRow("SELECT name FROM companies WHERE igdb_id = ?", igdbId).Scan(&company.Name)
	if err != nil {
		return false
	}
	return true
}