51 lines
1.0 KiB
Go
51 lines
1.0 KiB
Go
|
package models
|
|||
|
|
|
|||
|
|
import "errors"
|
|||
|
|
|
|||
|
|
// Account 银行账户（公开）
|
|||
|
|
type Account struct {
|
|||
|
|
ID string // 账户ID（公开）
|
|||
|
|
Owner string // 账户所有者（公开）
|
|||
|
|
balance float64 // 余额（私有，不能直接修改）
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewAccount 创建账户
|
|||
|
|
func NewAccount(id, owner string) *Account {
|
|||
|
|
return &Account{
|
|||
|
|
ID: id,
|
|||
|
|
Owner: owner,
|
|||
|
|
balance: 0,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Deposit 存款（公开方法）
|
|||
|
|
func (a *Account) Deposit(amount float64) error {
|
|||
|
|
if !isValidAmount(amount) {
|
|||
|
|
return errors.New("金额必须大于0")
|
|||
|
|
}
|
|||
|
|
a.balance += amount
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Withdraw 取款（公开方法）
|
|||
|
|
func (a *Account) Withdraw(amount float64) error {
|
|||
|
|
if !isValidAmount(amount) {
|
|||
|
|
return errors.New("金额必须大于0")
|
|||
|
|
}
|
|||
|
|
if amount > a.balance {
|
|||
|
|
return errors.New("余额不足")
|
|||
|
|
}
|
|||
|
|
a.balance -= amount
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetBalance 获取余额（公开方法）
|
|||
|
|
func (a *Account) GetBalance() float64 {
|
|||
|
|
return a.balance
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// isValidAmount 验证金额（私有函数）
|
|||
|
|
func isValidAmount(amount float64) bool {
|
|||
|
|
return amount > 0
|
|||
|
|
}
|