|
| 1 | +package mysql |
| 2 | + |
| 3 | +import ( |
| 4 | + "context" |
| 5 | + "database/sql" |
| 6 | + "database/sql/driver" |
| 7 | + "fmt" |
| 8 | + "reflect" |
| 9 | + "sync" |
| 10 | + "time" |
| 11 | + |
| 12 | + "github.com/DataObserve/datav/query/pkg/colorlog" |
| 13 | + "github.com/DataObserve/datav/query/pkg/models" |
| 14 | + "github.com/gin-gonic/gin" |
| 15 | + |
| 16 | + _ "github.com/go-sql-driver/mysql" |
| 17 | +) |
| 18 | + |
| 19 | +var datasourceName = "mysql" |
| 20 | + |
| 21 | +type MysqlPlugin struct { |
| 22 | +} |
| 23 | + |
| 24 | +var ( |
| 25 | + conns = make(map[int64]*sql.DB) |
| 26 | + connsLock = &sync.Mutex{} |
| 27 | +) |
| 28 | + |
| 29 | +func (*MysqlPlugin) Query(c *gin.Context, ds *models.Datasource) models.PluginResult { |
| 30 | + query := c.Query("query") |
| 31 | + conn, ok := conns[ds.Id] |
| 32 | + if !ok { |
| 33 | + conn, err := connectToMysql(ds.URL, ds.Data["database"], ds.Data["username"], ds.Data["password"]) |
| 34 | + if err != nil { |
| 35 | + colorlog.RootLogger.Warn("connect to mysql error:", err, "ds_id", ds.Id, "url", ds.URL) |
| 36 | + return models.PluginResult{ |
| 37 | + Status: models.PluginStatusError, |
| 38 | + Error: err.Error(), |
| 39 | + } |
| 40 | + } |
| 41 | + connsLock.Lock() |
| 42 | + conns[ds.Id] = conn |
| 43 | + connsLock.Unlock() |
| 44 | + } |
| 45 | + rows, err := conn.Query(query) |
| 46 | + if err != nil { |
| 47 | + colorlog.RootLogger.Info("Error query mysql :", "error", err, "ds_id", ds.Id, "query", query) |
| 48 | + return models.PluginResult{ |
| 49 | + Status: models.PluginStatusError, |
| 50 | + Error: err.Error(), |
| 51 | + } |
| 52 | + } |
| 53 | + defer rows.Close() |
| 54 | + |
| 55 | + columns, err := rows.Columns() |
| 56 | + if err != nil { |
| 57 | + colorlog.RootLogger.Info("Error get rows columns :", "error", err, "ds_id", ds.Id, "query", query) |
| 58 | + return models.PluginResult{ |
| 59 | + Status: models.PluginStatusError, |
| 60 | + Error: err.Error(), |
| 61 | + } |
| 62 | + } |
| 63 | + columnTypes, err := rows.ColumnTypes() |
| 64 | + if err != nil { |
| 65 | + colorlog.RootLogger.Info("Error get rows column types :", "error", err, "ds_id", ds.Id, "query", query) |
| 66 | + return models.PluginResult{ |
| 67 | + Status: models.PluginStatusError, |
| 68 | + Error: err.Error(), |
| 69 | + } |
| 70 | + } |
| 71 | + |
| 72 | + types := make(map[string]string) |
| 73 | + data := make([][]interface{}, 0) |
| 74 | + for rows.Next() { |
| 75 | + v := make([]interface{}, len(columns)) |
| 76 | + for i := 0; i < len(columns); i++ { |
| 77 | + t := columnTypes[i].ScanType() |
| 78 | + v[i] = reflect.New(t).Interface() |
| 79 | + |
| 80 | + tp := t.String() |
| 81 | + |
| 82 | + if tp == reflect.TypeOf(sql.NullTime{}).String() { |
| 83 | + types[columns[i]] = "time" |
| 84 | + } |
| 85 | + } |
| 86 | + |
| 87 | + err = rows.Scan(v...) |
| 88 | + if err != nil { |
| 89 | + colorlog.RootLogger.Info("Error scan mysql :", "error", err, "ds_id", ds.Id) |
| 90 | + continue |
| 91 | + } |
| 92 | + |
| 93 | + for i, v0 := range v { |
| 94 | + v1, ok := v0.(*sql.NullTime) |
| 95 | + if ok { |
| 96 | + v[i] = v1.Time.Unix() |
| 97 | + } else { |
| 98 | + v2, ok := v0.(driver.Valuer) |
| 99 | + if ok { |
| 100 | + v[i], _ = v2.Value() |
| 101 | + } |
| 102 | + } |
| 103 | + } |
| 104 | + data = append(data, v) |
| 105 | + } |
| 106 | + |
| 107 | + return models.PluginResult{ |
| 108 | + Status: models.PluginStatusSuccess, |
| 109 | + Error: "", |
| 110 | + Data: map[string]interface{}{ |
| 111 | + "columns": columns, |
| 112 | + "data": data, |
| 113 | + "types": types, |
| 114 | + }} |
| 115 | +} |
| 116 | + |
| 117 | +func (*MysqlPlugin) TestDatasource(c *gin.Context) models.PluginResult { |
| 118 | + return TestMysqlDatasource(c) |
| 119 | +} |
| 120 | + |
| 121 | +func init() { |
| 122 | + // register datasource |
| 123 | + models.RegisterPlugin(datasourceName, &MysqlPlugin{}) |
| 124 | +} |
| 125 | + |
| 126 | +func TestMysqlDatasource(c *gin.Context) models.PluginResult { |
| 127 | + url := c.Query("url") |
| 128 | + database := c.Query("database") |
| 129 | + username := c.Query("username") |
| 130 | + password := c.Query("password") |
| 131 | + db, err := connectToMysql(url, database, username, password) |
| 132 | + if err != nil { |
| 133 | + return models.GenPluginResult(models.PluginStatusError, err.Error(), nil) |
| 134 | + } |
| 135 | + if err = db.PingContext(context.Background()); err != nil { |
| 136 | + return models.GenPluginResult(models.PluginStatusError, err.Error(), nil) |
| 137 | + } |
| 138 | + return models.GenPluginResult(models.PluginStatusSuccess, "", nil) |
| 139 | +} |
| 140 | + |
| 141 | +func connectToMysql(url, database, username, password string) (*sql.DB, error) { |
| 142 | + dsn := fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8&parseTime=true", username, password, url, database) |
| 143 | + colorlog.RootLogger.Debug("connect to mysql dsn: ", dsn) |
| 144 | + |
| 145 | + db, err := sql.Open("mysql", dsn) |
| 146 | + if err != nil { |
| 147 | + return nil, err |
| 148 | + } |
| 149 | + db.SetMaxIdleConns(5) |
| 150 | + db.SetMaxOpenConns(5) |
| 151 | + db.SetConnMaxLifetime(time.Duration(10) * time.Minute) |
| 152 | + db.SetConnMaxIdleTime(time.Duration(10) * time.Minute) |
| 153 | + ctx, cancel := context.WithTimeout(context.Background(), time.Second*30) |
| 154 | + defer cancel() |
| 155 | + if err = db.PingContext(ctx); err != nil { |
| 156 | + return nil, err |
| 157 | + } |
| 158 | + return db, nil |
| 159 | +} |
0 commit comments