当前位置: 首页>>代码示例>>Golang>>正文


Golang baseDbConnection.DbConnection类代码示例

本文整理汇总了Golang中github.com/jrbudnack/cf-sql-broker/dbconnection/baseDbConnection.DbConnection的典型用法代码示例。如果您正苦于以下问题:Golang DbConnection类的具体用法?Golang DbConnection怎么用?Golang DbConnection使用的例子?那么恭喜您, 这里精选的类代码示例或许可以为您提供帮助。


在下文中一共展示了DbConnection类的8个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Golang代码示例。

示例1: UserExists

func (pgDbc PostgresDbConnection) UserExists(userName string, dbConn *baseDbConnection.DbConnection) (bool, error) {
	queryResults, err := dbConn.Query("SELECT 1 FROM pg_roles WHERE rolname = '" + userName + "'")

	if err != nil {
		return false, err
	}

	users := baseDbConnection.ParseResults(queryResults)
	for _ = range users {
		return true, nil
	}
	return false, nil
}
开发者ID:jrbudnack,项目名称:cf-sql-broker,代码行数:13,代码来源:postgresdbconnection.go

示例2: DbExists

func (pgDbc PostgresDbConnection) DbExists(databaseName string, dbConn *baseDbConnection.DbConnection) (bool, error) {
	queryResults, err := dbConn.Query("SELECT 1 FROM pg_database WHERE datname = '" + databaseName + "'")

	if err != nil {
		return false, err
	}

	databases := baseDbConnection.ParseResults(queryResults)

	for _ = range databases {
		return true, nil
	}
	return false, nil
}
开发者ID:jrbudnack,项目名称:cf-sql-broker,代码行数:14,代码来源:postgresdbconnection.go

示例3: GetTableDefinition

func (pgDbc PostgresDbConnection) GetTableDefinition(tableName string, dbConn *baseDbConnection.DbConnection) (baseDbConnection.TableDefinition, error) {
	// In PostgreSQL, we query the system tables in information_schema to get the list
	// of tables and their columns.
	sqlGetTable := `
SELECT c.table_name, c.column_name, c.ordinal_position,
c.data_type, CASE WHEN primary_key = 1 THEN TRUE ELSE FALSE END AS primary_key,
c.column_default
FROM information_schema.columns c LEFT OUTER JOIN
(SELECT kcu.table_name, kcu.column_name, 1 AS primary_key
FROM information_schema.key_column_usage kcu INNER JOIN
information_schema.table_constraints tc ON
tc.table_catalog = kcu.table_catalog AND
tc.table_schema = kcu.table_schema AND
tc.table_name = kcu.table_name
WHERE tc.constraint_type = 'PRIMARY KEY') pk
ON pk.table_name = c.table_name AND
pk.column_name = c.column_name
WHERE c.table_catalog = '` + dbConn.DBName + `' AND
c.table_schema = '` + dbConn.Schema + `' AND c.table_name = '` + tableName + `'
ORDER BY c.table_name, c.ordinal_position, c.column_name`

	tableDefinition := baseDbConnection.TableDefinition{}
	tableDefinition.Columns = make([]baseDbConnection.ColumnDefinition, 0)

	queryResults, err := dbConn.Query(sqlGetTable)
	tableColumns := baseDbConnection.ParseResults(queryResults)

	if err != nil {
		return baseDbConnection.TableDefinition{}, err
	}

	for _, tableColumn := range tableColumns {
		tableDefinition.Name = tableColumn["table_name"].(string)
		newColumn :=
			baseDbConnection.ColumnDefinition{
				tableColumn["column_name"].(string),
				tableColumn["data_type"].(string),
				false,
				tableColumn["column_default"].(string),
			}
		if tableColumn["primary_key"].(bool) {
			newColumn.PrimaryKey = true
		}
		tableDefinition.Columns = append(tableDefinition.Columns, newColumn)
	}

	return tableDefinition, nil
}
开发者ID:jrbudnack,项目名称:cf-sql-broker,代码行数:48,代码来源:postgresdbconnection.go

示例4: CreateTable

func (pgDbc PostgresDbConnection) CreateTable(table baseDbConnection.TableDefinition, dbConn *baseDbConnection.DbConnection) error {
	existingTable, err := dbConn.GetTableDefinition(table.Name)

	if err != nil {
		return err
	}

	if existingTable.Name != table.Name {
		sqlCreateTable := "CREATE TABLE IF NOT EXISTS " + dbConn.Schema + "." + table.Name + " (\n"

		fields := make([]string, 0)
		primaryKey := make([]string, 0)

		for _, tableColumn := range table.Columns {
			fieldDef := "\"" + tableColumn.Name + "\" " + tableColumn.Type
			if tableColumn.Default != "" {
				fieldDef += " DEFAULT " + tableColumn.Default
			}
			fields = append(fields, fieldDef)

			if tableColumn.PrimaryKey {
				primaryKey = append(primaryKey, tableColumn.Name)
			}
		}
		fields = append(fields, "CONSTRAINT pk_"+existingTable.Name+" PRIMARY KEY ("+strings.Join(primaryKey, ",")+")")
		sqlCreateTable = sqlCreateTable + strings.Join(fields, "\n,")
		sqlCreateTable = sqlCreateTable + ") WITH (OIDS = FALSE);\n"
		//TODO:  Default Table Grants
		//TODO:  Indexes
		//TODO:  Partitioning
		_, err := dbConn.Exec(sqlCreateTable)
		return err
	} else {
		return errors.New("Table already exists.")
	}
}
开发者ID:jrbudnack,项目名称:cf-sql-broker,代码行数:36,代码来源:postgresdbconnection.go

示例5: CreateUser

func (pgDbc PostgresDbConnection) CreateUser(userName string, password string, dbConn *baseDbConnection.DbConnection) error {
	_, err := dbConn.Exec("CREATE USER " + userName + " WITH PASSWORD '" + password + "'")

	if err == nil {
		_, err = dbConn.Exec("GRANT ALL PRIVILEGES ON DATABASE " + dbConn.DBName + " TO " + userName)
	}
	if err == nil {
		_, err = dbConn.Exec("GRANT ALL PRIVILEGES ON SCHEMA public TO " + userName)
	}

	return err
}
开发者ID:jrbudnack,项目名称:cf-sql-broker,代码行数:12,代码来源:postgresdbconnection.go

示例6: DropUser

func (pgDbc PostgresDbConnection) DropUser(userName string, dbConn *baseDbConnection.DbConnection) error {
	_, err := dbConn.Exec("REVOKE ALL PRIVILEGES ON SCHEMA public FROM " + userName)

	if err == nil {
		_, err = dbConn.Exec("REVOKE ALL PRIVILEGES ON DATABASE " + dbConn.DBName + " FROM " + userName)
	}
	if err == nil {
		//RISK ITEM:  If the user owns any objects, this will fail.
		//HOW DO WE WANT TO HANDLE THIS?
		_, err = dbConn.Exec("DROP USER IF EXISTS " + userName)
	}

	return err
}
开发者ID:jrbudnack,项目名称:cf-sql-broker,代码行数:14,代码来源:postgresdbconnection.go

示例7: DropDatabase

func (pgDbc PostgresDbConnection) DropDatabase(databaseName string, dbConn *baseDbConnection.DbConnection) error {
	_, err := dbConn.Exec("DROP DATABASE " + databaseName)
	return err
}
开发者ID:jrbudnack,项目名称:cf-sql-broker,代码行数:4,代码来源:postgresdbconnection.go

示例8: CreateDatabase

func (pgDbc PostgresDbConnection) CreateDatabase(databaseName string, dbConn *baseDbConnection.DbConnection) error {
	_, err := dbConn.Exec("CREATE DATABASE " + databaseName)
	return err
}
开发者ID:jrbudnack,项目名称:cf-sql-broker,代码行数:4,代码来源:postgresdbconnection.go


注:本文中的github.com/jrbudnack/cf-sql-broker/dbconnection/baseDbConnection.DbConnection类示例由纯净天空整理自Github/MSDocs等开源代码及文档管理平台,相关代码片段筛选自各路编程大神贡献的开源项目,源码版权归原作者所有,传播和使用请参考对应项目的License;未经允许,请勿转载。