odbclient_test.go 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  1. package client_test
  2. import (
  3. "fmt"
  4. "os"
  5. "strings"
  6. "testing"
  7. "time"
  8. "git.wecise.com/wecise/odb-go/odb"
  9. "gitee.com/wecisecode/util/logger"
  10. "gitee.com/wecisecode/util/set/strset"
  11. //"git.wecise.com/wecise/odbserver/odb/upsertupsert"
  12. )
  13. var odbcfg *odb.Config
  14. func init() {
  15. odbpath := os.Getenv("ODBPATH")
  16. if odbpath == "" {
  17. odbpath = "47.92.151.165:11001"
  18. }
  19. odbpaths := strset.New(strings.Split(odbpath, ",")...).List()
  20. keyspace := os.Getenv("KEYSPACE")
  21. if keyspace == "" {
  22. keyspace = "ooootest"
  23. }
  24. odbcfg = &odb.Config{
  25. Hosts: odbpaths,
  26. Keyspace: keyspace,
  27. User: fmt.Sprint("测试客户端"),
  28. Pass: "********",
  29. PoolSize: 20,
  30. Debug: true,
  31. }
  32. }
  33. func getClient() (c odb.Client, err error) {
  34. client, err := odb.NewClient(odbcfg)
  35. if err != nil {
  36. return nil, err
  37. }
  38. return client, nil
  39. }
  40. func TestClient2(t *testing.T) {
  41. time.Sleep(2 * time.Second)
  42. client, err := getClient()
  43. if err != nil {
  44. logger.Error(err)
  45. }
  46. defer func() {
  47. if err := client.Close(); err != nil {
  48. logger.Error(err)
  49. }
  50. }()
  51. //_, err = client.Query(`insert into /test/bucketpromdb (id, prom) values (?, ?)`, "bucketpromdb:wecise11", []map[string]interface{}{ map[string]interface{}{ "__name__":"key1", "__value__":2, "__timestamp__":"2024-08-27 11:24:17", "disk":"/opt/matrix", "CPU":"CPU1"}} ).Do()
  52. _, err = client.Query(`insert into /test/bucketpromdb (id, prom) values (?, ?)`, "bucketpromdb:wecise11", []map[string]interface{}{map[string]interface{}{"__name__": "key1", "__value__": 2, "__timestamp__": 1724729057000, "disk": "/opt/matrix", "CPU": "CPU1"}}).Do()
  53. if err != nil {
  54. logger.Error(err)
  55. }
  56. }