package db const ( UserStateNone = UserState("") UserStateGo = UserState("Go") UserStateEat = UserState("Eat") UserStatePause = UserState("Pause") UserStateWeight = UserState("Weight") ) type UserState string type Chat struct { ChatID int64 `bson:"chat_id"` } type ChatInfo struct { ChatID int64 `bson:"chat_id"` Status string `bson:"status"` Username string `bson:"username"` } func (c *ChatInfo) GetStatus() UserState { return UserState(c.Status) }