client.go 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957
  1. package dpsv1
  2. import (
  3. "context"
  4. "encoding/json"
  5. "errors"
  6. "git.sxidc.com/service-supports/dps-sdk/client"
  7. "git.sxidc.com/service-supports/dps-sdk/pb/v1"
  8. "git.sxidc.com/service-supports/dps-sdk/pb/v1/request"
  9. "google.golang.org/grpc"
  10. "google.golang.org/grpc/credentials/insecure"
  11. "io"
  12. "sync"
  13. )
  14. type Client struct {
  15. databaseID string
  16. conn *grpc.ClientConn
  17. commandServiceClient v1.CommandServiceClient
  18. queryServiceClient v1.QueryServiceClient
  19. eventQueryServiceClient v1.EventQueryServiceClient
  20. transactionMutex sync.Mutex
  21. }
  22. func NewClient(address string, databaseID string) (*Client, error) {
  23. conn, err := grpc.DialContext(context.Background(), address,
  24. grpc.WithTransportCredentials(insecure.NewCredentials()))
  25. if err != nil {
  26. return nil, err
  27. }
  28. return &Client{
  29. databaseID: databaseID,
  30. conn: conn,
  31. commandServiceClient: v1.NewCommandServiceClient(conn),
  32. queryServiceClient: v1.NewQueryServiceClient(conn),
  33. eventQueryServiceClient: v1.NewEventQueryServiceClient(conn),
  34. }, nil
  35. }
  36. func DestroyClient(client *Client) error {
  37. if client == nil {
  38. return nil
  39. }
  40. client.transactionMutex.Lock()
  41. defer client.transactionMutex.Unlock()
  42. err := client.conn.Close()
  43. if err != nil {
  44. return err
  45. }
  46. client.databaseID = ""
  47. client.conn = nil
  48. client.commandServiceClient = nil
  49. client.queryServiceClient = nil
  50. return nil
  51. }
  52. func (c *Client) AutoMigrate(req *client.AutoMigrateRequest) error {
  53. items := make([]*request.AutoMigrateItem, 0)
  54. for _, reqItem := range req.Items {
  55. tableModelDescribeJsonBytes, err := json.Marshal(reqItem.TableModelDescribe)
  56. if err != nil {
  57. return err
  58. }
  59. items = append(items, &request.AutoMigrateItem{
  60. TablePrefixWithSchema: reqItem.TablePrefixWithSchema,
  61. Version: reqItem.Version,
  62. TableModelDescribe: tableModelDescribeJsonBytes,
  63. NoEvent: reqItem.NoEvent,
  64. })
  65. }
  66. _, err := c.commandServiceClient.AutoMigrate(context.Background(), &request.AutoMigrateRequest{
  67. DatabaseID: c.databaseID,
  68. Items: items,
  69. })
  70. if err != nil {
  71. return err
  72. }
  73. return nil
  74. }
  75. func (c *Client) Transaction(txFunc client.TransactionFunc) error {
  76. stream, err := c.commandServiceClient.Transaction(context.Background())
  77. if err != nil {
  78. return err
  79. }
  80. defer func() {
  81. innerErr := stream.CloseSend()
  82. if innerErr != nil {
  83. panic(innerErr)
  84. }
  85. }()
  86. err = stream.Send(&request.TransactionOperation{
  87. Request: &request.TransactionOperation_TransactionBeginRequest{
  88. TransactionBeginRequest: &request.TransactionBeginRequest{DatabaseID: c.databaseID},
  89. }})
  90. if err != nil {
  91. return err
  92. }
  93. err = txFunc(&Transaction{
  94. stream: stream,
  95. client: c,
  96. })
  97. if err != nil {
  98. return err
  99. }
  100. err = stream.Send(&request.TransactionOperation{
  101. Request: &request.TransactionOperation_TransactionEndRequest{
  102. TransactionEndRequest: &request.TransactionEndRequest{},
  103. }})
  104. if err != nil {
  105. return err
  106. }
  107. _, err = stream.Recv()
  108. if err != nil && err != io.EOF {
  109. return err
  110. }
  111. return nil
  112. }
  113. func (c *Client) Insert(req *client.InsertRequest) (string, error) {
  114. if req.TableRow == nil {
  115. return "", nil
  116. }
  117. tableRow, err := req.TableRow.ToDPSTableRow()
  118. if err != nil {
  119. return "", err
  120. }
  121. reply, err := c.commandServiceClient.Insert(context.Background(), &request.InsertRequest{
  122. DatabaseID: c.databaseID,
  123. TablePrefixWithSchema: req.TablePrefixWithSchema,
  124. Version: req.Version,
  125. KeyColumns: req.KeyColumns,
  126. TableRow: tableRow,
  127. UserID: req.UserID,
  128. })
  129. if err != nil {
  130. return "", err
  131. }
  132. return reply.Statement, nil
  133. }
  134. func (c *Client) InsertBatch(req *client.InsertBatchRequest) (string, error) {
  135. tableRowItems := make([]*request.InsertTableRowItem, 0)
  136. for _, reqTableItem := range req.Items {
  137. tableRows := make([]*request.TableRow, 0)
  138. for _, reqTableRow := range reqTableItem.TableRows {
  139. if reqTableRow == nil {
  140. continue
  141. }
  142. tableRow, err := reqTableRow.ToDPSTableRow()
  143. if err != nil {
  144. return "", err
  145. }
  146. tableRows = append(tableRows, tableRow)
  147. }
  148. tableRowItems = append(tableRowItems, &request.InsertTableRowItem{
  149. TablePrefixWithSchema: reqTableItem.TablePrefixWithSchema,
  150. Version: reqTableItem.Version,
  151. KeyColumns: reqTableItem.KeyColumns,
  152. TableRows: tableRows,
  153. })
  154. }
  155. reply, err := c.commandServiceClient.InsertBatch(context.Background(), &request.InsertBatchRequest{
  156. DatabaseID: c.databaseID,
  157. Items: tableRowItems,
  158. UserID: req.UserID,
  159. })
  160. if err != nil {
  161. return "", err
  162. }
  163. return reply.Statement, nil
  164. }
  165. func (c *Client) Delete(req *client.DeleteRequest) (string, error) {
  166. reply, err := c.commandServiceClient.Delete(context.Background(), &request.DeleteRequest{
  167. DatabaseID: c.databaseID,
  168. TablePrefixWithSchema: req.TablePrefixWithSchema,
  169. Version: req.Version,
  170. KeyColumns: req.KeyValues.Columns(),
  171. KeyValues: req.KeyValues.Values(),
  172. UserID: req.UserID,
  173. })
  174. if err != nil {
  175. return "", err
  176. }
  177. return reply.Statement, nil
  178. }
  179. func (c *Client) DeleteWhere(req *client.DeleteWhereRequest) (string, error) {
  180. if req.Where == nil {
  181. return "", errors.New("没有传递Where条件")
  182. }
  183. whereJsonBytes, err := req.Where.ToJson()
  184. if err != nil {
  185. return "", err
  186. }
  187. reply, err := c.commandServiceClient.DeleteWhere(context.Background(), &request.DeleteWhereRequest{
  188. DatabaseID: c.databaseID,
  189. TablePrefixWithSchema: req.TablePrefixWithSchema,
  190. Version: req.Version,
  191. KeyColumns: req.KeyColumns,
  192. Where: whereJsonBytes,
  193. UserID: req.UserID,
  194. })
  195. if err != nil {
  196. return "", err
  197. }
  198. return reply.Statement, nil
  199. }
  200. func (c *Client) Update(req *client.UpdateRequest) (string, error) {
  201. if req.NewTableRow == nil {
  202. return "", nil
  203. }
  204. newTableRow, err := req.NewTableRow.ToDPSTableRow()
  205. if err != nil {
  206. return "", err
  207. }
  208. reply, err := c.commandServiceClient.Update(context.Background(), &request.UpdateRequest{
  209. DatabaseID: c.databaseID,
  210. TablePrefixWithSchema: req.TablePrefixWithSchema,
  211. Version: req.Version,
  212. KeyColumns: req.KeyValues.Columns(),
  213. KeyValues: req.KeyValues.Values(),
  214. NewTableRow: newTableRow,
  215. UserID: req.UserID,
  216. })
  217. if err != nil {
  218. return "", err
  219. }
  220. return reply.Statement, nil
  221. }
  222. func (c *Client) UpdateWhere(req *client.UpdateWhereRequest) (string, error) {
  223. if req.Where == nil {
  224. return "", errors.New("没有传递Where条件")
  225. }
  226. if req.NewTableRow == nil {
  227. return "", nil
  228. }
  229. whereJsonBytes, err := req.Where.ToJson()
  230. if err != nil {
  231. return "", err
  232. }
  233. newTableRow, err := req.NewTableRow.ToDPSTableRow()
  234. if err != nil {
  235. return "", err
  236. }
  237. reply, err := c.commandServiceClient.UpdateWhere(context.Background(), &request.UpdateWhereRequest{
  238. DatabaseID: c.databaseID,
  239. TablePrefixWithSchema: req.TablePrefixWithSchema,
  240. Version: req.Version,
  241. KeyColumns: req.KeyColumns,
  242. Where: whereJsonBytes,
  243. NewTableRow: newTableRow,
  244. UserID: req.UserID,
  245. })
  246. if err != nil {
  247. return "", err
  248. }
  249. return reply.Statement, nil
  250. }
  251. func (c *Client) Replay(req *client.ReplayRequest) (string, error) {
  252. reply, err := c.commandServiceClient.Replay(context.Background(), &request.ReplayRequest{
  253. DatabaseID: c.databaseID,
  254. TablePrefixWithSchema: req.TablePrefixWithSchema,
  255. Version: req.Version,
  256. KeyColumns: req.KeyValues.Columns(),
  257. KeyValues: req.KeyValues.Values(),
  258. UserID: req.UserID,
  259. })
  260. if err != nil {
  261. return "", err
  262. }
  263. return reply.Statement, nil
  264. }
  265. func (c *Client) QueryByWhereAndOrderBy(req *client.QueryByWhereAndOrderByRequest) (string, []client.TableRow, int64, error) {
  266. var selectJsonBytes []byte
  267. if req.Select != nil {
  268. innerJsonBytes, err := req.Select.ToJson()
  269. if err != nil {
  270. return "", nil, 0, err
  271. }
  272. selectJsonBytes = innerJsonBytes
  273. }
  274. var whereJsonBytes []byte
  275. if req.Where != nil {
  276. innerJsonBytes, err := req.Where.ToJson()
  277. if err != nil {
  278. return "", nil, 0, err
  279. }
  280. whereJsonBytes = innerJsonBytes
  281. }
  282. reply, err := c.queryServiceClient.QueryByWhereAndOrderBy(context.Background(), &request.QueryByWhereAndOrderByRequest{
  283. DatabaseID: c.databaseID,
  284. TablePrefixWithSchema: req.TablePrefixWithSchema,
  285. Version: req.Version,
  286. Select: selectJsonBytes,
  287. Where: whereJsonBytes,
  288. OrderBy: req.OrderBy,
  289. PageNo: int32(req.PageNo),
  290. PageSize: int32(req.PageSize),
  291. })
  292. if err != nil {
  293. return "", nil, 0, err
  294. }
  295. tableRows, err := client.FromDSPInfosData(reply.Infos)
  296. if err != nil {
  297. return "", nil, 0, err
  298. }
  299. return reply.Statement, tableRows, reply.TotalCount, nil
  300. }
  301. func (c *Client) CommonQuery(req *client.CommonQueryRequest) (string, []client.TableRow, int64, error) {
  302. var selectJsonBytes []byte
  303. if req.Select != nil {
  304. innerJsonBytes, err := req.Select.ToJson()
  305. if err != nil {
  306. return "", nil, 0, err
  307. }
  308. selectJsonBytes = innerJsonBytes
  309. }
  310. var whereJsonBytes []byte
  311. if req.Where != nil {
  312. innerJsonBytes, err := req.Where.ToJson()
  313. if err != nil {
  314. return "", nil, 0, err
  315. }
  316. whereJsonBytes = innerJsonBytes
  317. }
  318. var orJsonBytes []byte
  319. if req.Or != nil {
  320. innerJsonBytes, err := req.Or.ToJson()
  321. if err != nil {
  322. return "", nil, 0, err
  323. }
  324. orJsonBytes = innerJsonBytes
  325. }
  326. var joinsJsonBytes []byte
  327. if req.Joins != nil {
  328. innerJsonBytes, err := req.Joins.ToJson()
  329. if err != nil {
  330. return "", nil, 0, err
  331. }
  332. joinsJsonBytes = innerJsonBytes
  333. }
  334. var havingJsonBytes []byte
  335. if req.Having != nil {
  336. innerJsonBytes, err := req.Having.ToJson()
  337. if err != nil {
  338. return "", nil, 0, err
  339. }
  340. havingJsonBytes = innerJsonBytes
  341. }
  342. reply, err := c.queryServiceClient.CommonQuery(context.Background(), &request.CommonQueryRequest{
  343. DatabaseID: c.databaseID,
  344. TablePrefixWithSchema: req.TablePrefixWithSchema,
  345. Version: req.Version,
  346. Select: selectJsonBytes,
  347. Where: whereJsonBytes,
  348. OrderBy: req.OrderBy,
  349. Or: orJsonBytes,
  350. GroupBy: req.GroupBy,
  351. Joins: joinsJsonBytes,
  352. Having: havingJsonBytes,
  353. PageNo: int32(req.PageNo),
  354. PageSize: int32(req.PageSize),
  355. })
  356. if err != nil {
  357. return "", nil, 0, err
  358. }
  359. tableRows, err := client.FromDSPInfosData(reply.Infos)
  360. if err != nil {
  361. return "", nil, 0, err
  362. }
  363. return reply.Statement, tableRows, reply.TotalCount, nil
  364. }
  365. func (c *Client) QueryOnlyByWhereAndOrderBy(req *client.QueryByWhereAndOrderByRequest) (string, []client.TableRow, error) {
  366. var selectJsonBytes []byte
  367. if req.Select != nil {
  368. innerJsonBytes, err := req.Select.ToJson()
  369. if err != nil {
  370. return "", nil, err
  371. }
  372. selectJsonBytes = innerJsonBytes
  373. }
  374. var whereJsonBytes []byte
  375. if req.Where != nil {
  376. innerJsonBytes, err := req.Where.ToJson()
  377. if err != nil {
  378. return "", nil, err
  379. }
  380. whereJsonBytes = innerJsonBytes
  381. }
  382. reply, err := c.queryServiceClient.QueryOnlyByWhereAndOrderBy(context.Background(), &request.QueryByWhereAndOrderByRequest{
  383. DatabaseID: c.databaseID,
  384. TablePrefixWithSchema: req.TablePrefixWithSchema,
  385. Version: req.Version,
  386. Select: selectJsonBytes,
  387. Where: whereJsonBytes,
  388. OrderBy: req.OrderBy,
  389. PageNo: int32(req.PageNo),
  390. PageSize: int32(req.PageSize),
  391. })
  392. if err != nil {
  393. return "", nil, err
  394. }
  395. tableRows, err := client.FromDSPInfosData(reply.Infos)
  396. if err != nil {
  397. return "", nil, err
  398. }
  399. return reply.Statement, tableRows, nil
  400. }
  401. func (c *Client) CommonQueryOnly(req *client.CommonQueryRequest) (string, []client.TableRow, error) {
  402. var selectJsonBytes []byte
  403. if req.Select != nil {
  404. innerJsonBytes, err := req.Select.ToJson()
  405. if err != nil {
  406. return "", nil, err
  407. }
  408. selectJsonBytes = innerJsonBytes
  409. }
  410. var whereJsonBytes []byte
  411. if req.Where != nil {
  412. innerJsonBytes, err := req.Where.ToJson()
  413. if err != nil {
  414. return "", nil, err
  415. }
  416. whereJsonBytes = innerJsonBytes
  417. }
  418. var orJsonBytes []byte
  419. if req.Or != nil {
  420. innerJsonBytes, err := req.Or.ToJson()
  421. if err != nil {
  422. return "", nil, err
  423. }
  424. orJsonBytes = innerJsonBytes
  425. }
  426. var joinsJsonBytes []byte
  427. if req.Joins != nil {
  428. innerJsonBytes, err := req.Joins.ToJson()
  429. if err != nil {
  430. return "", nil, err
  431. }
  432. joinsJsonBytes = innerJsonBytes
  433. }
  434. var havingJsonBytes []byte
  435. if req.Having != nil {
  436. innerJsonBytes, err := req.Having.ToJson()
  437. if err != nil {
  438. return "", nil, err
  439. }
  440. havingJsonBytes = innerJsonBytes
  441. }
  442. reply, err := c.queryServiceClient.CommonQueryOnly(context.Background(), &request.CommonQueryRequest{
  443. DatabaseID: c.databaseID,
  444. TablePrefixWithSchema: req.TablePrefixWithSchema,
  445. Version: req.Version,
  446. Select: selectJsonBytes,
  447. Where: whereJsonBytes,
  448. OrderBy: req.OrderBy,
  449. Or: orJsonBytes,
  450. GroupBy: req.GroupBy,
  451. Joins: joinsJsonBytes,
  452. Having: havingJsonBytes,
  453. PageNo: int32(req.PageNo),
  454. PageSize: int32(req.PageSize),
  455. })
  456. if err != nil {
  457. return "", nil, err
  458. }
  459. tableRows, err := client.FromDSPInfosData(reply.Infos)
  460. if err != nil {
  461. return "", nil, err
  462. }
  463. return reply.Statement, tableRows, nil
  464. }
  465. func (c *Client) QueryByKeys(req *client.QueryByKeysRequest) (string, *client.TableRow, error) {
  466. var selectJsonBytes []byte
  467. if req.Select != nil {
  468. innerJsonBytes, err := req.Select.ToJson()
  469. if err != nil {
  470. return "", nil, err
  471. }
  472. selectJsonBytes = innerJsonBytes
  473. }
  474. reply, err := c.queryServiceClient.QueryByKeys(context.Background(), &request.QueryByKeysRequest{
  475. DatabaseID: c.databaseID,
  476. TablePrefixWithSchema: req.TablePrefixWithSchema,
  477. Version: req.Version,
  478. Select: selectJsonBytes,
  479. KeyColumns: req.KeyValues.Columns(),
  480. KeyValues: req.KeyValues.Values(),
  481. })
  482. if err != nil {
  483. return "", nil, err
  484. }
  485. tableRow := client.NewTableRow()
  486. err = tableRow.FromDSPInfoData(reply.Info)
  487. if err != nil {
  488. return "", nil, err
  489. }
  490. return reply.Statement, tableRow, nil
  491. }
  492. func (c *Client) CountWhere(req *client.CountWhereRequest) (string, int64, error) {
  493. var whereJsonBytes []byte
  494. if req.Where != nil {
  495. innerJsonBytes, err := req.Where.ToJson()
  496. if err != nil {
  497. return "", 0, err
  498. }
  499. whereJsonBytes = innerJsonBytes
  500. }
  501. reply, err := c.queryServiceClient.CountWhere(context.Background(), &request.CountWhereRequest{
  502. DatabaseID: c.databaseID,
  503. TablePrefixWithSchema: req.TablePrefixWithSchema,
  504. Version: req.Version,
  505. Where: whereJsonBytes,
  506. })
  507. if err != nil {
  508. return "", 0, err
  509. }
  510. return reply.Statement, reply.Count, nil
  511. }
  512. func (c *Client) CommonCount(req *client.CommonCountRequest) (string, int64, error) {
  513. var whereJsonBytes []byte
  514. if req.Where != nil {
  515. innerJsonBytes, err := req.Where.ToJson()
  516. if err != nil {
  517. return "", 0, err
  518. }
  519. whereJsonBytes = innerJsonBytes
  520. }
  521. var orJsonBytes []byte
  522. if req.Or != nil {
  523. innerJsonBytes, err := req.Or.ToJson()
  524. if err != nil {
  525. return "", 0, err
  526. }
  527. orJsonBytes = innerJsonBytes
  528. }
  529. var joinsJsonBytes []byte
  530. if req.Joins != nil {
  531. innerJsonBytes, err := req.Joins.ToJson()
  532. if err != nil {
  533. return "", 0, err
  534. }
  535. joinsJsonBytes = innerJsonBytes
  536. }
  537. var havingJsonBytes []byte
  538. if req.Having != nil {
  539. innerJsonBytes, err := req.Having.ToJson()
  540. if err != nil {
  541. return "", 0, err
  542. }
  543. havingJsonBytes = innerJsonBytes
  544. }
  545. reply, err := c.queryServiceClient.CommonCount(context.Background(), &request.CommonCountRequest{
  546. DatabaseID: c.databaseID,
  547. TablePrefixWithSchema: req.TablePrefixWithSchema,
  548. Version: req.Version,
  549. Where: whereJsonBytes,
  550. Or: orJsonBytes,
  551. GroupBy: req.GroupBy,
  552. Joins: joinsJsonBytes,
  553. Having: havingJsonBytes,
  554. })
  555. if err != nil {
  556. return "", 0, err
  557. }
  558. return reply.Statement, reply.Count, nil
  559. }
  560. func (c *Client) EventQueryByKeys(req *client.EventQueryByKeysRequest) (string, []client.EventInfo, int64, error) {
  561. var selectJsonBytes []byte
  562. if req.Select != nil {
  563. innerJsonBytes, err := req.Select.ToJson()
  564. if err != nil {
  565. return "", nil, 0, err
  566. }
  567. selectJsonBytes = innerJsonBytes
  568. }
  569. reply, err := c.eventQueryServiceClient.EventQueryByKeys(context.Background(), &request.EventQueryByKeysRequest{
  570. DatabaseID: c.databaseID,
  571. TablePrefixWithSchema: req.TablePrefixWithSchema,
  572. Select: selectJsonBytes,
  573. KeyValues: req.KeyValues,
  574. PageNo: int32(req.PageNo),
  575. PageSize: int32(req.PageSize),
  576. })
  577. if err != nil {
  578. return "", nil, 0, err
  579. }
  580. return reply.Statement, client.FormEventInfoBatch(reply.Infos), reply.TotalCount, nil
  581. }
  582. func (c *Client) CommonEventQuery(req *client.CommonEventQueryRequest) (string, []client.EventInfo, int64, error) {
  583. var selectJsonBytes []byte
  584. if req.Select != nil {
  585. innerJsonBytes, err := req.Select.ToJson()
  586. if err != nil {
  587. return "", nil, 0, err
  588. }
  589. selectJsonBytes = innerJsonBytes
  590. }
  591. reply, err := c.eventQueryServiceClient.CommonEventQuery(context.Background(), &request.CommonEventQueryRequest{
  592. DatabaseID: c.databaseID,
  593. TablePrefixWithSchema: req.TablePrefixWithSchema,
  594. Select: selectJsonBytes,
  595. KeyValues: req.KeyValues,
  596. Version: req.Version,
  597. Operation: req.Operation,
  598. CreatorID: req.CreatorID,
  599. StartCreatedTime: req.StartCreatedTime,
  600. EndCreatedTime: req.EndCreatedTime,
  601. PageNo: int32(req.PageNo),
  602. PageSize: int32(req.PageSize),
  603. })
  604. if err != nil {
  605. return "", nil, 0, err
  606. }
  607. return reply.Statement, client.FormEventInfoBatch(reply.Infos), reply.TotalCount, nil
  608. }
  609. func (c *Client) EventQueryOnlyByKeys(req *client.EventQueryByKeysRequest) (string, []client.EventInfo, error) {
  610. var selectJsonBytes []byte
  611. if req.Select != nil {
  612. innerJsonBytes, err := req.Select.ToJson()
  613. if err != nil {
  614. return "", nil, err
  615. }
  616. selectJsonBytes = innerJsonBytes
  617. }
  618. reply, err := c.eventQueryServiceClient.EventQueryOnlyByKeys(context.Background(), &request.EventQueryByKeysRequest{
  619. DatabaseID: c.databaseID,
  620. TablePrefixWithSchema: req.TablePrefixWithSchema,
  621. Select: selectJsonBytes,
  622. KeyValues: req.KeyValues,
  623. PageNo: int32(req.PageNo),
  624. PageSize: int32(req.PageSize),
  625. })
  626. if err != nil {
  627. return "", nil, err
  628. }
  629. return reply.Statement, client.FormEventInfoBatch(reply.Infos), nil
  630. }
  631. func (c *Client) CommonEventQueryOnly(req *client.CommonEventQueryRequest) (string, []client.EventInfo, error) {
  632. var selectJsonBytes []byte
  633. if req.Select != nil {
  634. innerJsonBytes, err := req.Select.ToJson()
  635. if err != nil {
  636. return "", nil, err
  637. }
  638. selectJsonBytes = innerJsonBytes
  639. }
  640. reply, err := c.eventQueryServiceClient.CommonEventQueryOnly(context.Background(), &request.CommonEventQueryRequest{
  641. DatabaseID: c.databaseID,
  642. TablePrefixWithSchema: req.TablePrefixWithSchema,
  643. Select: selectJsonBytes,
  644. KeyValues: req.KeyValues,
  645. Version: req.Version,
  646. Operation: req.Operation,
  647. CreatorID: req.CreatorID,
  648. StartCreatedTime: req.StartCreatedTime,
  649. EndCreatedTime: req.EndCreatedTime,
  650. PageNo: int32(req.PageNo),
  651. PageSize: int32(req.PageSize),
  652. })
  653. if err != nil {
  654. return "", nil, err
  655. }
  656. return reply.Statement, client.FormEventInfoBatch(reply.Infos), nil
  657. }
  658. func (c *Client) CountEventByKeys(req *client.CountEventByKeysRequest) (string, int64, error) {
  659. reply, err := c.eventQueryServiceClient.CountEventByKeys(context.Background(), &request.CountEventByKeysRequest{
  660. DatabaseID: c.databaseID,
  661. TablePrefixWithSchema: req.TablePrefixWithSchema,
  662. KeyValues: req.KeyValues,
  663. })
  664. if err != nil {
  665. return "", 0, err
  666. }
  667. return reply.Statement, reply.Count, nil
  668. }
  669. func (c *Client) CommonCountEvent(req *client.CommonCountEventRequest) (string, int64, error) {
  670. reply, err := c.eventQueryServiceClient.CommonCountEvent(context.Background(), &request.CommonCountEventRequest{
  671. DatabaseID: c.databaseID,
  672. TablePrefixWithSchema: req.TablePrefixWithSchema,
  673. KeyValues: req.KeyValues,
  674. Version: req.Version,
  675. Operation: req.Operation,
  676. CreatorID: req.CreatorID,
  677. StartCreatedTime: req.StartCreatedTime,
  678. EndCreatedTime: req.EndCreatedTime,
  679. })
  680. if err != nil {
  681. return "", 0, err
  682. }
  683. return reply.Statement, reply.Count, nil
  684. }
  685. func (c *Client) EventHistoryQueryByKeys(req *client.EventQueryByKeysRequest) (string, []client.EventInfo, int64, error) {
  686. var selectJsonBytes []byte
  687. if req.Select != nil {
  688. innerJsonBytes, err := req.Select.ToJson()
  689. if err != nil {
  690. return "", nil, 0, err
  691. }
  692. selectJsonBytes = innerJsonBytes
  693. }
  694. reply, err := c.eventQueryServiceClient.EventHistoryQueryByKeys(context.Background(), &request.EventQueryByKeysRequest{
  695. DatabaseID: c.databaseID,
  696. TablePrefixWithSchema: req.TablePrefixWithSchema,
  697. Select: selectJsonBytes,
  698. KeyValues: req.KeyValues,
  699. PageNo: int32(req.PageNo),
  700. PageSize: int32(req.PageSize),
  701. })
  702. if err != nil {
  703. return "", nil, 0, err
  704. }
  705. return reply.Statement, client.FormEventInfoBatch(reply.Infos), reply.TotalCount, nil
  706. }
  707. func (c *Client) CommonEventHistoryQuery(req *client.CommonEventQueryRequest) (string, []client.EventInfo, int64, error) {
  708. var selectJsonBytes []byte
  709. if req.Select != nil {
  710. innerJsonBytes, err := req.Select.ToJson()
  711. if err != nil {
  712. return "", nil, 0, err
  713. }
  714. selectJsonBytes = innerJsonBytes
  715. }
  716. reply, err := c.eventQueryServiceClient.CommonEventHistoryQuery(context.Background(), &request.CommonEventQueryRequest{
  717. DatabaseID: c.databaseID,
  718. TablePrefixWithSchema: req.TablePrefixWithSchema,
  719. Select: selectJsonBytes,
  720. KeyValues: req.KeyValues,
  721. Version: req.Version,
  722. Operation: req.Operation,
  723. CreatorID: req.CreatorID,
  724. StartCreatedTime: req.StartCreatedTime,
  725. EndCreatedTime: req.EndCreatedTime,
  726. PageNo: int32(req.PageNo),
  727. PageSize: int32(req.PageSize),
  728. })
  729. if err != nil {
  730. return "", nil, 0, err
  731. }
  732. return reply.Statement, client.FormEventInfoBatch(reply.Infos), reply.TotalCount, nil
  733. }
  734. func (c *Client) EventHistoryQueryOnlyByKeys(req *client.EventQueryByKeysRequest) (string, []client.EventInfo, error) {
  735. var selectJsonBytes []byte
  736. if req.Select != nil {
  737. innerJsonBytes, err := req.Select.ToJson()
  738. if err != nil {
  739. return "", nil, err
  740. }
  741. selectJsonBytes = innerJsonBytes
  742. }
  743. reply, err := c.eventQueryServiceClient.EventHistoryQueryOnlyByKeys(context.Background(), &request.EventQueryByKeysRequest{
  744. DatabaseID: c.databaseID,
  745. TablePrefixWithSchema: req.TablePrefixWithSchema,
  746. Select: selectJsonBytes,
  747. KeyValues: req.KeyValues,
  748. PageNo: int32(req.PageNo),
  749. PageSize: int32(req.PageSize),
  750. })
  751. if err != nil {
  752. return "", nil, err
  753. }
  754. return reply.Statement, client.FormEventInfoBatch(reply.Infos), nil
  755. }
  756. func (c *Client) CommonEventHistoryQueryOnly(req *client.CommonEventQueryRequest) (string, []client.EventInfo, error) {
  757. var selectJsonBytes []byte
  758. if req.Select != nil {
  759. innerJsonBytes, err := req.Select.ToJson()
  760. if err != nil {
  761. return "", nil, err
  762. }
  763. selectJsonBytes = innerJsonBytes
  764. }
  765. reply, err := c.eventQueryServiceClient.CommonEventHistoryQueryOnly(context.Background(), &request.CommonEventQueryRequest{
  766. DatabaseID: c.databaseID,
  767. TablePrefixWithSchema: req.TablePrefixWithSchema,
  768. Select: selectJsonBytes,
  769. KeyValues: req.KeyValues,
  770. Version: req.Version,
  771. Operation: req.Operation,
  772. CreatorID: req.CreatorID,
  773. StartCreatedTime: req.StartCreatedTime,
  774. EndCreatedTime: req.EndCreatedTime,
  775. PageNo: int32(req.PageNo),
  776. PageSize: int32(req.PageSize),
  777. })
  778. if err != nil {
  779. return "", nil, err
  780. }
  781. return reply.Statement, client.FormEventInfoBatch(reply.Infos), nil
  782. }
  783. func (c *Client) CountEventHistoryByKeys(req *client.CountEventByKeysRequest) (string, int64, error) {
  784. reply, err := c.eventQueryServiceClient.CountEventHistoryByKeys(context.Background(), &request.CountEventByKeysRequest{
  785. DatabaseID: c.databaseID,
  786. TablePrefixWithSchema: req.TablePrefixWithSchema,
  787. KeyValues: req.KeyValues,
  788. })
  789. if err != nil {
  790. return "", 0, err
  791. }
  792. return reply.Statement, reply.Count, nil
  793. }
  794. func (c *Client) CommonCountEventHistory(req *client.CommonCountEventRequest) (string, int64, error) {
  795. reply, err := c.eventQueryServiceClient.CommonCountEventHistory(context.Background(), &request.CommonCountEventRequest{
  796. DatabaseID: c.databaseID,
  797. TablePrefixWithSchema: req.TablePrefixWithSchema,
  798. KeyValues: req.KeyValues,
  799. Version: req.Version,
  800. Operation: req.Operation,
  801. CreatorID: req.CreatorID,
  802. StartCreatedTime: req.StartCreatedTime,
  803. EndCreatedTime: req.EndCreatedTime,
  804. })
  805. if err != nil {
  806. return "", 0, err
  807. }
  808. return reply.Statement, reply.Count, nil
  809. }