liubin/ContextOS
1
0
Fork 0
forked from x/ContextOS
Code Pull Requests Activity

Merge branch 'master' of github.com:shylinux/context

Browse Source
This commit is contained in:
shylinux 2018-04-20 09:03:47 +08:00
commit e463fe87cd
5 changed files with 279 additions and 58 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
Makefile Normal file
View File

@ -0,0 +1,8 @@
BENCH=src/example/bench.go
install:
go install $(BENCH)
build:
go build $(BENCH)

5
etc/init.shy
View File

@ -1,2 +1,5 @@
~aaa login root root
source etc/local.shy
~web serve ./ ":9090"
# source etc/local.shy

2
src/contexts/cli/cli.go
View File

@ -102,7 +102,7 @@ func (cli *CLI) Begin(m *ctx.Message, arg ...string) ctx.Server { // {{{
yac.Cmd("train", "num", "num", "mul{", "[1-9][0-9]*", "0[0-9]+", "0x[0-9]+", "}")
yac.Cmd("train", "str", "str", "mul{", "\"[^\"]*\"", "'[^']*'", "}")
yac.Cmd("train", "tran", "tran", "mul{", "@", "$", "}", "opt{", "[a-zA-Z0-9]+", "}")
yac.Cmd("train", "tran", "tran", "mul{", "@", "$", "}", "opt{", "$", "[a-zA-Z0-9]+", "}")
yac.Cmd("train", "word", "word", "mul{", "~", "!", "tran", "str", "[a-zA-Z0-9_/.]+", "}")
yac.Cmd("train", "op1", "op1", "mul{", "$", "@", "}")

123
src/contexts/mdb/mdb.go
View File

@ -1,7 +1,9 @@
package mdb
import (
package mdb // {{{
// }}}
import ( // {{{
"contexts"
"strconv"
"strings"
"database/sql"
_ "github.com/go-sql-driver/mysql"
@ -9,12 +11,19 @@ import (
"fmt"
)
// }}}
type MDB struct {
*sql.DB
list map[string][]string
list_key []string
table []string
*ctx.Context
}
func (mdb *MDB) Spawn(m *ctx.Message, c *ctx.Context, arg ...string) ctx.Server {
func (mdb *MDB) Spawn(m *ctx.Message, c *ctx.Context, arg ...string) ctx.Server { // {{{
c.Caches = map[string]*ctx.Cache{
"source": &ctx.Cache{Name: "数据库参数", Value: "", Help: "数据库参数"},
"driver": &ctx.Cache{Name: "数据库驱动", Value: "", Help: "数据库驱动"},
@ -26,14 +35,17 @@ func (mdb *MDB) Spawn(m *ctx.Message, c *ctx.Context, arg ...string) ctx.Server
return s
}
func (mdb *MDB) Begin(m *ctx.Message, arg ...string) ctx.Server {
// }}}
func (mdb *MDB) Begin(m *ctx.Message, arg ...string) ctx.Server { // {{{
mdb.Context.Master(nil)
if mdb.Context == Index {
Pulse = m
}
return mdb
}
func (mdb *MDB) Start(m *ctx.Message, arg ...string) bool {
// }}}
func (mdb *MDB) Start(m *ctx.Message, arg ...string) bool { // {{{
if len(arg) > 0 {
m.Cap("source", arg[0])
}
@ -54,7 +66,8 @@ func (mdb *MDB) Start(m *ctx.Message, arg ...string) bool {
return false
}
func (mdb *MDB) Close(m *ctx.Message, arg ...string) bool {
// }}}
func (mdb *MDB) Close(m *ctx.Message, arg ...string) bool { // {{{
switch mdb.Context {
case m.Target():
if mdb.DB != nil {
@ -67,9 +80,12 @@ func (mdb *MDB) Close(m *ctx.Message, arg ...string) bool {
return true
}
// }}}
var Pulse *ctx.Message
var Index = &ctx.Context{Name: "mdb", Help: "数据中心",
Caches: map[string]*ctx.Cache{
"count": &ctx.Cache{Name: "count", Value: "0", Help: "主机协议"},
"nsource": &ctx.Cache{Name: "数据源数量", Value: "0", Help: "已打开数据库的数量"},
},
Configs: map[string]*ctx.Config{
@ -77,15 +93,16 @@ var Index = &ctx.Context{Name: "mdb", Help: "数据中心",
},
Commands: map[string]*ctx.Command{
"open": &ctx.Command{Name: "open source [driver]", Help: "打开数据库", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) {
m.Assert(len(arg) > 0, "缺少参数")
m.Assert(len(arg) > 0, "缺少参数") // {{{
m.Start(fmt.Sprintf("db%d", Pulse.Capi("nsource", 1)), "数据存储", arg...)
Pulse.Cap("stream", Pulse.Cap("nsource"))
m.Echo(m.Target().Name)
// }}}
}},
"exec": &ctx.Command{Name: "exec sql [arg]", Help: "操作数据库",
Appends: map[string]string{"last": "最后插入元组的标识", "nrow": "修改元组的数量"},
Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) {
mdb, ok := m.Target().Server.(*MDB)
mdb, ok := m.Target().Server.(*MDB) // {{{
m.Assert(ok, "目标模块类型错误")
m.Assert(len(arg) > 0, "缺少参数")
m.Assert(mdb.DB != nil, "数据库未打开")
@ -106,9 +123,10 @@ var Index = &ctx.Context{Name: "mdb", Help: "数据中心",
m.Add("append", "last", fmt.Sprintf("%d", id))
m.Add("append", "nrow", fmt.Sprintf("%d", n))
m.Log("info", nil, "last(%d) nrow(%d)", id, n)
// }}}
}},
"query": &ctx.Command{Name: "query sql [arg]", Help: "执行查询语句", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) {
mdb, ok := m.Target().Server.(*MDB)
mdb, ok := m.Target().Server.(*MDB) // {{{
m.Assert(ok, "目标模块类型错误")
m.Assert(len(arg) > 0, "缺少参数")
m.Assert(mdb.DB != nil, "数据库未打开")
@ -151,6 +169,91 @@ var Index = &ctx.Context{Name: "mdb", Help: "数据中心",
} else {
m.Log("info", nil, "rows(0) cols(0)")
}
// }}}
}},
"table": &ctx.Command{Name: "table", Help: "执行查询语句", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) {
mdb, ok := m.Target().Server.(*MDB) // {{{
m.Assert(ok)
msg := m.Spawn(m.Target())
if len(arg) > 0 {
table := arg[0]
index, e := strconv.Atoi(arg[0])
if e == nil && index < len(mdb.table) {
table = mdb.table[index]
}
msg.Cmd("query", fmt.Sprintf("desc %s", table))
if len(arg) > 1 {
for i, v := range msg.Meta[msg.Meta["append"][0]] {
if v == arg[1] {
for _, k := range msg.Meta["append"] {
m.Echo("%s: %s\n", k, msg.Meta[k][i])
}
}
}
} else {
for _, v := range msg.Meta[msg.Meta["append"][0]] {
m.Echo("%s\n", v)
}
}
} else {
msg.Cmd("query", "show tables")
mdb.table = []string{}
for i, v := range msg.Meta[msg.Meta["append"][0]] {
mdb.table = append(mdb.table, v)
m.Echo("%d: %s\n", i, v)
}
}
// }}}
}},
"list": &ctx.Command{Name: "list add table field", Help: "执行查询语句", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) {
mdb, ok := m.Target().Server.(*MDB) // {{{
m.Assert(ok)
if len(arg) == 0 {
for _, k := range mdb.list_key {
m.Echo("%s: %v\n", k, mdb.list[k])
}
return
}
switch arg[0] {
case "add":
if mdb.list == nil {
mdb.list = make(map[string][]string)
}
mdb.list[m.Cap("count")] = arg[1:]
mdb.list_key = append(mdb.list_key, m.Cap("count"))
m.Capi("count", 1)
case "set":
mdb.list[arg[1]] = arg[2:]
default:
if table, ok := mdb.list[arg[0]]; ok {
msg := m.Spawn(m.Target())
fields := strings.Join(table[1:], ",")
condition := ""
if len(arg) > 1 && len(arg[1]) > 0 {
condition = fmt.Sprintf("where %s", arg[1])
}
other := ""
if len(arg) > 2 {
other = strings.Join(arg[2:], " ")
}
msg.Cmd("query", fmt.Sprintf("select %s from %s %s %s", fields, table[0], condition, other))
for _, k := range msg.Meta["append"] {
m.Echo("\t%s", k)
}
m.Echo("\n")
for i := 0; i < len(msg.Meta[msg.Meta["append"][0]]); i++ {
for _, k := range msg.Meta["append"] {
m.Echo("\t%s", msg.Meta[k][i])
}
m.Echo("\n")
}
}
}
// }}}
}},
},
Index: map[string]*ctx.Context{

143
src/contexts/web/web.go
View File

@ -32,15 +32,18 @@ type WEB struct {
*http.ServeMux
*http.Server
list map[string]string
client *http.Client
cookie map[string]*http.Cookie
list map[string][]string
list_key []string
*ctx.Message
*ctx.Context
}
func (web *WEB) generate(m *ctx.Message, arg string) string { // {{{
add, e := url.Parse(arg)
func (web *WEB) generate(m *ctx.Message, uri string, arg ...string) string { // {{{
add, e := url.Parse(uri)
m.Assert(e)
adds := []string{}
@ -76,12 +79,31 @@ func (web *WEB) generate(m *ctx.Message, arg string) string { // {{{
}
}
args := []string{}
for i := 0; i < len(arg)-1; i += 2 {
args = append(args, arg[i]+"="+arg[i+1])
}
p := strings.Join(args, "&")
if add.RawQuery != "" {
adds = append(adds, "?")
adds = append(adds, add.RawQuery)
if p != "" {
adds = append(adds, "&")
adds = append(adds, p)
}
} else if m.Confs("query") {
adds = append(adds, "?")
adds = append(adds, m.Conf("query"))
if p != "" {
adds = append(adds, "&")
adds = append(adds, p)
}
} else {
if p != "" {
adds = append(adds, "?")
adds = append(adds, p)
}
}
return strings.Join(adds, "")
@ -147,22 +169,23 @@ func (web *WEB) Trans(m *ctx.Message, key string, hand func(*ctx.Message, *ctx.C
// }}}
func (web *WEB) ServeHTTP(w http.ResponseWriter, r *http.Request) { // {{{
web.Log("fuck", nil, "why")
if web.Message != nil {
log.Println()
web.Log("cmd", nil, "%v %s %s", r.RemoteAddr, r.Method, r.URL)
if web.Conf("logheaders") == "yes" {
for k, v := range r.Header {
log.Printf("%s: %v", k, v)
web.Log("info", nil, "%s: %v", k, v)
}
log.Println()
web.Log("info", nil, "")
}
if r.ParseForm(); len(r.PostForm) > 0 {
for k, v := range r.PostForm {
log.Printf("%s: %v", k, v)
web.Log("info", nil, "%s: %v", k, v)
}
log.Println()
web.Log("info", nil, "")
}
}
@ -170,9 +193,9 @@ func (web *WEB) ServeHTTP(w http.ResponseWriter, r *http.Request) { // {{{
if web.Message != nil && web.Conf("logheaders") == "yes" {
for k, v := range w.Header() {
log.Printf("%s: %v", k, v)
web.Log("info", nil, "%s: %v", k, v)
}
log.Println()
web.Log("info", nil, "")
}
}
@ -207,7 +230,7 @@ func (web *WEB) Begin(m *ctx.Message, arg ...string) ctx.Server { // {{{
}
}
web.list = map[string]string{}
web.list = map[string][]string{}
return web
}
@ -348,16 +371,83 @@ var Index = &ctx.Context{Name: "web", Help: "应用中心",
})
} // }}}
}},
"get": &ctx.Command{Name: "get url", Help: "访问URL", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) {
"cookie": &ctx.Command{Name: "cookie add|del arg...", Help: "访问URL", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) {
web, ok := m.Target().Server.(*WEB) // {{{
m.Assert(ok)
uri := web.generate(m, arg[0])
switch len(arg) {
case 0:
for k, v := range web.cookie {
m.Echo("%s: %v\n", k, v.Value)
}
case 1:
if v, ok := web.cookie[arg[0]]; ok {
m.Echo("%s", v.Value)
}
default:
if web.cookie == nil {
web.cookie = make(map[string]*http.Cookie)
}
if v, ok := web.cookie[arg[0]]; ok {
v.Value = arg[1]
} else {
web.cookie[arg[0]] = &http.Cookie{Name: arg[0], Value: arg[1]}
}
}
// }}}
}},
"get": &ctx.Command{Name: "get [method GET|POST] [file filename] arg...", Help: "访问URL",
Formats: map[string]int{"method": 1, "file": 1},
Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) {
web, ok := m.Target().Server.(*WEB) // {{{
m.Assert(ok)
uri := web.generate(m, arg[0], arg[1:]...)
m.Log("info", nil, "GET %s", uri)
res, e := http.Get(uri)
if web.client == nil {
web.client = &http.Client{}
}
method := "GET"
if m.Options("method") {
method = m.Option("method")
}
var body io.Reader
index := strings.Index(uri, "?")
switch method {
case "POST":
if index > 0 {
body = strings.NewReader(uri[index+1:])
uri = uri[:index]
}
}
req, e := http.NewRequest(method, uri, body)
m.Assert(e)
switch method {
case "POST":
if index > 0 {
req.Header.Add("Content-Type", "application/x-www-form-urlencoded")
}
}
for _, v := range web.cookie {
req.AddCookie(v)
}
res, e := web.client.Do(req)
m.Assert(e)
if web.cookie == nil {
web.cookie = make(map[string]*http.Cookie)
}
for _, v := range res.Cookies() {
web.cookie[v.Name] = v
}
for k, v := range res.Header {
m.Log("info", nil, "%s: %v", k, v)
}
@ -386,7 +476,7 @@ var Index = &ctx.Context{Name: "web", Help: "应用中心",
m.Echo(string(buf))
// }}}
}},
"list": &ctx.Command{Name: "list [index|add|del [url]]", Help: "查看、访问、添加url", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) {
"list": &ctx.Command{Name: "list [set|add|del [url]]", Help: "查看、访问、添加url", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) {
web, ok := m.Target().Server.(*WEB) // {{{
m.Assert(ok)
@ -400,16 +490,33 @@ var Index = &ctx.Context{Name: "web", Help: "应用中心",
case 1:
msg := m.Spawn(m.Target()).Cmd("get", web.list[arg[0]])
m.Copy(msg, "result")
case 2:
default:
switch arg[0] {
case "add":
m.Capi("count", 1)
web.list[m.Cap("count")] = arg[1]
web.list[m.Cap("count")] = arg[1:]
web.list_key = append(web.list_key, m.Cap("count"))
m.Capi("count", 1)
case "del":
delete(web.list, arg[1])
case "set":
web.list[arg[1]] = arg[2:]
default:
web.list[arg[0]] = arg[1]
list := []string{}
j := 1
for _, v := range web.list[arg[0]] {
if v == "_" && j < len(arg) {
list = append(list, arg[j])
j++
} else {
list = append(list, v)
}
}
for ; j < len(arg); j++ {
list = append(list, arg[j])
}
msg := m.Spawn(m.Target()).Cmd("get", list)
m.Copy(msg, "result")
}
} // }}}
}},