Paul Lecuq
5b95e817de
All checks were successful
continuous-integration/drone/push Build is passing
* removed unused columns in tables * added multithreaded support for ip scans * added health endpoint
59 lines
1.2 KiB
Go
59 lines
1.2 KiB
Go
package models
|
|
|
|
import (
|
|
"time"
|
|
|
|
"xorm.io/xorm"
|
|
)
|
|
|
|
func (country *Country) GetOrCreate(session *xorm.Session) (apicountry *APICountry, err error) {
|
|
has, err := session.Get(country)
|
|
if err != nil {
|
|
return
|
|
}
|
|
if !has {
|
|
session.Insert(country)
|
|
} else {
|
|
session.ID(country.ID).Update(country)
|
|
}
|
|
apicountry = country.APIFormat()
|
|
country.Get(session)
|
|
return
|
|
}
|
|
|
|
func (country *Country) Get(session *xorm.Session) (apicountry *APICountry, err error) {
|
|
has, err := session.Get(country)
|
|
if !has || err != nil {
|
|
return
|
|
}
|
|
apicountry = country.APIFormat()
|
|
return
|
|
}
|
|
|
|
func (country *Country) APIFormat() *APICountry {
|
|
if country == nil {
|
|
return &APICountry{}
|
|
}
|
|
return &APICountry{
|
|
CountryName: country.CountryName,
|
|
}
|
|
}
|
|
|
|
func (country *Country) APIParse(apicountry APICountry) (err error) {
|
|
*country = Country{
|
|
CountryName: apicountry.CountryName}
|
|
return
|
|
}
|
|
|
|
type Country struct {
|
|
ID int `xorm:"pk autoincr"`
|
|
CountryName string `xorm:"text unique(countryindex) country_name" json:"country_name"`
|
|
Created time.Time `xorm:"created notnull"`
|
|
Updated time.Time `xorm:"updated notnull"`
|
|
}
|
|
|
|
type APICountry struct {
|
|
ID int `json:"-"`
|
|
CountryName string `json:"country_name"`
|
|
}
|