package service
import (
"bufio"
"compress/gzip"
"context"
"encoding/json"
"fmt"
"io"
"io/ioutil"
"mime/multipart"
"os"
"os/exec"
"path/filepath"
"regexp"
"strconv"
"strings"
"time"
"github.com/1Panel-dev/1Panel/backend/app/dto"
"github.com/1Panel-dev/1Panel/backend/app/model"
"github.com/1Panel-dev/1Panel/backend/constant"
"github.com/1Panel-dev/1Panel/backend/global"
"github.com/1Panel-dev/1Panel/backend/utils/compose"
"github.com/1Panel-dev/1Panel/backend/utils/files"
_ "github.com/go-sql-driver/mysql"
"github.com/jinzhu/copier"
"github.com/pkg/errors"
)
type MysqlService struct { }
type IMysqlService interface {
SearchWithPage ( search dto . SearchDBWithPage ) ( int64 , interface { } , error )
ListDBByVersion ( name string ) ( [ ] string , error )
SearchBackupsWithPage ( search dto . SearchBackupsWithPage ) ( int64 , interface { } , error )
Create ( mysqlDto dto . MysqlDBCreate ) error
ChangeInfo ( info dto . ChangeDBInfo ) error
UpdateVariables ( mysqlName string , updatas [ ] dto . MysqlVariablesUpdate ) error
UpdateConfByFile ( info dto . MysqlConfUpdateByFile ) error
UpFile ( mysqlName string , files [ ] * multipart . FileHeader ) error
RecoverByUpload ( req dto . UploadRecover ) error
SearchUpListWithPage ( req dto . SearchDBWithPage ) ( int64 , interface { } , error )
Backup ( db dto . BackupDB ) error
Recover ( db dto . RecoverDB ) error
Delete ( name string , ids [ ] uint ) error
LoadStatus ( name string ) ( * dto . MysqlStatus , error )
LoadVariables ( vernamesion string ) ( * dto . MysqlVariables , error )
LoadRunningVersion ( ) ( [ ] string , error )
LoadBaseInfo ( name string ) ( * dto . DBBaseInfo , error )
}
func NewIMysqlService ( ) IMysqlService {
return & MysqlService { }
}
func ( u * MysqlService ) SearchWithPage ( search dto . SearchDBWithPage ) ( int64 , interface { } , error ) {
total , mysqls , err := mysqlRepo . Page ( search . Page , search . PageSize , mysqlRepo . WithByMysqlName ( search . MysqlName ) )
var dtoMysqls [ ] dto . MysqlDBInfo
for _ , mysql := range mysqls {
var item dto . MysqlDBInfo
if err := copier . Copy ( & item , & mysql ) ; err != nil {
return 0 , nil , errors . WithMessage ( constant . ErrStructTransform , err . Error ( ) )
}
dtoMysqls = append ( dtoMysqls , item )
}
return total , dtoMysqls , err
}
func ( u * MysqlService ) SearchUpListWithPage ( req dto . SearchDBWithPage ) ( int64 , interface { } , error ) {
var (
list [ ] dto . DatabaseFileRecords
backDatas [ ] dto . DatabaseFileRecords
)
localDir , appKey , err := loadBackupDirAndKey ( req . MysqlName )
if err != nil {
return 0 , list , nil
}
uploadDir := fmt . Sprintf ( "%s/database/%s/%s/upload" , localDir , appKey , req . MysqlName )
if _ , err := os . Stat ( uploadDir ) ; err != nil {
return 0 , list , nil
}
_ = filepath . Walk ( uploadDir , func ( path string , info os . FileInfo , err error ) error {
if err != nil {
return nil
}
if ! info . IsDir ( ) {
list = append ( list , dto . DatabaseFileRecords {
CreatedAt : info . ModTime ( ) . Format ( "2006-01-02 15:04:05" ) ,
Size : int ( info . Size ( ) ) ,
FileDir : uploadDir ,
FileName : info . Name ( ) ,
} )
}
return nil
} )
total , start , end := len ( list ) , ( req . Page - 1 ) * req . PageSize , req . Page * req . PageSize
if start > total {
backDatas = make ( [ ] dto . DatabaseFileRecords , 0 )
} else {
if end >= total {
end = total
}
backDatas = list [ start : end ]
}
return int64 ( total ) , backDatas , nil
}
func ( u * MysqlService ) UpFile ( mysqlName string , files [ ] * multipart . FileHeader ) error {
localDir , appKey , err := loadBackupDirAndKey ( mysqlName )
if err != nil {
return err
}
dstDir := fmt . Sprintf ( "%s/database/%s/%s/upload" , localDir , appKey , mysqlName )
if _ , err := os . Stat ( dstDir ) ; err != nil && os . IsNotExist ( err ) {
if err = os . MkdirAll ( dstDir , os . ModePerm ) ; err != nil {
if err != nil {
return fmt . Errorf ( "mkdir %s failed, err: %v" , dstDir , err )
}
}
}
for _ , file := range files {
src , err := file . Open ( )
if err != nil {
return err
}
defer src . Close ( )
out , err := os . Create ( dstDir + "/" + file . Filename )
if err != nil {
return err
}
defer out . Close ( )
_ , _ = io . Copy ( out , src )
}
return nil
}
func ( u * MysqlService ) RecoverByUpload ( req dto . UploadRecover ) error {
app , err := mysqlRepo . LoadBaseInfoByName ( req . MysqlName )
if err != nil {
return err
}
localDir , err := loadLocalDir ( )
if err != nil {
return err
}
file := req . FileDir + "/" + req . FileName
if ! strings . HasSuffix ( req . FileName , ".sql" ) && ! strings . HasSuffix ( req . FileName , ".gz" ) {
fileOp := files . NewFileOp ( )
fileNameItem := time . Now ( ) . Format ( "20060102150405" )
dstDir := fmt . Sprintf ( "%s/database/%s/%s/upload/tmp/%s" , localDir , app . Key , req . MysqlName , fileNameItem )
if _ , err := os . Stat ( dstDir ) ; err != nil && os . IsNotExist ( err ) {
if err = os . MkdirAll ( dstDir , os . ModePerm ) ; err != nil {
if err != nil {
return fmt . Errorf ( "mkdir %s failed, err: %v" , dstDir , err )
}
}
}
var compressType files . CompressType
switch {
case strings . HasSuffix ( req . FileName , ".tar.gz" ) , strings . HasSuffix ( req . FileName , ".tgz" ) :
compressType = files . TarGz
case strings . HasSuffix ( req . FileName , ".zip" ) :
compressType = files . Zip
}
if err := fileOp . Decompress ( req . FileDir + "/" + req . FileName , dstDir , compressType ) ; err != nil {
_ = os . RemoveAll ( dstDir )
return err
}
hasTestSql := false
_ = filepath . Walk ( dstDir , func ( path string , info os . FileInfo , err error ) error {
if err != nil {
return nil
}
if ! info . IsDir ( ) && info . Name ( ) == "test.sql" {
hasTestSql = true
file = path
}
return nil
} )
if ! hasTestSql {
_ = os . RemoveAll ( dstDir )
return fmt . Errorf ( "no such file named test.sql in %s, err: %v" , req . FileName , err )
}
defer func ( ) {
_ = os . RemoveAll ( dstDir )
} ( )
}
fi , _ := os . Open ( file )
defer fi . Close ( )
cmd := exec . Command ( "docker" , "exec" , "-i" , app . ContainerName , "mysql" , "-uroot" , "-p" + app . Password , req . DBName )
if strings . HasSuffix ( req . FileName , ".gz" ) {
gzipFile , err := os . Open ( file )
if err != nil {
return err
}
defer gzipFile . Close ( )
gzipReader , err := gzip . NewReader ( gzipFile )
if err != nil {
return err
}
defer gzipReader . Close ( )
cmd . Stdin = gzipReader
} else {
cmd . Stdin = fi
}
stdout , err := cmd . CombinedOutput ( )
stdStr := strings . ReplaceAll ( string ( stdout ) , "mysql: [Warning] Using a password on the command line interface can be insecure.\n" , "" )
if err != nil || strings . HasPrefix ( string ( stdStr ) , "ERROR " ) {
return errors . New ( stdStr )
}
return nil
}
func ( u * MysqlService ) ListDBByVersion ( name string ) ( [ ] string , error ) {
mysqls , err := mysqlRepo . List ( mysqlRepo . WithByMysqlName ( name ) )
var dbNames [ ] string
for _ , mysql := range mysqls {
dbNames = append ( dbNames , mysql . Name )
}
return dbNames , err
}
func ( u * MysqlService ) SearchBackupsWithPage ( search dto . SearchBackupsWithPage ) ( int64 , interface { } , error ) {
app , err := mysqlRepo . LoadBaseInfoByName ( search . MysqlName )
if err != nil {
return 0 , nil , err
}
searchDto := dto . BackupSearch {
Type : "database-mysql" ,
PageInfo : search . PageInfo ,
Name : app . Name ,
DetailName : search . DBName ,
}
return NewIBackupService ( ) . SearchRecordWithPage ( searchDto )
}
func ( u * MysqlService ) LoadRunningVersion ( ) ( [ ] string , error ) {
return mysqlRepo . LoadRunningVersion ( [ ] string { "Mysql5.7" , "Mysql8.0" } )
}
func ( u * MysqlService ) Create ( mysqlDto dto . MysqlDBCreate ) error {
if mysqlDto . Username == "root" {
return errors . New ( "Cannot set root as user name" )
}
app , err := mysqlRepo . LoadBaseInfoByName ( mysqlDto . MysqlName )
if err != nil {
return err
}
mysql , _ := mysqlRepo . Get ( commonRepo . WithByName ( mysqlDto . Name ) , mysqlRepo . WithByMysqlName ( app . Key ) )
if mysql . ID != 0 {
return constant . ErrRecordExist
}
if err := copier . Copy ( & mysql , & mysqlDto ) ; err != nil {
return errors . WithMessage ( constant . ErrStructTransform , err . Error ( ) )
}
if err := excuteSql ( app . ContainerName , app . Password , fmt . Sprintf ( "create database if not exists %s character set=%s" , mysqlDto . Name , mysqlDto . Format ) ) ; err != nil {
return err
}
tmpPermission := mysqlDto . Permission
if err := excuteSql ( app . ContainerName , app . Password , fmt . Sprintf ( "create user if not exists '%s'@'%s' identified by '%s';" , mysqlDto . Username , tmpPermission , mysqlDto . Password ) ) ; err != nil {
return err
}
grantStr := fmt . Sprintf ( "grant all privileges on %s.* to '%s'@'%s'" , mysqlDto . Name , mysqlDto . Username , tmpPermission )
if app . Key == "mysql5.7" {
grantStr = fmt . Sprintf ( "%s identified by '%s' with grant option;" , grantStr , mysqlDto . Password )
}
if err := excuteSql ( app . ContainerName , app . Password , grantStr ) ; err != nil {
return err
}
if err := mysqlRepo . Create ( context . TODO ( ) , & mysql ) ; err != nil {
return err
}
return nil
}
func ( u * MysqlService ) Backup ( db dto . BackupDB ) error {
localDir , appKey , err := loadBackupDirAndKey ( db . MysqlName )
if err != nil {
return err
}
backupDir := fmt . Sprintf ( "database/%s/%s/%s" , appKey , db . MysqlName , db . DBName )
fileName := fmt . Sprintf ( "%s_%s.sql.gz" , db . DBName , time . Now ( ) . Format ( "20060102150405" ) )
if err := backupMysql ( "LOCAL" , localDir , backupDir , db . MysqlName , db . DBName , fileName ) ; err != nil {
return err
}
return nil
}
func ( u * MysqlService ) Recover ( db dto . RecoverDB ) error {
app , err := mysqlRepo . LoadBaseInfoByName ( db . MysqlName )
if err != nil {
return err
}
gzipFile , err := os . Open ( db . BackupName )
if err != nil {
return err
}
defer gzipFile . Close ( )
gzipReader , err := gzip . NewReader ( gzipFile )
if err != nil {
return err
}
defer gzipReader . Close ( )
cmd := exec . Command ( "docker" , "exec" , "-i" , app . ContainerName , "mysql" , "-uroot" , "-p" + app . Password , db . DBName )
cmd . Stdin = gzipReader
stdout , err := cmd . CombinedOutput ( )
stdStr := strings . ReplaceAll ( string ( stdout ) , "mysql: [Warning] Using a password on the command line interface can be insecure.\n" , "" )
if err != nil || strings . HasPrefix ( string ( stdStr ) , "ERROR " ) {
return errors . New ( stdStr )
}
return nil
}
func ( u * MysqlService ) Delete ( name string , ids [ ] uint ) error {
app , err := mysqlRepo . LoadBaseInfoByName ( name )
if err != nil {
return err
}
dbs , err := mysqlRepo . List ( commonRepo . WithIdsIn ( ids ) )
if err != nil {
return err
}
for _ , db := range dbs {
if len ( db . Name ) != 0 {
if err := excuteSql ( app . ContainerName , app . Password , fmt . Sprintf ( "drop user if exists '%s'@'%s'" , db . Name , db . Permission ) ) ; err != nil {
return err
}
if err := excuteSql ( app . ContainerName , app . Password , fmt . Sprintf ( "drop database if exists %s" , db . Name ) ) ; err != nil {
return err
}
}
_ = mysqlRepo . Delete ( context . Background ( ) , commonRepo . WithByID ( db . ID ) )
}
return nil
}
func ( u * MysqlService ) ChangeInfo ( info dto . ChangeDBInfo ) error {
var (
mysql model . DatabaseMysql
err error
)
if info . ID != 0 {
mysql , err = mysqlRepo . Get ( commonRepo . WithByID ( info . ID ) )
if err != nil {
return err
}
}
app , err := mysqlRepo . LoadBaseInfoByName ( info . MysqlName )
if err != nil {
return err
}
if info . Operation == "password" {
if info . ID != 0 {
if err := excuteSql ( app . ContainerName , app . Password , fmt . Sprintf ( "set password for %s@%s = password('%s')" , mysql . Username , mysql . Permission , info . Value ) ) ; err != nil {
return err
}
_ = mysqlRepo . Update ( mysql . ID , map [ string ] interface { } { "password" : info . Value } )
return nil
}
hosts , err := excuteSqlForRows ( app . ContainerName , app . Password , "select host from mysql.user where user='root';" )
if err != nil {
return err
}
for _ , host := range hosts {
if host == "%" || host == "localhost" {
if err := excuteSql ( app . ContainerName , app . Password , fmt . Sprintf ( "set password for root@'%s' = password('%s')" , host , info . Value ) ) ; err != nil {
return err
}
}
}
_ = mysqlRepo . UpdateDatabaseInfo ( app . ID , map [ string ] interface { } {
"param" : strings . ReplaceAll ( app . Param , app . Password , info . Value ) ,
"env" : strings . ReplaceAll ( app . Env , app . Password , info . Value ) ,
} )
return nil
}
if info . ID == 0 {
mysql . Name = "*"
mysql . Username = "root"
mysql . Permission = "%"
mysql . Password = app . Password
}
if info . Value != mysql . Permission {
if err := excuteSql ( app . ContainerName , app . Password , fmt . Sprintf ( "drop user if exists '%s'@'%s'" , mysql . Username , mysql . Permission ) ) ; err != nil {
return err
}
if info . ID == 0 {
return nil
}
}
if err := excuteSql ( app . ContainerName , app . Password , fmt . Sprintf ( "create user if not exists '%s'@'%s' identified by '%s';" , mysql . Username , info . Value , mysql . Password ) ) ; err != nil {
return err
}
grantStr := fmt . Sprintf ( "grant all privileges on %s.* to '%s'@'%s'" , mysql . Name , mysql . Username , info . Value )
if app . Key == "mysql5.7" {
grantStr = fmt . Sprintf ( "%s identified by '%s' with grant option;" , grantStr , mysql . Password )
}
if err := excuteSql ( app . ContainerName , app . Password , grantStr ) ; err != nil {
return err
}
if err := excuteSql ( app . ContainerName , app . Password , "flush privileges" ) ; err != nil {
return err
}
if info . ID == 0 {
return nil
}
_ = mysqlRepo . Update ( mysql . ID , map [ string ] interface { } { "permission" : info . Value } )
return nil
}
func ( u * MysqlService ) UpdateConfByFile ( info dto . MysqlConfUpdateByFile ) error {
app , err := mysqlRepo . LoadBaseInfoByName ( info . MysqlName )
if err != nil {
return err
}
path := fmt . Sprintf ( "%s/%s/%s/conf/my.cnf" , constant . AppInstallDir , app . Key , app . Name )
file , err := os . OpenFile ( path , os . O_WRONLY | os . O_TRUNC , 0640 )
if err != nil {
return err
}
defer file . Close ( )
write := bufio . NewWriter ( file )
_ , _ = write . WriteString ( info . File )
write . Flush ( )
if _ , err := compose . Restart ( fmt . Sprintf ( "%s/%s/%s/docker-compose.yml" , constant . AppInstallDir , app . Key , app . Name ) ) ; err != nil {
return err
}
return nil
}
func ( u * MysqlService ) UpdateVariables ( mysqlName string , updatas [ ] dto . MysqlVariablesUpdate ) error {
app , err := mysqlRepo . LoadBaseInfoByName ( mysqlName )
if err != nil {
return err
}
var files [ ] string
path := fmt . Sprintf ( "%s/%s/%s/conf/my.cnf" , constant . AppInstallDir , app . Key , app . Name )
lineBytes , err := ioutil . ReadFile ( path )
if err != nil {
return err
} else {
files = strings . Split ( string ( lineBytes ) , "\n" )
}
group := ""
for _ , info := range updatas {
switch info . Param {
case "key_buffer_size" , "sort_buffer_size" :
group = "[myisamchk]"
default :
group = "[mysqld]"
}
files = updateMyCnf ( files , group , info . Param , info . Value )
}
file , err := os . OpenFile ( path , os . O_WRONLY , 0666 )
if err != nil {
return err
}
defer file . Close ( )
_ , err = file . WriteString ( strings . Join ( files , "\n" ) )
if err != nil {
return err
}
if _ , err := compose . Restart ( fmt . Sprintf ( "%s/%s/%s/docker-compose.yml" , constant . AppInstallDir , app . Key , app . Name ) ) ; err != nil {
return err
}
return nil
}
func ( u * MysqlService ) LoadBaseInfo ( name string ) ( * dto . DBBaseInfo , error ) {
var data dto . DBBaseInfo
app , err := mysqlRepo . LoadBaseInfoByName ( name )
if err != nil {
return nil , err
}
data . ContainerName = app . ContainerName
data . Name = app . Name
data . Port = int64 ( app . Port )
data . Password = app . Password
data . MysqlKey = app . Key
hosts , err := excuteSqlForRows ( app . ContainerName , app . Password , "select host from mysql.user where user='root';" )
if err != nil {
return nil , err
}
for _ , host := range hosts {
if host == "%" {
data . RemoteConn = true
break
}
}
return & data , nil
}
func ( u * MysqlService ) LoadVariables ( name string ) ( * dto . MysqlVariables , error ) {
app , err := mysqlRepo . LoadBaseInfoByName ( name )
if err != nil {
return nil , err
}
variableMap , err := excuteSqlForMaps ( app . ContainerName , app . Password , "show global variables;" )
if err != nil {
return nil , err
}
var info dto . MysqlVariables
arr , err := json . Marshal ( variableMap )
if err != nil {
return nil , err
}
_ = json . Unmarshal ( arr , & info )
return & info , nil
}
func ( u * MysqlService ) LoadStatus ( name string ) ( * dto . MysqlStatus , error ) {
app , err := mysqlRepo . LoadBaseInfoByName ( name )
if err != nil {
return nil , err
}
statusMap , err := excuteSqlForMaps ( app . ContainerName , app . Password , "show global status;" )
if err != nil {
return nil , err
}
var info dto . MysqlStatus
arr , err := json . Marshal ( statusMap )
if err != nil {
return nil , err
}
_ = json . Unmarshal ( arr , & info )
if value , ok := statusMap [ "Run" ] ; ok {
uptime , _ := strconv . Atoi ( value )
info . Run = time . Unix ( time . Now ( ) . Unix ( ) - int64 ( uptime ) , 0 ) . Format ( "2006-01-02 15:04:05" )
} else {
if value , ok := statusMap [ "Uptime" ] ; ok {
uptime , _ := strconv . Atoi ( value )
info . Run = time . Unix ( time . Now ( ) . Unix ( ) - int64 ( uptime ) , 0 ) . Format ( "2006-01-02 15:04:05" )
}
}
info . File = "OFF"
info . Position = "OFF"
rows , err := excuteSqlForRows ( app . ContainerName , app . Password , "show master status;" )
if err != nil {
return nil , err
}
if len ( rows ) > 2 {
itemValue := strings . Split ( rows [ 1 ] , "\t" )
if len ( itemValue ) > 2 {
info . File = itemValue [ 0 ]
info . Position = itemValue [ 1 ]
}
}
return & info , nil
}
func excuteSqlForMaps ( containerName , password , command string ) ( map [ string ] string , error ) {
cmd := exec . Command ( "docker" , "exec" , containerName , "mysql" , "-uroot" , "-p" + password , "-e" , command )
stdout , err := cmd . CombinedOutput ( )
stdStr := strings . ReplaceAll ( string ( stdout ) , "mysql: [Warning] Using a password on the command line interface can be insecure.\n" , "" )
if err != nil || strings . HasPrefix ( string ( stdStr ) , "ERROR " ) {
return nil , errors . New ( stdStr )
}
rows := strings . Split ( stdStr , "\n" )
rowMap := make ( map [ string ] string )
for _ , v := range rows {
itemRow := strings . Split ( v , "\t" )
if len ( itemRow ) == 2 {
rowMap [ itemRow [ 0 ] ] = itemRow [ 1 ]
}
}
return rowMap , nil
}
func excuteSqlForRows ( containerName , password , command string ) ( [ ] string , error ) {
cmd := exec . Command ( "docker" , "exec" , containerName , "mysql" , "-uroot" , "-p" + password , "-e" , command )
stdout , err := cmd . CombinedOutput ( )
stdStr := strings . ReplaceAll ( string ( stdout ) , "mysql: [Warning] Using a password on the command line interface can be insecure.\n" , "" )
if err != nil || strings . HasPrefix ( string ( stdStr ) , "ERROR " ) {
return nil , errors . New ( stdStr )
}
return strings . Split ( stdStr , "\n" ) , nil
}
func excuteSql ( containerName , password , command string ) error {
cmd := exec . Command ( "docker" , "exec" , containerName , "mysql" , "-uroot" , "-p" + password , "-e" , command )
stdout , err := cmd . CombinedOutput ( )
stdStr := strings . ReplaceAll ( string ( stdout ) , "mysql: [Warning] Using a password on the command line interface can be insecure.\n" , "" )
if err != nil || strings . HasPrefix ( string ( stdStr ) , "ERROR " ) {
return errors . New ( stdStr )
}
return nil
}
func backupMysql ( backupType , baseDir , backupDir , mysqlName , dbName , fileName string ) error {
app , err := mysqlRepo . LoadBaseInfoByName ( mysqlName )
if err != nil {
return err
}
fullDir := baseDir + "/" + backupDir
if _ , err := os . Stat ( fullDir ) ; err != nil && os . IsNotExist ( err ) {
if err = os . MkdirAll ( fullDir , os . ModePerm ) ; err != nil {
if err != nil {
return fmt . Errorf ( "mkdir %s failed, err: %v" , fullDir , err )
}
}
}
outfile , _ := os . OpenFile ( fullDir + "/" + fileName , os . O_RDWR | os . O_CREATE , 0755 )
cmd := exec . Command ( "docker" , "exec" , app . ContainerName , "mysqldump" , "-uroot" , "-p" + app . Password , dbName )
gzipCmd := exec . Command ( "gzip" , "-cf" )
gzipCmd . Stdin , _ = cmd . StdoutPipe ( )
gzipCmd . Stdout = outfile
_ = gzipCmd . Start ( )
_ = cmd . Run ( )
_ = gzipCmd . Wait ( )
record := & model . BackupRecord {
Type : "database-mysql" ,
Name : app . Name ,
DetailName : dbName ,
Source : backupType ,
BackupType : backupType ,
FileDir : backupDir ,
FileName : fileName ,
}
if baseDir != constant . TmpDir || backupType == "LOCAL" {
record . Source = "LOCAL"
record . FileDir = fullDir
}
if err := backupRepo . CreateRecord ( record ) ; err != nil {
global . LOG . Errorf ( "save backup record failed, err: %v" , err )
}
return nil
}
func updateMyCnf ( oldFiles [ ] string , group string , param string , value interface { } ) [ ] string {
isOn := false
hasGroup := false
hasKey := false
regItem , _ := regexp . Compile ( ` \[*\] ` )
var newFiles [ ] string
i := 0
for _ , line := range oldFiles {
i ++
if strings . HasPrefix ( line , group ) {
isOn = true
hasGroup = true
newFiles = append ( newFiles , line )
continue
}
if ! isOn {
newFiles = append ( newFiles , line )
continue
}
if strings . HasPrefix ( line , param ) || strings . HasPrefix ( line , "# " + param ) {
newFiles = append ( newFiles , fmt . Sprintf ( "%s=%v" , param , value ) )
hasKey = true
continue
}
if regItem . Match ( [ ] byte ( line ) ) || i == len ( oldFiles ) {
isOn = false
if ! hasKey {
newFiles = append ( newFiles , fmt . Sprintf ( "%s=%v" , param , value ) )
}
newFiles = append ( newFiles , line )
continue
}
newFiles = append ( newFiles , line )
}
if ! hasGroup {
newFiles = append ( newFiles , group + "\n" )
newFiles = append ( newFiles , fmt . Sprintf ( "%s=%v\n" , param , value ) )
}
return newFiles
}
func loadBackupDirAndKey ( mysqlName string ) ( string , string , error ) {
app , err := mysqlRepo . LoadBaseInfoByName ( mysqlName )
if err != nil {
return "" , "" , err
}
localDir , err := loadLocalDir ( )
if err != nil {
return "" , "" , err
}
return localDir , app . Key , nil
}