コード例 #1
13
ファイル: t_parser.go プロジェクト: Bootz/nzbcrawler
//download image from town
func (t *Townparser) downloadImage(url string, name string) error {
	defer func() {
		if err := recover(); err != nil {
			log.Info("%s recovered from panic", TAG)
			return
		}
	}()
	if url == "0" {
		return nil
	}
	exist, err := exists("templates/static/images/" + name + ".jpg")
	if err != nil {
		log.Error("%s %s", TAG, err.Error())
	}

	if !exist {
		resp, err := t.Tc.Get(url)
		if err != nil {
			log.Error("%s image download failed, name: %v, url: %v", TAG, name, url)
			log.Error("%s %s", TAG, err.Error())
			return err
		}
		defer resp.Body.Close()
		if strings.Contains(url, "jpg") || strings.Contains(url, "jpeg") {
			img, _ := jpeg.Decode(resp.Body)
			m := resize.Resize(300, 0, img, resize.Lanczos2Lut)
			out, err := os.Create("templates/static/images/" + name + ".jpg")
			if err != nil {
				log.Error("%s %s", TAG, err.Error())
				return nil
			}

			// write new image to file
			jpeg.Encode(out, m, nil)
			out.Close()
		} else if strings.Contains(url, "png") {
			img, err := png.Decode(resp.Body)
			if err != nil {
				log.Error("%s %s", TAG, err.Error())
			}
			m := resize.Resize(300, 0, img, resize.Lanczos2Lut)
			out, err := os.Create("templates/static/images/" + name + ".png")
			if err != nil {
				log.Error("%s %s", TAG, err.Error())
				return nil
			}

			// write new image to file
			jpeg.Encode(out, m, nil)
			out.Close()
		}
	}
	time.Sleep(200 * time.Millisecond)
	return nil
}
コード例 #2
0
ファイル: gomr.go プロジェクト: postfix/GoMR
//run all the mappers in parallel
func (mr *MapReduceJob) runMappers() {
	mappersGroup := sync.WaitGroup{}

	//open up N mappers in parallel go routines
	for i := 0; i < mr.numMappers; i++ {

		//we add each mapper to the waitgroup before starting, then when it finishes it calls Done
		//this way we can wait for the mappers to finish safely
		mappersGroup.Add(1)
		logging.Info("Starting mapper %d", i)
		go func(i int) {

			for record := range mr.inputChan {

				(*mr.mapReducer).Map(record, mr.mappersOutputChan)
			}

			mappersGroup.Done()
			logging.Info("Mapper %d done", i)
		}(i)
	}

	//wait for the mappers to finish up and then close their output channel
	logging.Info("Waiting for mappers to finish")
	mappersGroup.Wait()
	logging.Info("All mappers finished...")

	close(mr.mappersOutputChan)
}
コード例 #3
0
ファイル: t_client.go プロジェクト: Bootz/nzbcrawler
func (t *Townclient) getSValue() (sValue string) {
	log.Info("%s getting sValue for town login", TAG)
	sValue = ""
	var doc *goquery.Document
	var e error
	log.Info("%s[GET] url: %v", TAG, ROOT)
	if doc, e = goquery.NewDocument(ROOT); e != nil {
		log.Error("%s %s", TAG, e.Error())
		return
	}

	doc.Find("input").Each(func(i int, s *goquery.Selection) {
		attr, exists := s.Attr("name")
		if exists == true {
			if attr == "s" {
				bla, exists := s.Attr("value")
				if exists == true {
					sValue = bla
				}
			}
		}

	})
	log.Info("%s sValue: %v", TAG, sValue)
	return sValue
}
コード例 #4
0
ファイル: g_client.go プロジェクト: Bootz/nzbcrawler
func (g *Ghostclient) getFirstTimeShit() error {
	log.Info("%s[GET] url: %v", TAG, LOGIN)
	log.Info("%s getting cookies", TAG)
	client := &http.Client{}
	req, err := http.NewRequest("GET", LOGIN, nil)
	if err != nil {
		return err
	}

	req.Header.Add("Accept", "text/html, application/xhtml+xml, */*")
	req.Header.Add("Accept-Language", "de-DE")
	req.Header.Add("User-Agent", "Mozilla/5.0 (compatible; MSIE 10.0; Windows NT 6.2; WOW64; Trident/6.0)")
	req.Header.Add("Accept-Encoding", "gzip, deflate")
	req.Header.Add("Connection", "Keep-Alive")
	req.Header.Add("Host", "ghost-of-usenet.org")

	time1 := time.Now()
	g.dumpRequest(req, "ghost_first_req_"+strconv.Itoa(time1.Nanosecond()))

	//connect to sUrl
	resp, err := client.Do(req)
	if err != nil {
		return err
	}

	g.cookies = resp.Cookies()
	g.dumpResponse(resp, "ghost_first_resp_"+strconv.Itoa(time1.Nanosecond()))

	return nil
}
コード例 #5
0
ファイル: t_manager.go プロジェクト: Bootz/nzbcrawler
func (t *Townmanager) Start() {
	t.page = 1
	t.end = false

	tc := &Townclient{}
	tc.User = t.User
	tc.Password = t.Password

	err := t.init(tc)
	log.Info("%s init finished, starting to parse...", TAG)
	if err != nil {
		log.Error("%s init failed", TAG)
		log.Error("%s %s", TAG, err.Error())
		return
	}

	tp := &Townparser{Url: t.url, Tc: tc}

	count, err := tp.ParsePageCount()
	if err != nil {
		log.Error("%s %s", TAG, err.Error())
		return
	}
	t.maxpage = count
	log.Info("%s crawling approximately %v pages", TAG, t.maxpage)
	t.saveReleases(tp.Rel)

	i := 1
	for {
		if i == 1 {
			err = tp.ParseReleases(false)
			if err != nil {
				log.Error("%s %s", TAG, err.Error())
				break
			}
		} else {
			tp = nil
			tp = &Townparser{Url: t.url + "&pp=25&page=" + strconv.Itoa(i), Tc: tc}
			err = tp.ParseReleases(true)
			if err != nil {
				log.Error("%s %s", TAG, err.Error())
				break
			}
		}
		log.Info("%s crawled page %v/%v", TAG, i, t.maxpage)
		t.saveReleases(tp.Rel)
		time.Sleep(5 * time.Second)
		i++
		if i == t.maxpage+1 {
			break
		}
		if t.end {
			log.Info("%s found old end point", TAG)
			break
		}
	}
	log.Info("%s parser closing", TAG)

}
コード例 #6
0
ファイル: cache.go プロジェクト: Bootz/nzbcrawler
func GoRuntimeStats() {
	m := &runtime.MemStats{}

	log.Info("# goroutines: %v", runtime.NumGoroutine())
	runtime.ReadMemStats(m)
	log.Info("Memory Acquired: %vmb", (m.Sys / 1024 / 1024))
	log.Info("Memory Used    : %vmb", (m.Alloc / 1024 / 1024))
}
コード例 #7
0
ファイル: api.go プロジェクト: sguzwf/vertex
// configure registers the API's routes on a router. If the passed router is nil, we create a new one and return it.
// The nil mode is used when an API is run in stand-alone mode.
func (a *API) configure(router *httprouter.Router) *httprouter.Router {

	if router == nil {
		router = httprouter.New()
	}

	for i, route := range a.Routes {

		if err := route.parseInfo(route.Path); err != nil {
			logging.Error("Error parsing info for %s: %s", route.Path, err)
		}
		a.Routes[i] = route
		h := a.handler(route)

		pth := a.FullPath(route.Path)

		if route.Methods&GET == GET {
			logging.Info("Registering GET handler %v to path %s", h, pth)
			router.Handle("GET", pth, h)
		}
		if route.Methods&POST == POST {
			logging.Info("Registering POST handler %v to path %s", h, pth)
			router.Handle("POST", pth, h)

		}

	}

	chain := buildChain(a.SwaggerMiddleware...)
	if chain == nil {
		chain = buildChain(a.swaggerHandler())
	} else {
		chain.append(a.swaggerHandler())
	}

	// Server the API documentation swagger
	router.GET(a.FullPath("/swagger"), a.middlewareHandler(chain, nil, nil))

	chain = buildChain(a.TestMiddleware...)
	if chain == nil {
		chain = buildChain(a.testHandler())
	} else {
		chain.append(a.testHandler())
	}

	router.GET(path.Join("/test", a.root(), ":category"), a.middlewareHandler(chain, nil, nil))

	// Redirect /$api/$version/console => /console?url=/$api/$version/swagger
	uiPath := fmt.Sprintf("/console?url=%s", url.QueryEscape(a.FullPath("/swagger")))
	router.Handler("GET", a.FullPath("/console"), http.RedirectHandler(uiPath, 301))

	return router

}
コード例 #8
0
ファイル: runner.go プロジェクト: Bootz/nzbcrawler
func (r *Runner) Start() {
	go func() {
		timeout := time.Second * 1
		for {
			select {
			case <-time.After(timeout):
				if !r.checkTown() {
					timeout = time.Minute * 5
					log.Info("town: trying login again in %v minute", timeout)
				} else {
					tm := town.Townmanager{User: r.Server.Config2.TownName, Password: r.Server.Config2.TownPassword, DB: r.Server.RelDB}
					go tm.Start()
					dur, err := time.ParseDuration(r.Server.Config2.Timeout)
					if err != nil {
						log.Error(err.Error())
					}
					c := time.Tick(dur)
					for _ = range c {
						log.Info("tick start town")
						go tm.Start()
					}
				}
			}
			break
		}
		return
	}()
	go func() {
		timeout := time.Second * 2
		for {
			select {
			case <-time.After(timeout):
				if !r.checkTown() {
					timeout = time.Minute * 5
					log.Info("ghost: trying to login again in %v minute", timeout)
				} else {
					gm := ghost.Ghostmanager{User: r.Server.Config2.GhostName, Password: r.Server.Config2.GhostPassword, DB: r.Server.RelDB}
					go gm.Start()
					dur, err := time.ParseDuration(r.Server.Config2.Timeout)
					if err != nil {
						log.Error(err.Error())
					}
					c := time.Tick(dur)
					for _ = range c {
						log.Info("tick start ghost")
						go gm.Start()
					}
				}
			}
			break
		}
		return
	}()
}
コード例 #9
0
ファイル: g_client.go プロジェクト: Bootz/nzbcrawler
// Logs into town.ag and returns the response cookies
func (g *Ghostclient) Login() error {
	log.Info("%s login process started", TAG)

	g.getFirstTimeShit()

	param := url.Values{}
	param.Set("url", "index.php")
	param.Add("send", "send")
	param.Add("sid", "")
	param.Add("l_username", g.User)
	param.Add("l_password", g.Password)
	param.Add("submit", "Anmelden")

	client := &http.Client{}
	req, err := http.NewRequest("POST", LOGIN, strings.NewReader(param.Encode()))

	if err != nil {
		return err
	}

	log.Info("%s[POST] url: %v", TAG, LOGIN)

	if g.cookies != nil {
		for _, cookie := range g.cookies {
			req.AddCookie(cookie)
		}
	}
	req.Header.Add("Accept", "text/html, application/xhtml+xml, */*")
	req.Header.Add("Referer", "http://ghost-of-usenet.org/index.php")
	req.Header.Add("Accept-Language", "de-DE")
	req.Header.Add("User-Agent", "Mozilla/5.0 (compatible; MSIE 10.0; Windows NT 6.2; WOW64; Trident/6.0)")
	req.Header.Add("Content-Type", "application/x-www-form-urlencoded")
	req.Header.Add("Accept-Encoding", "gzip, deflate")
	req.Header.Add("Host", "ghost-of-usenet.org")

	length := strconv.Itoa(len(param.Encode()))
	req.Header.Add("Content-Length", length)
	req.Header.Add("Connection", "Keep-Alive")
	req.Header.Add("Pragma", "no-cache")

	g.dumpRequest(req, "town_login_req")

	resp, err := client.Do(req)
	if err != nil {
		return err
	}

	defer resp.Body.Close()

	g.dumpResponse(resp, "town_login_resp")
	g.cookies = resp.Cookies()
	return nil
}
コード例 #10
0
ファイル: g_manager.go プロジェクト: Bootz/nzbcrawler
func (g *Ghostmanager) Start() {

	g.end = false

	gc := &Ghostclient{}
	gc.User = g.User
	gc.Password = g.Password

	err := g.init(gc)
	log.Info("%s init finished, starting to parse...", TAG)
	if err != nil {
		log.Error("%s init failed", TAG)
		log.Error("%s %s", TAG, err.Error())
		return
	}

	tp := &Ghostparser{Url: g.url, Gc: gc}

	i := 1
	for {
		if i == 1 {
			err = tp.ParseReleases()
			if err != nil {
				log.Error("%s %s", TAG, err.Error())
				break
			}
			g.maxpage = tp.Count
			log.Info("%s crawling approximately %v pages", TAG, g.maxpage)
		} else {
			tp = nil
			tp = &Ghostparser{Url: g.url + "&page=" + strconv.Itoa(i), Gc: gc}
			err = tp.ParseReleases()
			if err != nil {
				log.Error("%s %s", TAG, err.Error())
				break
			}
		}
		g.saveReleases(tp.Rel)
		log.Info("%s crawled page %v/%v", TAG, i, g.maxpage)
		time.Sleep(5 * time.Second)
		i++
		if i == g.maxpage+1 {
			break
		}
		if g.end {
			log.Info("%s found old end point", TAG)
			break
		}
	}
	log.Info("%s closing", TAG)
}
コード例 #11
0
ファイル: gomr.go プロジェクト: postfix/GoMR
//collect the mappers' output into the aggregate dictionary
func (mr *MapReduceJob) collectMappersOutput() {

	logging.Info("Collecting mappers output")

	tracker := progressTracker(5, "items")
	for kv := range mr.mappersOutputChan {
		mr.aggregate[kv.Key] = append(mr.aggregate[kv.Key], kv.Val)
		tracker <- 1
	}

	close(tracker)
	logging.Info("FINISHED Collecting mappers output")

}
コード例 #12
0
ファイル: route.go プロジェクト: sguzwf/vertex
func (r *Route) parseInfo(path string) error {

	ri, err := schema.NewRequestInfo(reflect.TypeOf(r.Handler), path, r.Description, r.Returns)
	if err != nil {
		return err
	}

	// search for custom unmarshallers in the request info
	for _, param := range ri.Params {
		if param.Type.Kind() == reflect.Struct {

			logging.Debug("Checking unmarshaller for %s", param.Type)
			val := reflect.Zero(param.Type).Interface()

			if unm, ok := val.(Unmarshaler); ok {
				logging.Info("Registering unmarshaller for %#v", val)

				schemaDecoder.RegisterConverter(val, gorilla.Converter(func(s string) reflect.Value {
					return reflect.ValueOf(unm.UnmarshalRequestData(s))
				}))

			}
		}

	}

	r.requestInfo = ri
	return nil

}
コード例 #13
0
ファイル: gomr.go プロジェクト: postfix/GoMR
//run all steps
func (mr *MapReduceJob) Run() {

	//read the input in a seaparate goroutine
	go mr.readInput()

	//aggregate results in a separate goroutine while the mappers are working
	go mr.collectMappersOutput()

	//start mappers and block until they are all finished
	mr.runMappers()

	//run the output collector in a different goroutine

	//now run the reducers. For this to advance, some other gorooutine needs to collect output
	go mr.runReducers()
	outputHandler := sync.WaitGroup{}
	outputHandler.Add(1)
	go func() {
		(*mr.mapReducer).HandleOutput(mr.OutputChan)
		outputHandler.Done()
	}()

	outputHandler.Wait()

	logging.Info("Finished!")

}
コード例 #14
0
ファイル: config.go プロジェクト: sguzwf/vertex
func ReadConfigs() error {

	if err := autoflag.Load(gofigure.DefaultLoader, &Config); err != nil {
		logging.Error("Error loading configs: %v", err)
		return err
	}
	logging.Info("Read configs: %#v", &Config)

	for k, m := range Config.APIConfigs {

		if conf, found := Config.apiconfs[k]; found && conf != nil {

			b, err := yaml.Marshal(m)
			if err == nil {

				if err := yaml.Unmarshal(b, conf); err != nil {
					logging.Error("Error reading config for API %s: %s", k, err)
				} else {
					logging.Debug("Unmarshaled API config for %s: %#v", k, conf)
				}

			} else {

				logging.Error("Error marshalling config for API %s: %s", k, err)

			}
		} else {
			logging.Warning("API Section %s in config file not registered with server", k)
		}

	}

	return nil

}
コード例 #15
0
ファイル: t_client.go プロジェクト: Bootz/nzbcrawler
//http get to the given ressource
func (t *Townclient) Get(sUrl string) (*http.Response, error) {
	if strings.Contains(sUrl, "jpg") || strings.Contains(sUrl, "png") || strings.Contains(sUrl, "gif") || strings.Contains(sUrl, "jpeg") {
	} else {
		log.Info("%s[GET] url: %v", TAG, sUrl)
	}

	client := &http.Client{}
	req, err := http.NewRequest("GET", sUrl, nil)
	if err != nil {
		log.Error("%s couldn't create Request to: %v", TAG, sUrl)
		return nil, err
	}

	t.addHeader(req)

	if t.cookies != nil {
		for _, cookie := range t.cookies {
			req.AddCookie(cookie)
		}
	}

	time1 := time.Now()
	t.dumpRequest(req, "town_get_req_"+strconv.Itoa(time1.Nanosecond()))

	//connect to sUrl
	resp, err := client.Do(req)
	if err != nil {
		log.Error("%s couldn't connect to: %v", TAG, sUrl)
		return nil, err
	}

	t.dumpResponse(resp, "town_get_resp_"+strconv.Itoa(time1.Nanosecond()))

	return resp, nil
}
コード例 #16
0
ファイル: request.go プロジェクト: sguzwf/vertex
// Detect if the request is secure or not, based on either TLS info or http headers/url
func (r *Request) parseSecure() {

	logging.Info("Parsing secure. TLS: %v, URI: %s, Headers: %#v", r.TLS, r.RequestURI, r.Header)
	if r.TLS != nil {
		r.Secure = true
		return
	}

	if u, err := url.ParseRequestURI(r.RequestURI); err == nil {

		if u.Scheme == "https" {
			r.Secure = true
			return
		}
	}

	xfp := r.Header.Get("X-Forwarded-Proto")
	if xfp == "" {
		xfp = r.Header.Get("X-Scheme")
	}

	if xfp == "https" {
		r.Secure = true
	}
}
コード例 #17
0
ファイル: validate.go プロジェクト: sguzwf/vertex
func (rv *RequestValidator) Validate(request interface{}, r *http.Request) error {

	val := reflect.ValueOf(request)
	if val.Kind() == reflect.Ptr {
		val = val.Elem()
	}

	//go over all the validators
	for _, v := range rv.fieldValidators {

		// find the field in the struct. we assume it's there since we build the validators on start time
		field := val.FieldByName(v.GetKey())

		// if the arg is optional and not set, we set the default
		if v.IsOptional() && (!field.IsValid() || r.FormValue(v.GetParamName()) == "") {
			def, ok := v.GetDefault()
			if ok {
				logging.Info("Default value for %s: %v", v.GetKey(), def)
				field.Set(reflect.ValueOf(def).Convert(field.Type()))
			}
		}

		// now we validate!
		e := v.Validate(field, r)

		if e != nil {
			logging.Error("Could not validate field %s: %s", v.GetParamName(), e)
			return e
		}

	}

	return nil
}
コード例 #18
0
ファイル: cache.go プロジェクト: Bootz/nzbcrawler
func (c *Cache) freeMemory() {
	if c.isRunning {
		return
	}
	c.isRunning = true
	log.Info("[Cache] freeMemory() cachesize: %vmb, size to be freed: %vmb", c.GetSizeInMb(), (c.sizefree / 1024 / 1024))
	low := uint32(1)
	count := c.GetSize()
	start := time.Now()
	sec20 := int64(time.Second * 20)
	sec5 := int64(time.Second * 5)
	ignoreimmunity := false
	b := false
	for {
		for key, value := range c.cache {
			if c.sizemax-c.size < c.sizefree {
				if value.AccessCount <= low {
					//5 minute immunity to protect freshly added files
					if int64(start.Sub(value.Added)) >= sec20 || ignoreimmunity {
						c.Remove(key)
					}
				}
			} else {
				b = true
				break
			}
		}
		low++
		if int64(time.Now().Sub(start)) >= sec5 {
			ignoreimmunity = true
		}
		if b {
			break
		}
	}
	end := time.Now()
	log.Info("[Cache] removed %v elements in %fsec", count-c.GetSize(), end.Sub(start).Seconds())
	start = time.Now()
	runtime.GC()
	debug.FreeOSMemory()
	end = time.Now()
	log.Info("[Cache] run gc manually to free up memory asap, took %fsec", end.Sub(start).Seconds())
	c.isRunning = false
	//GoRuntimeStats()
}
コード例 #19
0
ファイル: gomr.go プロジェクト: postfix/GoMR
//Read the input from the user's reader and push it to the input chan
func (mr *MapReduceJob) readInput() {

	tracker := progressTracker(2, "docs")
	logging.Info("Reading input")
	for {
		record, err := (*mr.mapReducer).Read()

		if err != nil {
			logging.Error("Error: Aborting read: %s", err)
			break
		}
		mr.inputChan <- record
		tracker <- 1

	}
	close(tracker)
	close(mr.inputChan)
	logging.Info("Finished reading input!")
}
コード例 #20
0
ファイル: g_client.go プロジェクト: Bootz/nzbcrawler
//return the Daily url or "" if something went wrong
func (g *Ghostclient) GetDailyUrl() (string, error) {
	client := &http.Client{
		CheckRedirect: Redirect,
	}
	log.Info("%s[GET] url: %v", TAG, DAILY)
	req, err := http.NewRequest("GET", DAILY, nil)
	if err != nil {
		log.Error("%s %s", TAG, err.Error())
		return "", err
	}

	req.Header.Add("Accept", "text/html, application/xhtml+xml, */*")
	req.Header.Add("Referer", "http://ghost-of-usenet.org/index.php")
	req.Header.Add("Accept-Language", "de-DE")
	req.Header.Add("User-Agent", "Mozilla/5.0 (compatible; MSIE 10.0; Windows NT 6.2; WOW64; Trident/6.0)")
	req.Header.Add("Accept-Encoding", "gzip, deflate")
	req.Header.Add("Host", "ghost-of-usenet.org")
	req.Header.Add("Connection", "Keep-Alive")

	if g.cookies != nil {
		for _, cookie := range g.cookies {
			req.AddCookie(cookie)
		}
	}

	time1 := time.Now()
	g.dumpRequest(req, "daily_req_"+strconv.Itoa(time1.Nanosecond()))

	resp, err := client.Do(req)
	if resp == nil {
		return "", err
	}

	g.dumpResponse(resp, "daily_resp_"+strconv.Itoa(time1.Nanosecond()))

	url, err := resp.Location()
	if err != nil {
		return "", err
	}
	log.Info("%s daily url: %v", TAG, url.String())
	return url.String(), nil

}
コード例 #21
0
ファイル: t_client.go プロジェクト: Bootz/nzbcrawler
// Logs into town.ag and returns the response cookies
func (t *Townclient) Login() error {
	log.Info("%s login process started", TAG)

	sValue := t.getSValue()

	if sValue == "" {
		return errors.New("couldnt find SValue for the Town login")
	}

	param := url.Values{}
	param.Set("do", "login")
	param.Add("s", sValue)
	param.Add("securitytoken", "guest")
	param.Add("vb_login_username", t.User)
	param.Add("vb_login_password", "")
	param.Add("cookieuser", "1")
	param.Add("vb_login_md5password", t.Password)
	param.Add("vb_login_md5password_utf", t.Password)
	param.Add("url", "/v2/")

	client := &http.Client{}
	req, err := http.NewRequest("POST", LOGIN, strings.NewReader(param.Encode()))
	if err != nil {
		return err
	}

	log.Info("%s[POST] url: %v", TAG, LOGIN)
	t.addHeader(req)

	t.dumpRequest(req, "town_login_req")

	resp, err := client.Do(req)
	if err != nil {
		return err
	}

	defer resp.Body.Close()

	t.dumpResponse(resp, "town_login_resp")
	t.cookies = resp.Cookies()
	return nil
}
コード例 #22
0
ファイル: example.go プロジェクト: postfix/GoMR
func main() {

	runtime.GOMAXPROCS(8)

	my := MyMapReducer{scanner: *bufio.NewScanner(os.Stdin)}

	mr := gomr.NewMapReduceJob(&my, 80, 80)

	st := time.Now()
	mr.Run()

	logging.Info("Finished in %s", time.Since(st))
}
コード例 #23
0
ファイル: oauth.go プロジェクト: sguzwf/vertex
func (o *OAuthMiddleware) LoginHandler() vertex.Route {

	handler := func(w http.ResponseWriter, r *vertex.Request) (interface{}, error) {
		code := r.FormValue("code")
		logging.Info("Got code: %s", code)

		tok, err := o.conf.Exchange(oauth2.NoContext, code)
		if err != nil {
			return nil, vertex.UnauthorizedError("Could not log you in: %s", err)
		}

		user, err := o.userValidator.Login(tok)
		if err != nil {
			return nil, vertex.UnauthorizedError("Could not validate user for login: %s", err)
		}

		enc, err := o.userValidator.EncodeToken(user)
		if err != nil {
			return nil, vertex.UnauthorizedError("Could not validate encode user token: %s", err)
		}

		o.setCookie(w, enc, r.Host)

		if cook, err := r.Cookie(nextUrl); err == nil && cook != nil && cook.Value != "" {
			logging.Info("Found nextUrl from before auth denied. Redirecting to %s", cook.Value)
			http.Redirect(w, r.Request, cook.Value, http.StatusTemporaryRedirect)
			return nil, vertex.Hijacked
		}

		return "Success Logging In", nil
	}
	return vertex.Route{
		Path:        loginPath,
		Description: "OAuth Login",
		Handler:     vertex.HandlerFunc(handler),
		Methods:     vertex.GET,
	}

}
コード例 #24
0
ファイル: g_parser.go プロジェクト: Bootz/nzbcrawler
//parse the http resp from Townclient
func (g *Ghostparser) ParseReleases() error {
	log.Info("%s parsing %v", TAG, g.Url)

	resp, err := g.Gc.Get(g.Url)
	if err != nil {
		log.Error("%s %s", TAG, err.Error())
		return err
	}
	defer resp.Body.Close()

	respbody, err := html.Parse(resp.Body)
	doc := goquery.NewDocumentFromNode(respbody)

	var rel Release
	doc.Find("table").Each(func(a int, sa *goquery.Selection) {
		if a == 10 { //get the right table
			sa.Find("tr").Each(func(b int, sb *goquery.Selection) {
				sb.Find("td").Each(func(c int, sc *goquery.Selection) {
					if c == 2 {
						rel = Release{}
						g.getUrlAndTagAndName(&rel, sc)

						if rel.Name != "" {
							rel.Time = time.Now().Unix()
							rel.Checksum = g.encodeName(rel.Url)
							rel.checkQual()
							if rel.Name != "" {
								rel.Hits = 0
								rel.Rating = 0
								g.downloadImage(rel.Url, rel.Checksum)
								g.addRelease(rel)
							}
						}
					}
				})
			})
		}
		if g.Count == 0 { //get page count
			if a == 51 {
				sa.Find("a").Each(func(d int, sd *goquery.Selection) {
					if d == 3 {
						g.Count, err = strconv.Atoi(sd.Text())
					}
				})
			}
		}
	})

	return nil
}
コード例 #25
0
ファイル: gomr.go プロジェクト: postfix/GoMR
//run the reducers
func (mr *MapReduceJob) runReducers() {
	reducersGroup := sync.WaitGroup{}

	logging.Info("Runnign reducers!")

	//Run N reducers in parallel
	for i := 0; i < mr.numReducers; i++ {

		reducersGroup.Add(1)
		go func() {
			for record := range mr.reducersInputChan {
				(*mr.mapReducer).Reduce(record.Key, record.Val.([]interface{}), mr.OutputChan)

			}
			reducersGroup.Done()
		}()
	}

	tracker := progressTracker(5, "items")
	//push the output from the aggregate dictionary to the reducers' input channel
	for k := range mr.aggregate {

		mr.reducersInputChan <- Record{k, mr.aggregate[k]}
		tracker <- 1
	}
	close(tracker)

	//we close the input channel, causing all reduce loops to exit
	close(mr.reducersInputChan)
	//let's wait for them to actually finish
	reducersGroup.Wait()
	logging.Info("FINISHED Runnign reducers!")

	//we now close the output channel
	close(mr.OutputChan)

}
コード例 #26
0
ファイル: t_client.go プロジェクト: Bootz/nzbcrawler
//return the Daily url or "" if something went wrong
func (t *Townclient) GetDailyUrl() (string, error) {
	log.Info("%s getting Daily Url for town", TAG)
	client := &http.Client{
		CheckRedirect: Redirect,
	}
	req, err := http.NewRequest("GET", DAILY, nil)
	if err != nil {
		log.Error("%s %s", TAG, err.Error())
		return "", err
	}

	t.addHeader(req)

	if t.cookies != nil {
		for _, cookie := range t.cookies {
			req.AddCookie(cookie)
		}
	}

	resp, err := client.Do(req)
	if resp == nil {
		return "", err
	}
	defer resp.Body.Close()

	time1 := time.Now()
	t.dumpResponse(resp, "daily"+strconv.Itoa(time1.Nanosecond()))

	url, err := resp.Location()
	if err != nil {
		return "", err
	}

	log.Info("%s daily url: %v", TAG, url.String())
	return url.String(), nil

}
コード例 #27
0
ファイル: gomr.go プロジェクト: postfix/GoMR
//Create a new mapreduce job. Pass a mapreducer object and the number of mappers and reducers in the pools
func NewMapReduceJob(mr MapReducer, numReducers int, numMappers int) *MapReduceJob {

	job := &MapReduceJob{
		mappersOutputChan: make(chan Record),
		OutputChan:        make(chan Record),
		reducersInputChan: make(chan Record),
		inputChan:         make(chan interface{}),
		mapReducer:        &mr,
		aggregate:         make(map[string][]interface{}),
		numReducers:       numReducers,
		numMappers:        numMappers,
	}
	logging.Info("Created job with MapReducer %s, %d mappers, %d reducers", mr, numMappers, numReducers)
	return job
}
コード例 #28
0
ファイル: oauth.go プロジェクト: sguzwf/vertex
func (o *OAuthMiddleware) Handle(w http.ResponseWriter, r *vertex.Request, next vertex.HandlerFunc) (interface{}, error) {

	if strings.HasSuffix(r.URL.Path, loginPath) {
		return next(w, r)
	}
	user, err := o.getToken(r)
	if err != nil {
		o.redirect(w, r)
		return nil, vertex.Hijacked

	}

	logging.Info("Request authenticated. Continuing!")
	r.SetAttribute(AttrUser, user)

	return next(w, r)
}
コード例 #29
0
ファイル: render.go プロジェクト: sguzwf/vertex
func NewHTMLRendererFiles(funcMap map[string]interface{}, fileNames ...string) *HTMLRenderer {

	if funcMap == nil {
		funcMap = template.FuncMap{}
	}

	tpl, err := template.New("html").Funcs(funcMap).ParseFiles(fileNames...)
	if err != nil {
		panic(err)
	}

	logging.Info("Created template from files %s (%#v)", fileNames, tpl)
	tpl.ExecuteTemplate(os.Stderr, "html", nil)
	return &HTMLRenderer{
		template: tpl,
	}
}
コード例 #30
0
ファイル: testing.go プロジェクト: sguzwf/vertex
// invokeTest runs a tester and prints the output
func (t *testRunner) invokeTest(path string, tc Tester) *testResult {

	if t.shouldRun(tc) {

		var result testResult
		if tc == nil || t.shouldRun(tc) {
			result = t.runTest(tc, path)
			logging.Info("Test result for %s: %#v", path, result)
			if err := t.formatter.format(result); err != nil {
				logging.Error("Error running formatter: %s", err)
			}
			return &result
		}

	}

	return nil
}