wx.go 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406
  1. // Code generated by ent, DO NOT EDIT.
  2. package ent
  3. import (
  4. "encoding/json"
  5. "fmt"
  6. "strings"
  7. "time"
  8. "wechat-api/ent/agent"
  9. "wechat-api/ent/server"
  10. "wechat-api/ent/wx"
  11. "entgo.io/ent"
  12. "entgo.io/ent/dialect/sql"
  13. )
  14. // Wx is the model entity for the Wx schema.
  15. type Wx struct {
  16. config `json:"-"`
  17. // ID of the ent.
  18. ID uint64 `json:"id,omitempty"`
  19. // Create Time | 创建日期
  20. CreatedAt time.Time `json:"created_at,omitempty"`
  21. // Update Time | 修改日期
  22. UpdatedAt time.Time `json:"updated_at,omitempty"`
  23. // Status 1: normal 2: ban | 状态 1 正常 2 禁用
  24. Status uint8 `json:"status,omitempty"`
  25. // Delete Time | 删除日期
  26. DeletedAt time.Time `json:"deleted_at,omitempty"`
  27. // 服务器id
  28. ServerID uint64 `json:"server_id,omitempty"`
  29. // 端口号
  30. Port string `json:"port,omitempty"`
  31. // 进程号
  32. ProcessID string `json:"process_id,omitempty"`
  33. // 回调地址
  34. Callback string `json:"callback,omitempty"`
  35. // 微信id
  36. Wxid string `json:"wxid,omitempty"`
  37. // 微信账号
  38. Account string `json:"account,omitempty"`
  39. // 微信昵称
  40. Nickname string `json:"nickname,omitempty"`
  41. // 手机号
  42. Tel string `json:"tel,omitempty"`
  43. // 微信头像
  44. HeadBig string `json:"head_big,omitempty"`
  45. // 机构 ID
  46. OrganizationID uint64 `json:"organization_id,omitempty"`
  47. // 模式ID
  48. AgentID uint64 `json:"agent_id,omitempty"`
  49. // 大模型服务地址
  50. APIBase string `json:"api_base,omitempty"`
  51. // 大模型服务密钥
  52. APIKey string `json:"api_key,omitempty"`
  53. // 白名单
  54. AllowList []string `json:"allow_list,omitempty"`
  55. // 群白名单
  56. GroupAllowList []string `json:"group_allow_list,omitempty"`
  57. // 黑名单
  58. BlockList []string `json:"block_list,omitempty"`
  59. // 群黑名单
  60. GroupBlockList []string `json:"group_block_list,omitempty"`
  61. // 账号类型:1-个微 2-企微
  62. Ctype uint64 `json:"ctype,omitempty"`
  63. // Edges holds the relations/edges for other nodes in the graph.
  64. // The values are being populated by the WxQuery when eager-loading is set.
  65. Edges WxEdges `json:"edges"`
  66. agent_base_wx_agent *string
  67. selectValues sql.SelectValues
  68. }
  69. // WxEdges holds the relations/edges for other nodes in the graph.
  70. type WxEdges struct {
  71. // Server holds the value of the server edge.
  72. Server *Server `json:"server,omitempty"`
  73. // Agent holds the value of the agent edge.
  74. Agent *Agent `json:"agent,omitempty"`
  75. // loadedTypes holds the information for reporting if a
  76. // type was loaded (or requested) in eager-loading or not.
  77. loadedTypes [2]bool
  78. }
  79. // ServerOrErr returns the Server value or an error if the edge
  80. // was not loaded in eager-loading, or loaded but was not found.
  81. func (e WxEdges) ServerOrErr() (*Server, error) {
  82. if e.Server != nil {
  83. return e.Server, nil
  84. } else if e.loadedTypes[0] {
  85. return nil, &NotFoundError{label: server.Label}
  86. }
  87. return nil, &NotLoadedError{edge: "server"}
  88. }
  89. // AgentOrErr returns the Agent value or an error if the edge
  90. // was not loaded in eager-loading, or loaded but was not found.
  91. func (e WxEdges) AgentOrErr() (*Agent, error) {
  92. if e.Agent != nil {
  93. return e.Agent, nil
  94. } else if e.loadedTypes[1] {
  95. return nil, &NotFoundError{label: agent.Label}
  96. }
  97. return nil, &NotLoadedError{edge: "agent"}
  98. }
  99. // scanValues returns the types for scanning values from sql.Rows.
  100. func (*Wx) scanValues(columns []string) ([]any, error) {
  101. values := make([]any, len(columns))
  102. for i := range columns {
  103. switch columns[i] {
  104. case wx.FieldAllowList, wx.FieldGroupAllowList, wx.FieldBlockList, wx.FieldGroupBlockList:
  105. values[i] = new([]byte)
  106. case wx.FieldID, wx.FieldStatus, wx.FieldServerID, wx.FieldOrganizationID, wx.FieldAgentID, wx.FieldCtype:
  107. values[i] = new(sql.NullInt64)
  108. case wx.FieldPort, wx.FieldProcessID, wx.FieldCallback, wx.FieldWxid, wx.FieldAccount, wx.FieldNickname, wx.FieldTel, wx.FieldHeadBig, wx.FieldAPIBase, wx.FieldAPIKey:
  109. values[i] = new(sql.NullString)
  110. case wx.FieldCreatedAt, wx.FieldUpdatedAt, wx.FieldDeletedAt:
  111. values[i] = new(sql.NullTime)
  112. case wx.ForeignKeys[0]: // agent_base_wx_agent
  113. values[i] = new(sql.NullString)
  114. default:
  115. values[i] = new(sql.UnknownType)
  116. }
  117. }
  118. return values, nil
  119. }
  120. // assignValues assigns the values that were returned from sql.Rows (after scanning)
  121. // to the Wx fields.
  122. func (w *Wx) assignValues(columns []string, values []any) error {
  123. if m, n := len(values), len(columns); m < n {
  124. return fmt.Errorf("mismatch number of scan values: %d != %d", m, n)
  125. }
  126. for i := range columns {
  127. switch columns[i] {
  128. case wx.FieldID:
  129. value, ok := values[i].(*sql.NullInt64)
  130. if !ok {
  131. return fmt.Errorf("unexpected type %T for field id", value)
  132. }
  133. w.ID = uint64(value.Int64)
  134. case wx.FieldCreatedAt:
  135. if value, ok := values[i].(*sql.NullTime); !ok {
  136. return fmt.Errorf("unexpected type %T for field created_at", values[i])
  137. } else if value.Valid {
  138. w.CreatedAt = value.Time
  139. }
  140. case wx.FieldUpdatedAt:
  141. if value, ok := values[i].(*sql.NullTime); !ok {
  142. return fmt.Errorf("unexpected type %T for field updated_at", values[i])
  143. } else if value.Valid {
  144. w.UpdatedAt = value.Time
  145. }
  146. case wx.FieldStatus:
  147. if value, ok := values[i].(*sql.NullInt64); !ok {
  148. return fmt.Errorf("unexpected type %T for field status", values[i])
  149. } else if value.Valid {
  150. w.Status = uint8(value.Int64)
  151. }
  152. case wx.FieldDeletedAt:
  153. if value, ok := values[i].(*sql.NullTime); !ok {
  154. return fmt.Errorf("unexpected type %T for field deleted_at", values[i])
  155. } else if value.Valid {
  156. w.DeletedAt = value.Time
  157. }
  158. case wx.FieldServerID:
  159. if value, ok := values[i].(*sql.NullInt64); !ok {
  160. return fmt.Errorf("unexpected type %T for field server_id", values[i])
  161. } else if value.Valid {
  162. w.ServerID = uint64(value.Int64)
  163. }
  164. case wx.FieldPort:
  165. if value, ok := values[i].(*sql.NullString); !ok {
  166. return fmt.Errorf("unexpected type %T for field port", values[i])
  167. } else if value.Valid {
  168. w.Port = value.String
  169. }
  170. case wx.FieldProcessID:
  171. if value, ok := values[i].(*sql.NullString); !ok {
  172. return fmt.Errorf("unexpected type %T for field process_id", values[i])
  173. } else if value.Valid {
  174. w.ProcessID = value.String
  175. }
  176. case wx.FieldCallback:
  177. if value, ok := values[i].(*sql.NullString); !ok {
  178. return fmt.Errorf("unexpected type %T for field callback", values[i])
  179. } else if value.Valid {
  180. w.Callback = value.String
  181. }
  182. case wx.FieldWxid:
  183. if value, ok := values[i].(*sql.NullString); !ok {
  184. return fmt.Errorf("unexpected type %T for field wxid", values[i])
  185. } else if value.Valid {
  186. w.Wxid = value.String
  187. }
  188. case wx.FieldAccount:
  189. if value, ok := values[i].(*sql.NullString); !ok {
  190. return fmt.Errorf("unexpected type %T for field account", values[i])
  191. } else if value.Valid {
  192. w.Account = value.String
  193. }
  194. case wx.FieldNickname:
  195. if value, ok := values[i].(*sql.NullString); !ok {
  196. return fmt.Errorf("unexpected type %T for field nickname", values[i])
  197. } else if value.Valid {
  198. w.Nickname = value.String
  199. }
  200. case wx.FieldTel:
  201. if value, ok := values[i].(*sql.NullString); !ok {
  202. return fmt.Errorf("unexpected type %T for field tel", values[i])
  203. } else if value.Valid {
  204. w.Tel = value.String
  205. }
  206. case wx.FieldHeadBig:
  207. if value, ok := values[i].(*sql.NullString); !ok {
  208. return fmt.Errorf("unexpected type %T for field head_big", values[i])
  209. } else if value.Valid {
  210. w.HeadBig = value.String
  211. }
  212. case wx.FieldOrganizationID:
  213. if value, ok := values[i].(*sql.NullInt64); !ok {
  214. return fmt.Errorf("unexpected type %T for field organization_id", values[i])
  215. } else if value.Valid {
  216. w.OrganizationID = uint64(value.Int64)
  217. }
  218. case wx.FieldAgentID:
  219. if value, ok := values[i].(*sql.NullInt64); !ok {
  220. return fmt.Errorf("unexpected type %T for field agent_id", values[i])
  221. } else if value.Valid {
  222. w.AgentID = uint64(value.Int64)
  223. }
  224. case wx.FieldAPIBase:
  225. if value, ok := values[i].(*sql.NullString); !ok {
  226. return fmt.Errorf("unexpected type %T for field api_base", values[i])
  227. } else if value.Valid {
  228. w.APIBase = value.String
  229. }
  230. case wx.FieldAPIKey:
  231. if value, ok := values[i].(*sql.NullString); !ok {
  232. return fmt.Errorf("unexpected type %T for field api_key", values[i])
  233. } else if value.Valid {
  234. w.APIKey = value.String
  235. }
  236. case wx.FieldAllowList:
  237. if value, ok := values[i].(*[]byte); !ok {
  238. return fmt.Errorf("unexpected type %T for field allow_list", values[i])
  239. } else if value != nil && len(*value) > 0 {
  240. if err := json.Unmarshal(*value, &w.AllowList); err != nil {
  241. return fmt.Errorf("unmarshal field allow_list: %w", err)
  242. }
  243. }
  244. case wx.FieldGroupAllowList:
  245. if value, ok := values[i].(*[]byte); !ok {
  246. return fmt.Errorf("unexpected type %T for field group_allow_list", values[i])
  247. } else if value != nil && len(*value) > 0 {
  248. if err := json.Unmarshal(*value, &w.GroupAllowList); err != nil {
  249. return fmt.Errorf("unmarshal field group_allow_list: %w", err)
  250. }
  251. }
  252. case wx.FieldBlockList:
  253. if value, ok := values[i].(*[]byte); !ok {
  254. return fmt.Errorf("unexpected type %T for field block_list", values[i])
  255. } else if value != nil && len(*value) > 0 {
  256. if err := json.Unmarshal(*value, &w.BlockList); err != nil {
  257. return fmt.Errorf("unmarshal field block_list: %w", err)
  258. }
  259. }
  260. case wx.FieldGroupBlockList:
  261. if value, ok := values[i].(*[]byte); !ok {
  262. return fmt.Errorf("unexpected type %T for field group_block_list", values[i])
  263. } else if value != nil && len(*value) > 0 {
  264. if err := json.Unmarshal(*value, &w.GroupBlockList); err != nil {
  265. return fmt.Errorf("unmarshal field group_block_list: %w", err)
  266. }
  267. }
  268. case wx.FieldCtype:
  269. if value, ok := values[i].(*sql.NullInt64); !ok {
  270. return fmt.Errorf("unexpected type %T for field ctype", values[i])
  271. } else if value.Valid {
  272. w.Ctype = uint64(value.Int64)
  273. }
  274. case wx.ForeignKeys[0]:
  275. if value, ok := values[i].(*sql.NullString); !ok {
  276. return fmt.Errorf("unexpected type %T for field agent_base_wx_agent", values[i])
  277. } else if value.Valid {
  278. w.agent_base_wx_agent = new(string)
  279. *w.agent_base_wx_agent = value.String
  280. }
  281. default:
  282. w.selectValues.Set(columns[i], values[i])
  283. }
  284. }
  285. return nil
  286. }
  287. // Value returns the ent.Value that was dynamically selected and assigned to the Wx.
  288. // This includes values selected through modifiers, order, etc.
  289. func (w *Wx) Value(name string) (ent.Value, error) {
  290. return w.selectValues.Get(name)
  291. }
  292. // QueryServer queries the "server" edge of the Wx entity.
  293. func (w *Wx) QueryServer() *ServerQuery {
  294. return NewWxClient(w.config).QueryServer(w)
  295. }
  296. // QueryAgent queries the "agent" edge of the Wx entity.
  297. func (w *Wx) QueryAgent() *AgentQuery {
  298. return NewWxClient(w.config).QueryAgent(w)
  299. }
  300. // Update returns a builder for updating this Wx.
  301. // Note that you need to call Wx.Unwrap() before calling this method if this Wx
  302. // was returned from a transaction, and the transaction was committed or rolled back.
  303. func (w *Wx) Update() *WxUpdateOne {
  304. return NewWxClient(w.config).UpdateOne(w)
  305. }
  306. // Unwrap unwraps the Wx entity that was returned from a transaction after it was closed,
  307. // so that all future queries will be executed through the driver which created the transaction.
  308. func (w *Wx) Unwrap() *Wx {
  309. _tx, ok := w.config.driver.(*txDriver)
  310. if !ok {
  311. panic("ent: Wx is not a transactional entity")
  312. }
  313. w.config.driver = _tx.drv
  314. return w
  315. }
  316. // String implements the fmt.Stringer.
  317. func (w *Wx) String() string {
  318. var builder strings.Builder
  319. builder.WriteString("Wx(")
  320. builder.WriteString(fmt.Sprintf("id=%v, ", w.ID))
  321. builder.WriteString("created_at=")
  322. builder.WriteString(w.CreatedAt.Format(time.ANSIC))
  323. builder.WriteString(", ")
  324. builder.WriteString("updated_at=")
  325. builder.WriteString(w.UpdatedAt.Format(time.ANSIC))
  326. builder.WriteString(", ")
  327. builder.WriteString("status=")
  328. builder.WriteString(fmt.Sprintf("%v", w.Status))
  329. builder.WriteString(", ")
  330. builder.WriteString("deleted_at=")
  331. builder.WriteString(w.DeletedAt.Format(time.ANSIC))
  332. builder.WriteString(", ")
  333. builder.WriteString("server_id=")
  334. builder.WriteString(fmt.Sprintf("%v", w.ServerID))
  335. builder.WriteString(", ")
  336. builder.WriteString("port=")
  337. builder.WriteString(w.Port)
  338. builder.WriteString(", ")
  339. builder.WriteString("process_id=")
  340. builder.WriteString(w.ProcessID)
  341. builder.WriteString(", ")
  342. builder.WriteString("callback=")
  343. builder.WriteString(w.Callback)
  344. builder.WriteString(", ")
  345. builder.WriteString("wxid=")
  346. builder.WriteString(w.Wxid)
  347. builder.WriteString(", ")
  348. builder.WriteString("account=")
  349. builder.WriteString(w.Account)
  350. builder.WriteString(", ")
  351. builder.WriteString("nickname=")
  352. builder.WriteString(w.Nickname)
  353. builder.WriteString(", ")
  354. builder.WriteString("tel=")
  355. builder.WriteString(w.Tel)
  356. builder.WriteString(", ")
  357. builder.WriteString("head_big=")
  358. builder.WriteString(w.HeadBig)
  359. builder.WriteString(", ")
  360. builder.WriteString("organization_id=")
  361. builder.WriteString(fmt.Sprintf("%v", w.OrganizationID))
  362. builder.WriteString(", ")
  363. builder.WriteString("agent_id=")
  364. builder.WriteString(fmt.Sprintf("%v", w.AgentID))
  365. builder.WriteString(", ")
  366. builder.WriteString("api_base=")
  367. builder.WriteString(w.APIBase)
  368. builder.WriteString(", ")
  369. builder.WriteString("api_key=")
  370. builder.WriteString(w.APIKey)
  371. builder.WriteString(", ")
  372. builder.WriteString("allow_list=")
  373. builder.WriteString(fmt.Sprintf("%v", w.AllowList))
  374. builder.WriteString(", ")
  375. builder.WriteString("group_allow_list=")
  376. builder.WriteString(fmt.Sprintf("%v", w.GroupAllowList))
  377. builder.WriteString(", ")
  378. builder.WriteString("block_list=")
  379. builder.WriteString(fmt.Sprintf("%v", w.BlockList))
  380. builder.WriteString(", ")
  381. builder.WriteString("group_block_list=")
  382. builder.WriteString(fmt.Sprintf("%v", w.GroupBlockList))
  383. builder.WriteString(", ")
  384. builder.WriteString("ctype=")
  385. builder.WriteString(fmt.Sprintf("%v", w.Ctype))
  386. builder.WriteByte(')')
  387. return builder.String()
  388. }
  389. // Wxes is a parsable slice of Wx.
  390. type Wxes []*Wx